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

Commit 91cbc689 authored by Jigar Thakkar's avatar Jigar Thakkar
Browse files

Add tests for private profile delayed locking 

This change adds UserControllerTests for testing the delayed locking
flows for private profile. The following scenarios are covered:
- Private profile should be stopped and locked if stopUser is called
- Private profile should be stopped and locked if
  stopUserWithDelayedLocking is called but the private profile feature
  flags are disabled
- Private profile should be stopped and locked if
  stopUserWithDelayedLocking is called but mMaxRunningUsers is less than
  the total running users
- Private profile should be stopped and unlocked if
  stopUserWithDelayedLocking is called with private profile flags turned
  on and mMaxRunningUsers greater than the total running users

Test: atest UserControllerTests
Bug: 312184187

Change-Id: I2af727edf52340c9f4b64614715f70226b88fd75
parent 90e467e1

services/tests/servicestests/src/com/android/server/am/UserControllerTest.java

+91 −8
Original line number Diff line number Diff line
@@ -92,6 +92,7 @@ import android.os.UserHandle; 
import android.os.UserManager;

import android.os.storage.IStorageManager;

import android.platform.test.annotations.Presubmit;

import android.platform.test.flag.junit.SetFlagsRule;

import android.util.Log;

import android.view.Display;
@@ -104,11 +105,14 @@ import com.android.server.am.UserState.KeyEvictedCallback; 
import com.android.server.pm.UserJourneyLogger;

import com.android.server.pm.UserManagerInternal;

import com.android.server.pm.UserManagerService;

import com.android.server.pm.UserTypeDetails;

import com.android.server.pm.UserTypeFactory;

import com.android.server.wm.ActivityTaskManagerInternal;

import com.android.server.wm.WindowManagerService;



import org.junit.After;

