common/device-side/bedstead/nene/src/test/java/com/android/bedstead/nene/users/UserReferenceTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2021 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.bedstead.nene.users;

 import static android.Manifest.permission.CREATE_USERS;
 import static android.os.Build.VERSION_CODES.Q;
 import static android.os.Build.VERSION_CODES.S;

 import static com.android.bedstead.harrier.OptionalBoolean.FALSE;

 import static com.google.common.truth.Truth.assertThat;

 import static org.testng.Assert.assertThrows;

 import android.content.Context;
 import android.os.Process;
 import android.os.UserManager;

 import com.android.bedstead.harrier.BedsteadJUnit4;
 import com.android.bedstead.harrier.DeviceState;
 import com.android.bedstead.harrier.annotations.EnsureHasPermission;
 import com.android.bedstead.harrier.annotations.EnsureHasSecondaryUser;
 import com.android.bedstead.harrier.annotations.EnsureHasWorkProfile;
 import com.android.bedstead.harrier.annotations.RequireRunNotOnSecondaryUser;
 import com.android.bedstead.harrier.annotations.RequireRunOnPrimaryUser;
 import com.android.bedstead.harrier.annotations.RequireRunOnWorkProfile;
 import com.android.bedstead.harrier.annotations.RequireSdkVersion;
 import com.android.bedstead.nene.TestApis;
 import com.android.bedstead.nene.exceptions.NeneException;

 import org.junit.ClassRule;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;

 @RunWith(BedsteadJUnit4.class)
 public class UserReferenceTest {
     private static final int NON_EXISTING_USER_ID = 10000;
     private static final int USER_ID = NON_EXISTING_USER_ID;
     private static final String USER_NAME = "userName";
     private static final String TEST_ACTIVITY_NAME = "com.android.bedstead.nene.test.Activity";
     private static final int SERIAL_NO = 1000;
     private static final UserType USER_TYPE = new UserType(new UserType.MutableUserType());
     private static final Context sContext = TestApis.context().instrumentedContext();
     private static final UserManager sUserManager = sContext.getSystemService(UserManager.class);

     @ClassRule @Rule
     public static final DeviceState sDeviceState = new DeviceState();

     @Test
     public void id_returnsId() {
         assertThat(TestApis.users().find(USER_ID).id()).isEqualTo(USER_ID);
     }

     @Test
     public void userHandle_referencesId() {
         assertThat(TestApis.users().find(USER_ID).userHandle().getIdentifier()).isEqualTo(USER_ID);
     }

     @Test
     public void exists_doesNotExist_returnsFalse() {
         assertThat(TestApis.users().find(NON_EXISTING_USER_ID).exists()).isFalse();
     }

     @Test
     @EnsureHasSecondaryUser
     public void exists_doesExist_returnsTrue() {
         assertThat(sDeviceState.secondaryUser().exists()).isTrue();
     }

     @Test
     public void remove_userDoesNotExist_throwsException() {
         assertThrows(NeneException.class, () -> TestApis.users().find(USER_ID).remove());
     }

     @Test
     public void remove_userExists_removesUser() {
         UserReference user = TestApis.users().createUser().create();

         user.remove();

         assertThat(TestApis.users().all()).doesNotContain(user);
     }

     @Test
     public void start_userDoesNotExist_throwsException() {
         assertThrows(NeneException.class,
                 () -> TestApis.users().find(NON_EXISTING_USER_ID).start());
     }

     @Test
     public void start_userNotStarted_userIsUnlocked() {
         UserReference user = TestApis.users().createUser().create().stop();

         user.start();

         try {
             assertThat(user.isUnlocked()).isTrue();
         } finally {
             user.remove();
         }
     }

     @Test
     @EnsureHasSecondaryUser
     public void start_userAlreadyStarted_doesNothing() {
         sDeviceState.secondaryUser().start();

         sDeviceState.secondaryUser().start();

         assertThat(sDeviceState.secondaryUser().isUnlocked()).isTrue();
     }

     @Test
     public void stop_userDoesNotExist_doesNothing() {
         TestApis.users().find(NON_EXISTING_USER_ID).stop();
     }

     @Test
     @EnsureHasSecondaryUser
     @RequireRunNotOnSecondaryUser
     public void stop_userStarted_userIsStopped() {
         sDeviceState.secondaryUser().stop();

         assertThat(sDeviceState.secondaryUser().isRunning()).isFalse();
     }

     @Test
     @EnsureHasSecondaryUser
     @RequireRunNotOnSecondaryUser
     public void stop_userNotStarted_doesNothing() {
         sDeviceState.secondaryUser().stop();

         sDeviceState.secondaryUser().stop();

         assertThat(sDeviceState.secondaryUser().isRunning()).isFalse();
     }

     @Test
     @EnsureHasSecondaryUser
     @RequireRunNotOnSecondaryUser
     @RequireSdkVersion(min = Q)
     public void switchTo_userIsSwitched() {
         sDeviceState.secondaryUser().switchTo();

         assertThat(TestApis.users().current()).isEqualTo(sDeviceState.secondaryUser());
     }

     @Test
     @RequireRunOnWorkProfile(switchedToParentUser = FALSE)
     public void switchTo_profile_switchesToParent() {
         sDeviceState.workProfile().switchTo();

         assertThat(TestApis.users().current()).isEqualTo(sDeviceState.workProfile().parent());
     }

     @Test
     @RequireRunOnPrimaryUser
     @EnsureHasWorkProfile
     public void stop_isWorkProfileOfCurrentUser_stops() {
         sDeviceState.workProfile().stop();

         assertThat(sDeviceState.workProfile().isRunning()).isFalse();
     }

     @Test
     public void serialNo_returnsSerialNo() {
         UserReference user = TestApis.users().instrumented();

         assertThat(user.serialNo())
                 .isEqualTo(sUserManager.getSerialNumberForUser(Process.myUserHandle()));
     }

     @Test
     public void serialNo_userDoesNotExist_throwsException() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThrows(NeneException.class, user::serialNo);
     }

     @Test
     @EnsureHasPermission(CREATE_USERS)
     @RequireSdkVersion(min = S, reason = "getUserName is only available on S+")
     public void name_returnsName() {
         UserReference user = TestApis.users().instrumented();

         assertThat(user.name()).isEqualTo(sUserManager.getUserName());
     }

     @Test
     public void name_userDoesNotExist_throwsException() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThrows(NeneException.class, user::name);
     }

     @Test
     @EnsureHasPermission(CREATE_USERS)
     @RequireSdkVersion(min = S, reason = "getUserType is only available on S+")
     public void type_returnsType() {
         UserReference user = TestApis.users().instrumented();

         assertThat(user.type().name()).isEqualTo(sUserManager.getUserType());
     }

     @Test
     public void type_userDoesNotExist_throwsException() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThrows(NeneException.class, user::type);
     }

     @Test
     @RequireRunOnPrimaryUser
     public void isPrimary_isPrimary_returnsTrue() {
         UserReference user = TestApis.users().instrumented();

         assertThat(user.isPrimary()).isTrue();
     }

     @Test
     @RequireRunOnPrimaryUser
     @EnsureHasSecondaryUser
     public void isPrimary_isNotPrimary_returnsFalse() {
         UserReference user = sDeviceState.secondaryUser();

         assertThat(user.isPrimary()).isFalse();
     }

     @Test
     public void isPrimary_userDoesNotExist_throwsException() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThrows(NeneException.class, user::isPrimary);
     }

     @Test
     public void isRunning_userNotStarted_returnsFalse() {
         UserReference user = TestApis.users().createUser().create();
         user.stop();

         try {
             assertThat(user.isRunning()).isFalse();
         } finally {
             user.remove();
         }
     }

     @Test
     public void isRunning_userIsRunning_returnsTrue() {
         UserReference user = TestApis.users().createUser().create();
         user.start();

         try {
             assertThat(user.isRunning()).isTrue();
         } finally {
             user.remove();
         }
     }

     @Test
     public void isRunning_userDoesNotExist_returnsFalse() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThat(user.isRunning()).isFalse();
     }

     @Test
     public void isUnlocked_userIsUnlocked_returnsTrue() {
         UserReference user = TestApis.users().createUser().createAndStart();

         try {
             assertThat(user.isUnlocked()).isTrue();
         } finally {
             user.remove();
         }
     }

     // TODO(b/203542772): add tests for locked state

     @Test
     public void isUnlocked_userDoesNotExist_returnsFalse() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThat(user.isUnlocked()).isFalse();
     }

     @Test
     @EnsureHasWorkProfile
     public void parent_returnsParent() {
         assertThat(sDeviceState.workProfile().parent()).isNotNull();
     }

     @Test
     @RequireRunOnPrimaryUser
     public void parent_noParent_returnsNull() {
         UserReference user = TestApis.users().instrumented();

         assertThat(user.parent()).isNull();
     }

     @Test
     @RequireRunOnPrimaryUser
     public void parent_userDoesNotExist_throwsException() {
         UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

         assertThrows(NeneException.class, user::parent);
     }

     @Test
     public void autoclose_removesUser() {
         int numUsers = TestApis.users().all().size();

         try (UserReference user = TestApis.users().createUser().create()) {
             // We intentionally don't do anything here, just rely on the auto-close behaviour
         }

         assertThat(TestApis.users().all()).hasSize(numUsers);
     }
 }
	/*
	* Copyright (C) 2021 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.bedstead.nene.users;

	import static android.Manifest.permission.CREATE_USERS;
	import static android.os.Build.VERSION_CODES.Q;
	import static android.os.Build.VERSION_CODES.S;

	import static com.android.bedstead.harrier.OptionalBoolean.FALSE;

	import static com.google.common.truth.Truth.assertThat;

	import static org.testng.Assert.assertThrows;

	import android.content.Context;
	import android.os.Process;
	import android.os.UserManager;

	import com.android.bedstead.harrier.BedsteadJUnit4;
	import com.android.bedstead.harrier.DeviceState;
	import com.android.bedstead.harrier.annotations.EnsureHasPermission;
	import com.android.bedstead.harrier.annotations.EnsureHasSecondaryUser;
	import com.android.bedstead.harrier.annotations.EnsureHasWorkProfile;
	import com.android.bedstead.harrier.annotations.RequireRunNotOnSecondaryUser;
	import com.android.bedstead.harrier.annotations.RequireRunOnPrimaryUser;
	import com.android.bedstead.harrier.annotations.RequireRunOnWorkProfile;
	import com.android.bedstead.harrier.annotations.RequireSdkVersion;
	import com.android.bedstead.nene.TestApis;
	import com.android.bedstead.nene.exceptions.NeneException;

	import org.junit.ClassRule;
	import org.junit.Rule;
	import org.junit.Test;
	import org.junit.runner.RunWith;

	@RunWith(BedsteadJUnit4.class)
	public class UserReferenceTest {
	private static final int NON_EXISTING_USER_ID = 10000;
	private static final int USER_ID = NON_EXISTING_USER_ID;
	private static final String USER_NAME = "userName";
	private static final String TEST_ACTIVITY_NAME = "com.android.bedstead.nene.test.Activity";
	private static final int SERIAL_NO = 1000;
	private static final UserType USER_TYPE = new UserType(new UserType.MutableUserType());
	private static final Context sContext = TestApis.context().instrumentedContext();
	private static final UserManager sUserManager = sContext.getSystemService(UserManager.class);

	@ClassRule @Rule
	public static final DeviceState sDeviceState = new DeviceState();

	@Test
	public void id_returnsId() {
	assertThat(TestApis.users().find(USER_ID).id()).isEqualTo(USER_ID);
	}

	@Test
	public void userHandle_referencesId() {
	assertThat(TestApis.users().find(USER_ID).userHandle().getIdentifier()).isEqualTo(USER_ID);
	}

	@Test
	public void exists_doesNotExist_returnsFalse() {
	assertThat(TestApis.users().find(NON_EXISTING_USER_ID).exists()).isFalse();
	}

	@Test
	@EnsureHasSecondaryUser
	public void exists_doesExist_returnsTrue() {
	assertThat(sDeviceState.secondaryUser().exists()).isTrue();
	}

	@Test
	public void remove_userDoesNotExist_throwsException() {
	assertThrows(NeneException.class, () -> TestApis.users().find(USER_ID).remove());
	}

	@Test
	public void remove_userExists_removesUser() {
	UserReference user = TestApis.users().createUser().create();

	user.remove();

	assertThat(TestApis.users().all()).doesNotContain(user);
	}

	@Test
	public void start_userDoesNotExist_throwsException() {
	assertThrows(NeneException.class,
	() -> TestApis.users().find(NON_EXISTING_USER_ID).start());
	}

	@Test
	public void start_userNotStarted_userIsUnlocked() {
	UserReference user = TestApis.users().createUser().create().stop();

	user.start();

	try {
	assertThat(user.isUnlocked()).isTrue();
	} finally {
	user.remove();
	}
	}

	@Test
	@EnsureHasSecondaryUser
	public void start_userAlreadyStarted_doesNothing() {
	sDeviceState.secondaryUser().start();

	sDeviceState.secondaryUser().start();

	assertThat(sDeviceState.secondaryUser().isUnlocked()).isTrue();
	}

	@Test
	public void stop_userDoesNotExist_doesNothing() {
	TestApis.users().find(NON_EXISTING_USER_ID).stop();
	}

	@Test
	@EnsureHasSecondaryUser
	@RequireRunNotOnSecondaryUser
	public void stop_userStarted_userIsStopped() {
	sDeviceState.secondaryUser().stop();

	assertThat(sDeviceState.secondaryUser().isRunning()).isFalse();
	}

	@Test
	@EnsureHasSecondaryUser
	@RequireRunNotOnSecondaryUser
	public void stop_userNotStarted_doesNothing() {
	sDeviceState.secondaryUser().stop();

	sDeviceState.secondaryUser().stop();

	assertThat(sDeviceState.secondaryUser().isRunning()).isFalse();
	}

	@Test
	@EnsureHasSecondaryUser
	@RequireRunNotOnSecondaryUser
	@RequireSdkVersion(min = Q)
	public void switchTo_userIsSwitched() {
	sDeviceState.secondaryUser().switchTo();

	assertThat(TestApis.users().current()).isEqualTo(sDeviceState.secondaryUser());
	}

	@Test
	@RequireRunOnWorkProfile(switchedToParentUser = FALSE)
	public void switchTo_profile_switchesToParent() {
	sDeviceState.workProfile().switchTo();

	assertThat(TestApis.users().current()).isEqualTo(sDeviceState.workProfile().parent());
	}

	@Test
	@RequireRunOnPrimaryUser
	@EnsureHasWorkProfile
	public void stop_isWorkProfileOfCurrentUser_stops() {
	sDeviceState.workProfile().stop();

	assertThat(sDeviceState.workProfile().isRunning()).isFalse();
	}

	@Test
	public void serialNo_returnsSerialNo() {
	UserReference user = TestApis.users().instrumented();

	assertThat(user.serialNo())
	.isEqualTo(sUserManager.getSerialNumberForUser(Process.myUserHandle()));
	}

	@Test
	public void serialNo_userDoesNotExist_throwsException() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThrows(NeneException.class, user::serialNo);
	}

	@Test
	@EnsureHasPermission(CREATE_USERS)
	@RequireSdkVersion(min = S, reason = "getUserName is only available on S+")
	public void name_returnsName() {
	UserReference user = TestApis.users().instrumented();

	assertThat(user.name()).isEqualTo(sUserManager.getUserName());
	}

	@Test
	public void name_userDoesNotExist_throwsException() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThrows(NeneException.class, user::name);
	}

	@Test
	@EnsureHasPermission(CREATE_USERS)
	@RequireSdkVersion(min = S, reason = "getUserType is only available on S+")
	public void type_returnsType() {
	UserReference user = TestApis.users().instrumented();

	assertThat(user.type().name()).isEqualTo(sUserManager.getUserType());
	}

	@Test
	public void type_userDoesNotExist_throwsException() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThrows(NeneException.class, user::type);
	}

	@Test
	@RequireRunOnPrimaryUser
	public void isPrimary_isPrimary_returnsTrue() {
	UserReference user = TestApis.users().instrumented();

	assertThat(user.isPrimary()).isTrue();
	}

	@Test
	@RequireRunOnPrimaryUser
	@EnsureHasSecondaryUser
	public void isPrimary_isNotPrimary_returnsFalse() {
	UserReference user = sDeviceState.secondaryUser();

	assertThat(user.isPrimary()).isFalse();
	}

	@Test
	public void isPrimary_userDoesNotExist_throwsException() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThrows(NeneException.class, user::isPrimary);
	}

	@Test
	public void isRunning_userNotStarted_returnsFalse() {
	UserReference user = TestApis.users().createUser().create();
	user.stop();

	try {
	assertThat(user.isRunning()).isFalse();
	} finally {
	user.remove();
	}
	}

	@Test
	public void isRunning_userIsRunning_returnsTrue() {
	UserReference user = TestApis.users().createUser().create();
	user.start();

	try {
	assertThat(user.isRunning()).isTrue();
	} finally {
	user.remove();
	}
	}

	@Test
	public void isRunning_userDoesNotExist_returnsFalse() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThat(user.isRunning()).isFalse();
	}

	@Test
	public void isUnlocked_userIsUnlocked_returnsTrue() {
	UserReference user = TestApis.users().createUser().createAndStart();

	try {
	assertThat(user.isUnlocked()).isTrue();
	} finally {
	user.remove();
	}
	}

	// TODO(b/203542772): add tests for locked state

	@Test
	public void isUnlocked_userDoesNotExist_returnsFalse() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThat(user.isUnlocked()).isFalse();
	}

	@Test
	@EnsureHasWorkProfile
	public void parent_returnsParent() {
	assertThat(sDeviceState.workProfile().parent()).isNotNull();
	}

	@Test
	@RequireRunOnPrimaryUser
	public void parent_noParent_returnsNull() {
	UserReference user = TestApis.users().instrumented();

	assertThat(user.parent()).isNull();
	}

	@Test
	@RequireRunOnPrimaryUser
	public void parent_userDoesNotExist_throwsException() {
	UserReference user = TestApis.users().find(NON_EXISTING_USER_ID);

	assertThrows(NeneException.class, user::parent);
	}

	@Test
	public void autoclose_removesUser() {
	int numUsers = TestApis.users().all().size();

	try (UserReference user = TestApis.users().createUser().create()) {
	// We intentionally don't do anything here, just rely on the auto-close behaviour
	}

	assertThat(TestApis.users().all()).hasSize(numUsers);
	}
	}