diff options
| -rw-r--r-- | services/core/java/com/android/server/pm/UserManagerService.java | 5 | ||||
| -rw-r--r-- | services/tests/mockingservicestests/src/com/android/server/pm/UserManagerServiceDemoModeTest.kt | 80 |
2 files changed, 85 insertions, 0 deletions
diff --git a/services/core/java/com/android/server/pm/UserManagerService.java b/services/core/java/com/android/server/pm/UserManagerService.java index 2305d6c9fba9..75b453184db8 100644 --- a/services/core/java/com/android/server/pm/UserManagerService.java +++ b/services/core/java/com/android/server/pm/UserManagerService.java @@ -2285,6 +2285,11 @@ public class UserManagerService extends IUserManager.Stub { throw new SecurityException("You need MANAGE_USERS permission to query if u=" + userId + " is a demo user"); } + + if (SystemProperties.getBoolean("ro.boot.arc_demo_mode", false)) { + return true; + } + synchronized (mUsersLock) { UserInfo userInfo = getUserInfoLU(userId); return userInfo != null && userInfo.isDemo(); diff --git a/services/tests/mockingservicestests/src/com/android/server/pm/UserManagerServiceDemoModeTest.kt b/services/tests/mockingservicestests/src/com/android/server/pm/UserManagerServiceDemoModeTest.kt new file mode 100644 index 000000000000..dfdb0c7241c4 --- /dev/null +++ b/services/tests/mockingservicestests/src/com/android/server/pm/UserManagerServiceDemoModeTest.kt @@ -0,0 +1,80 @@ +/* + * Copyright (C) 2023 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package com.android.server.pm + +import android.content.res.Configuration +import android.os.Looper +import android.os.SystemProperties +import android.os.UserHandle +import android.util.ArrayMap +import com.android.server.LockGuard +import com.android.server.extendedtestutils.wheneverStatic +import com.android.server.testutils.whenever +import com.google.common.truth.Truth.assertThat +import org.junit.Before +import org.junit.Rule +import org.junit.Test +import org.junit.runner.RunWith +import org.junit.runners.JUnit4 +import org.mockito.MockitoAnnotations + +@RunWith(JUnit4::class) +class UserManagerServiceDemoModeTest { + private lateinit var ums: UserManagerService + + @Rule + @JvmField + val rule = MockSystemRule() + + @Before + @Throws(Exception::class) + fun setUp() { + MockitoAnnotations.initMocks(this) + rule.system().stageNominalSystemState() + + if (Looper.myLooper() == null) { + Looper.prepare() + } + + wheneverStatic { LockGuard.installNewLock(LockGuard.INDEX_USER) }.thenReturn(Object()) + whenever(rule.mocks().systemConfig.getAndClearPackageToUserTypeWhitelist()).thenReturn(ArrayMap<String, Set<String>>()) + whenever(rule.mocks().systemConfig.getAndClearPackageToUserTypeBlacklist()).thenReturn(ArrayMap<String, Set<String>>()) + whenever(rule.mocks().resources.getStringArray(com.android.internal.R.array.config_defaultFirstUserRestrictions)).thenReturn(arrayOf<String>()) + whenever(rule.mocks().resources.configuration).thenReturn(Configuration()) + + ums = UserManagerService(rule.mocks().context) + } + + @Test + fun isDemoUser_returnsTrue_whenSystemPropertyIsSet() { + wheneverStatic { SystemProperties.getBoolean("ro.boot.arc_demo_mode", false) }.thenReturn(true) + + assertThat(ums.isDemoUser(0)).isTrue() + } + + @Test + fun isDemoUser_returnsFalse_whenSystemPropertyIsSet() { + wheneverStatic { SystemProperties.getBoolean("ro.boot.arc_demo_mode", false) }.thenReturn(false) + + assertThat(ums.isDemoUser(0)).isFalse() + } + + @Test + fun isDemoUser_returnsFalse_whenSystemPropertyIsNotSet() { + assertThat(ums.isDemoUser(0)).isFalse() + } +}
\ No newline at end of file |