import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import org.mockito.ArgumentCaptor;
@@ -175,6 +179,9 @@ public class UserControllerTest { 
            USER_START_MSG,

            REPORT_LOCKED_BOOT_COMPLETE_MSG);



    @Rule

    public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();



    @Before

    public void setUp() throws Exception {

        runWithDexmakerShareClassLoader(() -> {
@@ -789,28 +796,99 @@ public class UserControllerTest { 
    }



    @Test

    public void testStartProfile() throws Exception {

        setUpAndStartProfileInBackground(TEST_USER_ID1);

    public void testStartManagedProfile() throws Exception {

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_MANAGED);



        startBackgroundUserAssertions();

        verifyUserAssignedToDisplay(TEST_USER_ID1, Display.DEFAULT_DISPLAY);

    }



    @Test

    public void testStartProfile_whenUsersOnSecondaryDisplaysIsEnabled() throws Exception {

    public void testStartManagedProfile_whenUsersOnSecondaryDisplaysIsEnabled() throws Exception {

        mockIsUsersOnSecondaryDisplaysEnabled(true);



        setUpAndStartProfileInBackground(TEST_USER_ID1);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_MANAGED);



        startBackgroundUserAssertions();

        verifyUserAssignedToDisplay(TEST_USER_ID1, Display.DEFAULT_DISPLAY);

    }



    @Test

    public void testStopProfile() throws Exception {

        setUpAndStartProfileInBackground(TEST_USER_ID1);

    public void testStopManagedProfile() throws Exception {

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_MANAGED);

        assertProfileLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* expectLocking= */ true);

        verifyUserUnassignedFromDisplay(TEST_USER_ID1);

    }



    @Test

    public void testStopPrivateProfile() throws Exception {

        mUserController.setInitialConfig(/* mUserSwitchUiEnabled */ true,

                /* maxRunningUsers= */ 3, /* delayUserDataLocking= */ false);

        mSetFlagsRule.enableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE,

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_PRIVATE);

        assertProfileLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* expectLocking= */ true);

        verifyUserUnassignedFromDisplay(TEST_USER_ID1);



        mSetFlagsRule.disableFlags(

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID2, UserManager.USER_TYPE_PROFILE_PRIVATE);

        assertProfileLockedOrUnlockedAfterStopping(TEST_USER_ID2, /* expectLocking= */ true);

        verifyUserUnassignedFromDisplay(TEST_USER_ID2);

    }



    @Test

    public void testStopPrivateProfileWithDelayedLocking() throws Exception {

        mUserController.setInitialConfig(/* mUserSwitchUiEnabled */ true,

                /* maxRunningUsers= */ 3, /* delayUserDataLocking= */ false);

        mSetFlagsRule.enableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE,

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_PRIVATE);

        assertUserLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* delayedLocking= */ true,

                /* keyEvictedCallback */ null, /* expectLocking= */ false);

    }



    @Test

    public void testStopPrivateProfileWithDelayedLocking_flagDisabled() throws Exception {

        mUserController.setInitialConfig(/* mUserSwitchUiEnabled */ true,

                /* maxRunningUsers= */ 3, /* delayUserDataLocking= */ false);

        mSetFlagsRule.enableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE);

        mSetFlagsRule.disableFlags(

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_PRIVATE);

        assertUserLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* delayedLocking= */ true,

                /* keyEvictedCallback */ null, /* expectLocking= */ true);



        mSetFlagsRule.disableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE);

        mSetFlagsRule.enableFlags(

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID2, UserManager.USER_TYPE_PROFILE_PRIVATE);

        assertUserLockedOrUnlockedAfterStopping(TEST_USER_ID2, /* delayedLocking= */ true,

                /* keyEvictedCallback */ null, /* expectLocking= */ true);

    }



    @Test

    public void testStopPrivateProfileWithDelayedLocking_maxRunningUsersBreached()

            throws Exception {

        mUserController.setInitialConfig(/* mUserSwitchUiEnabled */ true,

                /* maxRunningUsers= */ 1, /* delayUserDataLocking= */ false);

        mSetFlagsRule.enableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE,

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_PRIVATE);

        setUpAndStartProfileInBackground(TEST_USER_ID2, UserManager.USER_TYPE_PROFILE_MANAGED);

        assertUserLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* delayedLocking= */ true,

                /* keyEvictedCallback */ null, /* expectLocking= */ true);

    }



    @Test

    public void testStopManagedProfileWithDelayedLocking() throws Exception {

        mUserController.setInitialConfig(/* mUserSwitchUiEnabled */ true,

                /* maxRunningUsers= */ 3, /* delayUserDataLocking= */ false);

        mSetFlagsRule.enableFlags(android.os.Flags.FLAG_ALLOW_PRIVATE_PROFILE,

                android.multiuser.Flags.FLAG_ENABLE_BIOMETRICS_TO_UNLOCK_PRIVATE_SPACE);

        setUpAndStartProfileInBackground(TEST_USER_ID1, UserManager.USER_TYPE_PROFILE_MANAGED);

        assertUserLockedOrUnlockedAfterStopping(TEST_USER_ID1, /* delayedLocking= */ true,

                /* keyEvictedCallback */ null, /* expectLocking= */ true);

    }



    /** Tests handleIncomingUser() for a variety of permissions and situations. */
@@ -1001,8 +1079,8 @@ public class UserControllerTest { 
        mUserStates.put(userId, mUserController.getStartedUserState(userId));

    }



    private void setUpAndStartProfileInBackground(int userId) throws Exception {

        setUpUser(userId, UserInfo.FLAG_PROFILE, false, UserManager.USER_TYPE_PROFILE_MANAGED);

    private void setUpAndStartProfileInBackground(int userId, String userType) throws Exception {

        setUpUser(userId, UserInfo.FLAG_PROFILE, false, userType);

        assertThat(mUserController.startProfile(userId, /* evenWhenDisabled=*/ false,

                /* unlockListener= */ null)).isTrue();
@@ -1070,6 +1148,11 @@ public class UserControllerTest { 
        userInfo.preCreated = preCreated;

        when(mInjector.mUserManagerMock.getUserInfo(eq(userId))).thenReturn(userInfo);

        when(mInjector.mUserManagerMock.isPreCreated(userId)).thenReturn(preCreated);



        UserTypeDetails userTypeDetails = UserTypeFactory.getUserTypes().get(userType);

        assertThat(userTypeDetails).isNotNull();

        when(mInjector.mUserManagerInternalMock.getUserProperties(eq(userId)))

                .thenReturn(userTypeDetails.getDefaultUserPropertiesReference());

    }



    private static List<String> getActions(List<Intent> intents) {