108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway/* 208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * Copyright (C) 2015 The Android Open Source Project 308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * 408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * Licensed under the Apache License, Version 2.0 (the "License"); 508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * you may not use this file except in compliance with the License. 608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * You may obtain a copy of the License at 708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * 808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * http://www.apache.org/licenses/LICENSE-2.0 908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * 1008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * Unless required by applicable law or agreed to in writing, software 1108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * distributed under the License is distributed on an "AS IS" BASIS, 1208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * See the License for the specific language governing permissions and 1408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * limitations under the License. 1508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway */ 1608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwaypackage com.android.managedprovisioning.task; 1808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 199df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.any; 209df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.anyBoolean; 219df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.anyString; 229df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.never; 239df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.verify; 249df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franzimport static org.mockito.Mockito.when; 259df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz 2608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.content.Context; 2708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.content.pm.UserInfo; 2808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.os.UserHandle; 2908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.os.UserManager; 3008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.test.AndroidTestCase; 3108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport android.test.suitebuilder.annotation.SmallTest; 3208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 3308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport org.mockito.Mock; 3408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport org.mockito.MockitoAnnotations; 3508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 3608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport java.util.Arrays; 3708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwayimport java.util.Collections; 3808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 3908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway/** 4008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway * Unit-tests for {@link DisallowAddUserTask}. 4108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway */ 4208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadwaypublic class DisallowAddUserTaskTest extends AndroidTestCase { 4308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @Mock private Context mockContext; 4408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @Mock private UserManager mockUserManager; 459df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz @Mock private AbstractProvisioningTask.Callback mCallback; 4608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 4708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway // Normal cases. 4808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway private UserInfo primaryUser = new UserInfo(0, "Primary", 4908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway UserInfo.FLAG_PRIMARY | UserInfo.FLAG_ADMIN); 5008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 5108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway // Split-system-user cases. 5208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway private UserInfo systemUser = new UserInfo(UserHandle.USER_SYSTEM, "System", 0 /* flags */); 5308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway private UserInfo meatUser = new UserInfo(10, "Primary", 5408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway UserInfo.FLAG_PRIMARY | UserInfo.FLAG_ADMIN); 5508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 5608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @Override 5708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway public void setUp() { 5808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway // this is necessary for mockito to work 5908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway System.setProperty("dexmaker.dexcache", getContext().getCacheDir().toString()); 6008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 6108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway MockitoAnnotations.initMocks(this); 6208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 639df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz when(mockContext.getSystemService(Context.USER_SERVICE)).thenReturn(mockUserManager); 6408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway // Setup sensible default responses. 6508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.hasUserRestriction(anyString(), any(UserHandle.class))) 6608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway .thenReturn(false); 6708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway } 6808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 6908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @SmallTest 7008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway public void testMaybeDisallowAddUsers_normalSystem() { 719df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // GIVEN that only one user exists on the device and the system doesn't have a split system 729df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // user 7308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.getUsers()).thenReturn(Collections.singletonList(primaryUser)); 749df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz final DisallowAddUserTask task = 759df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz new DisallowAddUserTask(false, mockContext, null, mCallback); 7608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 779df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // WHEN running the DisallowAddUserTask on the single user 789df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz task.run(primaryUser.id); 7908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 809df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // THEN the user restriction should be set 8108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway verify(mockUserManager).setUserRestriction(UserManager.DISALLOW_ADD_USER, true, 8208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway primaryUser.getUserHandle()); 839df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mCallback).onSuccess(task); 8408cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway } 8508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 8608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @SmallTest 8708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway public void testMaybeDisallowAddUsers_normalSystem_restrictionAlreadySetupForOneUser() { 889df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // GIVEN that only one user exists on the device and the system doesn't have a split system 899df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // user 9008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.getUsers()).thenReturn(Collections.singletonList(primaryUser)); 919df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz final DisallowAddUserTask task = 929df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz new DisallowAddUserTask(false, mockContext, null, mCallback); 9308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 949df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // GIVEN that the user restriction has already been set 9508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.hasUserRestriction(UserManager.DISALLOW_ADD_USER, 9608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway primaryUser.getUserHandle())) 9708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway .thenReturn(true); 9808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 999df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // WHEN running the DisallowAddUserTask on the single user 1009df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz task.run(primaryUser.id); 10108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1029df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // THEN the user restriction should not be set 1039df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mockUserManager, never()).setUserRestriction(anyString(), anyBoolean(), 1049df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz any(UserHandle.class)); 1059df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mCallback).onSuccess(task); 10608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway } 10708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 10808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @SmallTest 10908cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway public void testMaybeDisallowAddUsers_splitUserSystem_meatUserDeviceOwner() { 1109df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // GIVEN that we have a split system user and a single meat user on the device 11108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.getUsers()).thenReturn(Arrays.asList(new UserInfo[]{ 11208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway systemUser, meatUser})); 1139df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz final DisallowAddUserTask task = 1149df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz new DisallowAddUserTask(true, mockContext, null, mCallback); 11508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1169df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // WHEN running the DisallowAddUserTask on the meat user 1179df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz task.run(meatUser.id); 11808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1199df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // THEN the user restriction should be added on both users 12008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway verify(mockUserManager).setUserRestriction(UserManager.DISALLOW_ADD_USER, true, 12108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway systemUser.getUserHandle()); 12208cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway verify(mockUserManager).setUserRestriction(UserManager.DISALLOW_ADD_USER, true, 12308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway meatUser.getUserHandle()); 1249df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mCallback).onSuccess(task); 12508cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway } 12608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 12708cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway @SmallTest 12808cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway public void testMaybeDisallowAddUsers_splitUserSystem_systemDeviceOwner() { 1299df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // GIVEN that we have a split system user and only the system user on the device 13008cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway when(mockUserManager.getUsers()).thenReturn(Collections.singletonList(systemUser)); 1319df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz final DisallowAddUserTask task = 1329df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz new DisallowAddUserTask(true, mockContext, null, mCallback); 13308cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1349df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // WHEN running the DisallowAddUserTask on the system user 1359df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz task.run(systemUser.id); 13608cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway 1379df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz // THEN the user restriction should not be set 1389df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mockUserManager, never()).setUserRestriction(anyString(), anyBoolean(), 1399df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz any(UserHandle.class)); 1409df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz verify(mCallback).onSuccess(task); 14108cbb16d62d3ac60ce295354a0bc2cf7d71a7076Alan Treadway } 1429df4bac8ee77da37b1633e9f04ab9464697f6d41Benjamin Franz} 143