Add a flag to protect headless user 0 multi-user model.

Test: unit test
Bug: 80305146
Change-Id: Ibb029072756e29b57c747152bb779a1037aca753

1 files changed, 10 insertions, 6 deletions

diff --git a/tests/robotests/src/com/android/car/users/CarUserManagerRoboTest.java b/tests/robotests/src/com/android/car/users/CarUserManagerHelperRoboTest.java
index 977ee7c390..9c394459ea 100644
--- a/tests/robotests/src/com/android/car/users/CarUserManagerRoboTest.java
+++ b/tests/robotests/src/com/android/car/users/CarUserManagerHelperRoboTest.java
@@ -43,7 +43,7 @@ import org.robolectric.annotation.Config;
 @RunWith(CarServiceRobolectricTestRunner.class)
 @Config(shadows = { ShadowActivityManager.class,
         ShadowUserHandle.class, ShadowUserManager.class})
-public class CarUserManagerRoboTest {
+public class CarUserManagerHelperRoboTest {
     @Mock
     private Context mContext;
 
@@ -95,7 +95,7 @@ public class CarUserManagerRoboTest {
     }
 
     @Test
-    public void testGetAllUsersExcludesCurrentProcessUser() {
+    public void testGetAllUsers() {
         int currentProcessUserId = 12;
         ShadowUserManager userManager = ShadowUserManager.getShadow();
         userManager.setCurrentUser(currentProcessUserId);
@@ -111,10 +111,14 @@ public class CarUserManagerRoboTest {
         userManager.addUserInfo(otherUser1);
         userManager.addUserInfo(otherUser2);
 
-        // Should return 3 users that don't have currentProcessUser id.
-        assertThat(mHelper.getAllUsers()).hasSize(3);
-        assertThat(mHelper.getAllUsers())
-            .containsExactly(currentProcessUser, otherUser1, otherUser2);
+        if (mHelper.isHeadlessSystemUser()) {
+            // Should return 3 users that don't have system user id.
+            assertThat(mHelper.getAllUsers())
+                .containsExactly(currentProcessUser, otherUser1, otherUser2);
+        } else {
+            assertThat(mHelper.getAllUsers())
+                .containsExactly(systemUser, currentProcessUser, otherUser1, otherUser2);
+        }
     }
 
     @Test