From 0149363a394e3a565e6958cdca41b5621c4209d2 Mon Sep 17 00:00:00 2001 From: Ben Murdoch Date: Thu, 23 Jan 2020 15:41:26 +0000 Subject: [PATCH] Show "Add users from lock screen" dependent on secure lock screen. SysUI only enforces this setting while a secure lock screen is set, so it doesn't make sense to show it configurable while an insecure lock screen is set. Bug: 148015798 Test: make -j64 RunSettingsRoboTests ROBOTEST_FILTER="com.android.settings.users.AddUserWhenLockedPreferenceControllerTest" Change-Id: I186b2339e9575e20b55e2e5c1dab3de624848fc5 --- ...AddUserWhenLockedPreferenceController.java | 9 +++- .../android/settings/users/UserSettings.java | 5 +- ...serWhenLockedPreferenceControllerTest.java | 49 ++++++++++++++++++- 3 files changed, 58 insertions(+), 5 deletions(-) diff --git a/src/com/android/settings/users/AddUserWhenLockedPreferenceController.java b/src/com/android/settings/users/AddUserWhenLockedPreferenceController.java index f931fa4f662..e0a9c26a88f 100644 --- a/src/com/android/settings/users/AddUserWhenLockedPreferenceController.java +++ b/src/com/android/settings/users/AddUserWhenLockedPreferenceController.java @@ -16,20 +16,25 @@ package com.android.settings.users; import android.content.Context; +import android.os.UserHandle; import android.provider.Settings; import androidx.preference.Preference; +import com.android.internal.widget.LockPatternUtils; import com.android.settings.core.TogglePreferenceController; import com.android.settingslib.RestrictedSwitchPreference; public class AddUserWhenLockedPreferenceController extends TogglePreferenceController { private final UserCapabilities mUserCaps; + private final LockPatternUtils mLockPatternUtils; - public AddUserWhenLockedPreferenceController(Context context, String key) { + public AddUserWhenLockedPreferenceController( + Context context, String key, LockPatternUtils lockPatternUtils) { super(context, key); mUserCaps = UserCapabilities.create(context); + mLockPatternUtils = lockPatternUtils; } @Override @@ -53,6 +58,8 @@ public class AddUserWhenLockedPreferenceController extends TogglePreferenceContr return DISABLED_FOR_USER; } else if (mUserCaps.disallowAddUser() || mUserCaps.disallowAddUserSetByAdmin()) { return DISABLED_FOR_USER; + } else if (!mLockPatternUtils.isSecure(UserHandle.myUserId())) { + return CONDITIONALLY_UNAVAILABLE; } else { return mUserCaps.mUserSwitcherEnabled ? AVAILABLE : CONDITIONALLY_UNAVAILABLE; } diff --git a/src/com/android/settings/users/UserSettings.java b/src/com/android/settings/users/UserSettings.java index cd305e97432..70af89985c8 100644 --- a/src/com/android/settings/users/UserSettings.java +++ b/src/com/android/settings/users/UserSettings.java @@ -240,7 +240,7 @@ public class UserSettings extends SettingsPreferenceFragment } mAddUserWhenLockedPreferenceController = new AddUserWhenLockedPreferenceController( - activity, KEY_ADD_USER_WHEN_LOCKED); + activity, KEY_ADD_USER_WHEN_LOCKED, new LockPatternUtils(getPrefContext())); mMultiUserFooterPreferenceController = new MultiUserFooterPreferenceController(activity, KEY_MULTIUSER_FOOTER); @@ -1290,7 +1290,8 @@ public class UserSettings extends SettingsPreferenceFragment boolean suppressAllPage) { final List niks = super.getNonIndexableKeysFromXml(context, xmlResId, suppressAllPage); - new AddUserWhenLockedPreferenceController(context, KEY_ADD_USER_WHEN_LOCKED) + new AddUserWhenLockedPreferenceController( + context, KEY_ADD_USER_WHEN_LOCKED, new LockPatternUtils(context)) .updateNonIndexableKeys(niks); new AutoSyncDataPreferenceController(context, null /* parent */) .updateNonIndexableKeys(niks); diff --git a/tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java index b27b6367879..cb5791ea06f 100644 --- a/tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java +++ b/tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java @@ -18,6 +18,7 @@ package com.android.settings.users; import static com.google.common.truth.Truth.assertThat; import static org.mockito.Answers.RETURNS_DEEP_STUBS; +import static org.mockito.ArgumentMatchers.anyInt; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; @@ -27,6 +28,8 @@ import android.provider.Settings.Global; import androidx.preference.PreferenceScreen; +import com.android.internal.widget.LockPatternUtils; +import com.android.settings.core.BasePreferenceController; import com.android.settings.testutils.shadow.ShadowUserManager; import com.android.settingslib.RestrictedSwitchPreference; @@ -51,13 +54,16 @@ public class AddUserWhenLockedPreferenceControllerTest { private Context mContext; private ShadowUserManager mUserManager; private AddUserWhenLockedPreferenceController mController; + private LockPatternUtils mLockPatternUtils; @Before public void setUp() { MockitoAnnotations.initMocks(this); mContext = RuntimeEnvironment.application; + mLockPatternUtils = mock(LockPatternUtils.class); mUserManager = ShadowUserManager.getShadow(); - mController = new AddUserWhenLockedPreferenceController(mContext, "fake_key"); + mController = new AddUserWhenLockedPreferenceController( + mContext, "fake_key", mLockPatternUtils); mUserManager.setSupportsMultipleUsers(true); } @@ -69,6 +75,7 @@ public class AddUserWhenLockedPreferenceControllerTest { @Test public void displayPref_NotAdmin_shouldNotDisplay() { mUserManager.setIsAdminUser(false); + when(mLockPatternUtils.isSecure(anyInt())).thenReturn(true); final RestrictedSwitchPreference preference = mock(RestrictedSwitchPreference.class); when(preference.getKey()).thenReturn(mController.getPreferenceKey()); when(mScreen.findPreference(preference.getKey())).thenReturn(preference); @@ -81,6 +88,7 @@ public class AddUserWhenLockedPreferenceControllerTest { @Test public void updateState_NotAdmin_shouldNotDisplayPreference() { mUserManager.setIsAdminUser(false); + when(mLockPatternUtils.isSecure(anyInt())).thenReturn(true); final RestrictedSwitchPreference preference = mock(RestrictedSwitchPreference.class); mController.updateState(preference); @@ -93,8 +101,9 @@ public class AddUserWhenLockedPreferenceControllerTest { mUserManager.setIsAdminUser(true); mUserManager.setUserSwitcherEnabled(true); mUserManager.setSupportsMultipleUsers(true); + when(mLockPatternUtils.isSecure(anyInt())).thenReturn(true); final AddUserWhenLockedPreferenceController controller = - new AddUserWhenLockedPreferenceController(mContext, "fake_key"); + new AddUserWhenLockedPreferenceController(mContext, "fake_key", mLockPatternUtils); final RestrictedSwitchPreference preference = mock(RestrictedSwitchPreference.class); controller.updateState(preference); @@ -143,4 +152,40 @@ public class AddUserWhenLockedPreferenceControllerTest { assertThat(Global.getInt(mContext.getContentResolver(), Global.ADD_USERS_WHEN_LOCKED, 0)) .isEqualTo(0); } + + @Test + public void updateState_insecureLockScreen_shouldNotDisplayPreference() { + mUserManager.setIsAdminUser(true); + mUserManager.setUserSwitcherEnabled(true); + mUserManager.setSupportsMultipleUsers(true); + when(mLockPatternUtils.isSecure(anyInt())).thenReturn(false); + final AddUserWhenLockedPreferenceController controller = + new AddUserWhenLockedPreferenceController( + mContext, "fake_key", mLockPatternUtils); + final RestrictedSwitchPreference preference = mock(RestrictedSwitchPreference.class); + + controller.updateState(preference); + + verify(preference).setVisible(false); + assertThat(controller.getAvailabilityStatus()).isEqualTo( + BasePreferenceController.CONDITIONALLY_UNAVAILABLE); + } + + @Test + public void updateState_secureLockScreen_shouldDisplayPreference() { + mUserManager.setIsAdminUser(true); + mUserManager.setUserSwitcherEnabled(true); + mUserManager.setSupportsMultipleUsers(true); + when(mLockPatternUtils.isSecure(anyInt())).thenReturn(true); + final AddUserWhenLockedPreferenceController controller = + new AddUserWhenLockedPreferenceController( + mContext, "fake_key", mLockPatternUtils); + final RestrictedSwitchPreference preference = mock(RestrictedSwitchPreference.class); + + controller.updateState(preference); + + verify(preference).setVisible(true); + assertThat(controller.getAvailabilityStatus()).isEqualTo( + BasePreferenceController.AVAILABLE); + } }