1 /* 2 * Copyright (C) 2018 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.car.settings.users; 18 19 import static org.mockito.Mockito.verify; 20 import static org.mockito.Mockito.when; 21 22 import android.car.user.CarUserManagerHelper; 23 import android.content.pm.UserInfo; 24 25 import com.android.car.settings.CarSettingsRobolectricTestRunner; 26 27 import org.junit.Before; 28 import org.junit.Test; 29 import org.junit.runner.RunWith; 30 import org.mockito.Mock; 31 import org.mockito.MockitoAnnotations; 32 import org.robolectric.Robolectric; 33 34 @RunWith(CarSettingsRobolectricTestRunner.class) 35 public class AddNewUserTaskTest { 36 @Mock 37 private CarUserManagerHelper mCarUserManagerHelper; 38 @Mock 39 private AddNewUserTask.AddNewUserListener mAddNewUserListener; 40 41 private AddNewUserTask mTask; 42 43 @Before 44 public void createAsyncTask() { 45 MockitoAnnotations.initMocks(this); 46 mTask = new AddNewUserTask(mCarUserManagerHelper, mAddNewUserListener); 47 } 48 49 @Test 50 public void testTaskCallsCreateNewNonAdminUser() { 51 mTask.execute("Test name"); 52 Robolectric.flushBackgroundThreadScheduler(); 53 54 verify(mCarUserManagerHelper).createNewNonAdminUser("Test name"); 55 } 56 57 @Test 58 public void testSwitchToNewUserIfUserCreated() { 59 UserInfo newUser = new UserInfo(10, "Test name", /* flags= */ 0); 60 when(mCarUserManagerHelper.createNewNonAdminUser("Test name")).thenReturn(newUser); 61 62 mTask.execute("Test name"); 63 Robolectric.flushBackgroundThreadScheduler(); 64 65 verify(mCarUserManagerHelper).switchToUser(newUser); 66 } 67 68 @Test 69 public void testAddNewUserListenerCalled() { 70 mTask.execute("Test name"); 71 Robolectric.flushBackgroundThreadScheduler(); 72 73 verify(mAddNewUserListener).onUserAdded(); 74 } 75 } 76