Home | History | Annotate | Download | only in appinfo 
      1 /*
      2  * Copyright (C) 2017 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.settings.applications.appinfo;
     18 
     19 import static com.google.common.truth.Truth.assertThat;
     20 import static org.mockito.ArgumentMatchers.anyInt;
     21 import static org.mockito.ArgumentMatchers.anyString;
     22 import static org.mockito.Mockito.any;
     23 import static org.mockito.Mockito.eq;
     24 import static org.mockito.Mockito.mock;
     25 import static org.mockito.Mockito.never;
     26 import static org.mockito.Mockito.spy;
     27 import static org.mockito.Mockito.verify;
     28 import static org.mockito.Mockito.when;
     29 
     30 import android.content.Context;
     31 import android.content.Intent;
     32 import android.content.pm.ActivityInfo;
     33 import android.content.pm.ApplicationInfo;
     34 import android.content.pm.PackageInfo;
     35 import android.content.pm.PackageManager;
     36 import android.content.pm.ResolveInfo;
     37 import android.os.UserManager;
     38 import android.support.v7.preference.Preference;
     39 
     40 import com.android.settings.core.BasePreferenceController;
     41 import com.android.settings.testutils.SettingsRobolectricTestRunner;
     42 
     43 import org.junit.Before;
     44 import org.junit.Test;
     45 import org.junit.runner.RunWith;
     46 import org.mockito.Mock;
     47 import org.mockito.MockitoAnnotations;
     48 import org.robolectric.RuntimeEnvironment;
     49 
     50 @RunWith(SettingsRobolectricTestRunner.class)
     51 public class AppInstallerInfoPreferenceControllerTest {
     52 
     53     @Mock
     54     private UserManager mUserManager;
     55     @Mock
     56     private PackageManager mPackageManager;
     57     @Mock
     58     private ApplicationInfo mAppInfo;
     59     @Mock
     60     private AppInfoDashboardFragment mFragment;
     61     @Mock
     62     private Preference mPreference;
     63 
     64     private Context mContext;
     65     private AppInstallerInfoPreferenceController mController;
     66 
     67     @Before
     68     public void setUp() throws PackageManager.NameNotFoundException {
     69         MockitoAnnotations.initMocks(this);
     70         mContext = spy(RuntimeEnvironment.application);
     71         when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mUserManager);
     72         when(mContext.getPackageManager()).thenReturn(mPackageManager);
     73         final String installerPackage = "Installer1";
     74         when(mPackageManager.getInstallerPackageName(anyString())).thenReturn(installerPackage);
     75         when(mPackageManager.getApplicationInfo(eq(installerPackage), anyInt()))
     76                 .thenReturn(mAppInfo);
     77         mController = new AppInstallerInfoPreferenceController(mContext, "test_key");
     78         mController.setPackageName("Package1");
     79         mController.setParentFragment(mFragment);
     80     }
     81 
     82     @Test
     83     public void getAvailabilityStatus_managedProfile_shouldReturnDisabled() {
     84         when(mUserManager.isManagedProfile()).thenReturn(true);
     85 
     86         assertThat(mController.getAvailabilityStatus())
     87                 .isEqualTo(BasePreferenceController.DISABLED_FOR_USER);
     88     }
     89 
     90     @Test
     91     public void getAvailabilityStatus_noAppLabel_shouldReturnDisabled() {
     92         when(mUserManager.isManagedProfile()).thenReturn(false);
     93 
     94         assertThat(mController.getAvailabilityStatus())
     95                 .isEqualTo(BasePreferenceController.DISABLED_FOR_USER);
     96     }
     97 
     98     @Test
     99     public void getAvailabilityStatus_hasAppLabel_shouldReturnAvailable() {
    100         when(mUserManager.isManagedProfile()).thenReturn(false);
    101         when(mAppInfo.loadLabel(mPackageManager)).thenReturn("Label1");
    102         mController = new AppInstallerInfoPreferenceController(mContext, "test_key");
    103         mController.setPackageName("Package1");
    104         mController.setParentFragment(mFragment);
    105 
    106         assertThat(mController.getAvailabilityStatus())
    107                 .isEqualTo(BasePreferenceController.AVAILABLE);
    108     }
    109 
    110     @Test
    111     public void updateState_shouldSetSummary() {
    112         final PackageInfo packageInfo = mock(PackageInfo.class);
    113         packageInfo.applicationInfo = mAppInfo;
    114         when(mFragment.getPackageInfo()).thenReturn(packageInfo);
    115 
    116         mController.updateState(mPreference);
    117 
    118         verify(mPreference).setSummary(any());
    119     }
    120 
    121     @Test
    122     public void updateState_noAppStoreLink_shouldDisablePreference() {
    123         final PackageInfo packageInfo = mock(PackageInfo.class);
    124         packageInfo.applicationInfo = mAppInfo;
    125         when(mFragment.getPackageInfo()).thenReturn(packageInfo);
    126         when(mPackageManager.resolveActivity(any(), anyInt())).thenReturn(null);
    127 
    128         mController.updateState(mPreference);
    129 
    130         verify(mPreference).setEnabled(false);
    131     }
    132 
    133     @Test
    134     public void updateState_hasAppStoreLink_shouldSetPreferenceIntent() {
    135         final PackageInfo packageInfo = mock(PackageInfo.class);
    136         packageInfo.applicationInfo = mAppInfo;
    137         when(mFragment.getPackageInfo()).thenReturn(packageInfo);
    138         final ResolveInfo resolveInfo = new ResolveInfo();
    139         resolveInfo.activityInfo = new ActivityInfo();
    140         resolveInfo.activityInfo.packageName = "Pkg1";
    141         resolveInfo.activityInfo.name = "Name1";
    142         when(mPackageManager.resolveActivity(any(), anyInt())).thenReturn(resolveInfo);
    143 
    144         mController.updateState(mPreference);
    145 
    146         verify(mPreference, never()).setEnabled(false);
    147         verify(mPreference).setIntent(any(Intent.class));
    148     }
    149 }
    150