Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 970d9167 authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Android (Google) Code Review
Browse files

Merge "Show "Add users from lock screen" dependent on secure lock screen."

parents 2db8d1d8 0149363a
Loading
Loading
Loading
Loading
+8 −1
Original line number Diff line number Diff line
@@ -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;
        }
+3 −2
Original line number Diff line number Diff line
@@ -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<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);
+47 −2
Original line number Diff line number Diff line
@@ -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);
    }
}