1 files changed, 78 insertions, 0 deletions
diff --git a/tests/src/com/android/providers/contacts/util/UserUtilsTest.java b/tests/src/com/android/providers/contacts/util/UserUtilsTest.java
index 93613cf5..c672697a 100644
--- a/tests/src/com/android/providers/contacts/util/UserUtilsTest.java
+++ b/tests/src/com/android/providers/contacts/util/UserUtilsTest.java
@@ -18,6 +18,7 @@ package com.android.providers.contacts.util;
 import static com.android.providers.contacts.ContactsActor.PACKAGE_GREY;
 
 import android.content.Context;
+import android.os.UserHandle;
 import android.provider.ContactsContract;
 import android.test.suitebuilder.annotation.SmallTest;
 
@@ -77,5 +78,82 @@ public class UserUtilsTest extends FixedAndroidTestCase {
 
         um.myUser = MockUserManager.SECONDARY_USER.id;
         assertEquals(-1, UserUtils.getCorpUserId(c));
+
+        // Primary + clone + corp
+        um.setUsers(MockUserManager.PRIMARY_USER, MockUserManager.CLONE_PROFILE_USER,
+                MockUserManager.CORP_USER);
+
+        um.myUser = MockUserManager.PRIMARY_USER.id;
+        assertEquals(MockUserManager.CORP_USER.id, UserUtils.getCorpUserId(c));
+
+        um.myUser = MockUserManager.CLONE_PROFILE_USER.id;
+        assertEquals(-1, UserUtils.getCorpUserId(c));
+
+        um.myUser = MockUserManager.CORP_USER.id;
+        assertEquals(-1, UserUtils.getCorpUserId(c));
+    }
+
+    public void testShouldUseParentsContacts() {
+        final Context c = mActor.getProviderContext();
+        final MockUserManager um = mActor.mockUserManager;
+
+        um.setUsers(MockUserManager.PRIMARY_USER, MockUserManager.SECONDARY_USER,
+                MockUserManager.CLONE_PROFILE_USER, MockUserManager.CORP_USER);
+
+        um.myUser = MockUserManager.PRIMARY_USER.id;
+        assertFalse(UserUtils.shouldUseParentsContacts(c));
+        assertFalse(UserUtils.shouldUseParentsContacts(c,
+                MockUserManager.PRIMARY_USER.getUserHandle()));
+
+        um.myUser = MockUserManager.SECONDARY_USER.id;
+        assertFalse(UserUtils.shouldUseParentsContacts(c));
+        assertFalse(UserUtils.shouldUseParentsContacts(c,
+                MockUserManager.SECONDARY_USER.getUserHandle()));
+
+        um.myUser = MockUserManager.CORP_USER.id;
+        assertFalse(UserUtils.shouldUseParentsContacts(c));
+        assertFalse(UserUtils.shouldUseParentsContacts(c,
+                MockUserManager.CORP_USER.getUserHandle()));
+
+        um.myUser = MockUserManager.CLONE_PROFILE_USER.id;
+        assertTrue(UserUtils.shouldUseParentsContacts(c));
+        assertTrue(UserUtils.shouldUseParentsContacts(c,
+                MockUserManager.CLONE_PROFILE_USER.getUserHandle()));
+
+    }
+
+    public void testIsParentUser() {
+        final Context c = mActor.getProviderContext();
+        final MockUserManager um = mActor.mockUserManager;
+        um.setUsers(MockUserManager.PRIMARY_USER, MockUserManager.SECONDARY_USER,
+                MockUserManager.CLONE_PROFILE_USER, MockUserManager.CORP_USER);
+
+        UserHandle primaryProfileUserHandle = MockUserManager.PRIMARY_USER.getUserHandle();
+        UserHandle cloneUserHandle = MockUserManager.CLONE_PROFILE_USER.getUserHandle();
+        UserHandle corpUserHandle = MockUserManager.CORP_USER.getUserHandle();
+
+        assertTrue(UserUtils.isParentUser(c, primaryProfileUserHandle, cloneUserHandle));
+        assertTrue(UserUtils.isParentUser(c, primaryProfileUserHandle, corpUserHandle));
+        assertFalse(UserUtils.isParentUser(c, primaryProfileUserHandle, primaryProfileUserHandle));
+        assertFalse(UserUtils.isParentUser(c, cloneUserHandle, cloneUserHandle));
+        assertFalse(UserUtils.isParentUser(c, cloneUserHandle, primaryProfileUserHandle));
+        assertFalse(UserUtils.isParentUser(c, corpUserHandle, primaryProfileUserHandle));
+    }
+
+    public void testGetProfileParent() {
+        final Context c = mActor.getProviderContext();
+        final MockUserManager um = mActor.mockUserManager;
+
+        um.setUsers(MockUserManager.PRIMARY_USER, MockUserManager.SECONDARY_USER,
+                MockUserManager.CLONE_PROFILE_USER, MockUserManager.CORP_USER);
+
+        um.myUser = MockUserManager.PRIMARY_USER.id;
+        assertNull(UserUtils.getProfileParentUser(c));
+
+        um.myUser = MockUserManager.CLONE_PROFILE_USER.id;
+        assertEquals(MockUserManager.PRIMARY_USER, UserUtils.getProfileParentUser(c));
+
+        um.myUser = MockUserManager.CORP_USER.id;
+        assertEquals(MockUserManager.PRIMARY_USER, UserUtils.getProfileParentUser(c));
     }
 }