| /* |
| * Copyright (C) 2017 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 android.content.pm.crossprofile; |
| |
| import static org.mockito.ArgumentMatchers.anyInt; |
| import static org.mockito.ArgumentMatchers.nullable; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| |
| import android.content.Context; |
| import android.content.res.Resources; |
| import android.graphics.drawable.Drawable; |
| import android.os.UserHandle; |
| import android.os.UserManager; |
| import android.platform.test.annotations.Presubmit; |
| |
| import com.android.internal.R; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.Answers; |
| import org.mockito.Mock; |
| import org.mockito.junit.MockitoJUnitRunner; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| /** |
| * Build/Install/Run: |
| * bit FrameworksCoreTests:android.content.pm.crossprofile.CrossProfileAppsTest |
| */ |
| @Presubmit |
| @RunWith(MockitoJUnitRunner.class) |
| public class CrossProfileAppsTest { |
| private static final UserHandle PERSONAL_PROFILE = UserHandle.of(0); |
| private static final UserHandle MANAGED_PROFILE = UserHandle.of(10); |
| private static final String MY_PACKAGE = "my.package"; |
| |
| private List<UserHandle> mTargetProfiles; |
| |
| @Mock |
| private Context mContext; |
| @Mock |
| private UserManager mUserManager; |
| @Mock |
| private ICrossProfileApps mService; |
| @Mock |
| private Resources mResources; |
| @Mock(answer = Answers.RETURNS_DEEP_STUBS) |
| private Drawable mDrawable; |
| private CrossProfileApps mCrossProfileApps; |
| |
| @Before |
| public void initCrossProfileApps() { |
| mCrossProfileApps = new CrossProfileApps(mContext, mService); |
| } |
| |
| @Before |
| public void mockContext() { |
| when(mContext.getPackageName()).thenReturn(MY_PACKAGE); |
| when(mContext.getSystemServiceName(UserManager.class)).thenReturn(Context.USER_SERVICE); |
| when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mUserManager); |
| } |
| |
| @Before |
| public void mockResources() { |
| when(mContext.getResources()).thenReturn(mResources); |
| when(mResources.getDrawable(anyInt(), nullable(Resources.Theme.class))) |
| .thenReturn(mDrawable); |
| } |
| |
| @Before |
| public void initUsers() throws Exception { |
| when(mUserManager.isManagedProfile(PERSONAL_PROFILE.getIdentifier())).thenReturn(false); |
| when(mUserManager.isManagedProfile(MANAGED_PROFILE.getIdentifier())).thenReturn(true); |
| |
| mTargetProfiles = new ArrayList<>(); |
| when(mService.getTargetUserProfiles(MY_PACKAGE)).thenReturn(mTargetProfiles); |
| } |
| |
| @Test |
| public void getProfileSwitchingLabel_managedProfile() { |
| setValidTargetProfile(MANAGED_PROFILE); |
| |
| mCrossProfileApps.getProfileSwitchingLabel(MANAGED_PROFILE); |
| verify(mResources).getString(R.string.managed_profile_label); |
| } |
| |
| @Test |
| public void getProfileSwitchingLabel_personalProfile() { |
| setValidTargetProfile(PERSONAL_PROFILE); |
| |
| mCrossProfileApps.getProfileSwitchingLabel(PERSONAL_PROFILE); |
| verify(mResources).getString(R.string.user_owner_label); |
| } |
| |
| @Test(expected = SecurityException.class) |
| public void getProfileSwitchingLabel_securityException() { |
| mCrossProfileApps.getProfileSwitchingLabel(PERSONAL_PROFILE); |
| } |
| |
| @Test |
| public void getProfileSwitchingIcon_managedProfile() { |
| setValidTargetProfile(MANAGED_PROFILE); |
| |
| mCrossProfileApps.getProfileSwitchingIcon(MANAGED_PROFILE); |
| verify(mResources).getDrawable(R.drawable.ic_corp_badge, null); |
| } |
| |
| @Test |
| public void getProfileSwitchingIcon_personalProfile() { |
| setValidTargetProfile(PERSONAL_PROFILE); |
| |
| mCrossProfileApps.getProfileSwitchingIcon(PERSONAL_PROFILE); |
| verify(mResources).getDrawable(R.drawable.ic_account_circle, null); |
| } |
| |
| @Test(expected = SecurityException.class) |
| public void getProfileSwitchingIcon_securityException() { |
| mCrossProfileApps.getProfileSwitchingIcon(PERSONAL_PROFILE); |
| } |
| |
| private void setValidTargetProfile(UserHandle userHandle) { |
| mTargetProfiles.add(userHandle); |
| } |
| } |