Loading src/com/android/settings/users/AddUserWhenLockedPreferenceController.java +8 −1 Original line number Diff line number Diff line Loading @@ -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 Loading @@ -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; } Loading src/com/android/settings/users/UserSettings.java +3 −2 Original line number Diff line number Diff line Loading @@ -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); Loading Loading @@ -1290,7 +1290,8 @@ public class UserSettings extends SettingsPreferenceFragment boolean suppressAllPage) { final List<String> 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); Loading tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java +47 −2 Original line number Diff line number Diff line Loading @@ -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; Loading @@ -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; Loading @@ -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); } Loading @@ -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); Loading @@ -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); Loading @@ -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); Loading Loading @@ -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); } } Loading
src/com/android/settings/users/AddUserWhenLockedPreferenceController.java +8 −1 Original line number Diff line number Diff line Loading @@ -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 Loading @@ -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; } Loading
src/com/android/settings/users/UserSettings.java +3 −2 Original line number Diff line number Diff line Loading @@ -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); Loading Loading @@ -1290,7 +1290,8 @@ public class UserSettings extends SettingsPreferenceFragment boolean suppressAllPage) { final List<String> 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); Loading
tests/robotests/src/com/android/settings/users/AddUserWhenLockedPreferenceControllerTest.java +47 −2 Original line number Diff line number Diff line Loading @@ -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; Loading @@ -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; Loading @@ -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); } Loading @@ -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); Loading @@ -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); Loading @@ -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); Loading Loading @@ -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); } }