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.Mockito.mock; 21 import static org.mockito.Mockito.spy; 22 import static org.mockito.Mockito.verify; 23 import static org.mockito.Mockito.when; 24 25 import android.content.Context; 26 import android.content.pm.ApplicationInfo; 27 import android.content.pm.IntentFilterVerificationInfo; 28 import android.content.pm.PackageInfo; 29 import android.content.pm.PackageManager; 30 import android.util.ArraySet; 31 32 import com.android.settings.applications.AppDomainsPreference; 33 import com.android.settings.core.BasePreferenceController; 34 import com.android.settings.testutils.SettingsRobolectricTestRunner; 35 import com.android.settingslib.applications.AppUtils; 36 import com.android.settingslib.applications.instantapps.InstantAppDataProvider; 37 38 import org.junit.Before; 39 import org.junit.Test; 40 import org.junit.runner.RunWith; 41 import org.mockito.Mock; 42 import org.mockito.MockitoAnnotations; 43 import org.robolectric.RuntimeEnvironment; 44 import org.robolectric.util.ReflectionHelpers; 45 46 import java.util.ArrayList; 47 import java.util.List; 48 49 @RunWith(SettingsRobolectricTestRunner.class) 50 public class InstantAppDomainsPreferenceControllerTest { 51 52 @Mock 53 private PackageManager mPackageManager; 54 @Mock 55 private ApplicationInfo mAppInfo; 56 @Mock 57 private AppInfoDashboardFragment mFragment; 58 @Mock 59 private AppDomainsPreference mPreference; 60 61 private Context mContext; 62 private InstantAppDomainsPreferenceController mController; 63 64 @Before 65 public void setUp() { 66 MockitoAnnotations.initMocks(this); 67 mContext = spy(RuntimeEnvironment.application); 68 when(mContext.getPackageManager()).thenReturn(mPackageManager); 69 final PackageInfo packageInfo = mock(PackageInfo.class); 70 packageInfo.applicationInfo = mAppInfo; 71 packageInfo.packageName = "Package1"; 72 when(mFragment.getPackageInfo()).thenReturn(packageInfo); 73 mController = new InstantAppDomainsPreferenceController(mContext, "test_key"); 74 mController.setParentFragment(mFragment); 75 } 76 77 @Test 78 public void getAvailabilityStatus_notInstantApp_shouldReturnDisabled() { 79 ReflectionHelpers.setStaticField(AppUtils.class, "sInstantAppDataProvider", 80 (InstantAppDataProvider) (i -> false)); 81 82 assertThat(mController.getAvailabilityStatus()) 83 .isEqualTo(BasePreferenceController.DISABLED_FOR_USER); 84 } 85 86 @Test 87 public void getAvailabilityStatus_isInstantApp_shouldReturnAvailable() { 88 ReflectionHelpers.setStaticField(AppUtils.class, "sInstantAppDataProvider", 89 (InstantAppDataProvider) (i -> true)); 90 91 assertThat(mController.getAvailabilityStatus()) 92 .isEqualTo(BasePreferenceController.AVAILABLE); 93 } 94 95 @Test 96 public void updateState_shouldSetPreferenceTitle() { 97 final String[] domain = {"Domain1"}; 98 final ArraySet<String> domains = new ArraySet<>(); 99 domains.add(domain[0]); 100 final List<IntentFilterVerificationInfo> infoList = new ArrayList<>(); 101 final IntentFilterVerificationInfo info = 102 new IntentFilterVerificationInfo("Package1", domains); 103 infoList.add(info); 104 105 when(mPackageManager.getIntentFilterVerifications("Package1")).thenReturn(infoList); 106 107 mController.updateState(mPreference); 108 109 verify(mPreference).setTitles(domain); 110 } 111 } 112