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.enterprise; 18 19 import android.content.Context; 20 import android.support.v7.preference.Preference; 21 22 import com.android.internal.logging.nano.MetricsProto.MetricsEvent; 23 import com.android.settings.R; 24 import com.android.settings.SettingsRobolectricTestRunner; 25 import com.android.settings.TestConfig; 26 import com.android.settings.applications.ApplicationFeatureProvider; 27 import com.android.settings.core.PreferenceAvailabilityObserver; 28 import com.android.settings.testutils.FakeFeatureFactory; 29 30 import org.junit.Before; 31 import org.junit.Test; 32 import org.junit.runner.RunWith; 33 import org.mockito.Answers; 34 import org.mockito.Mock; 35 import org.mockito.MockitoAnnotations; 36 import org.mockito.invocation.InvocationOnMock; 37 import org.mockito.stubbing.Answer; 38 import org.robolectric.annotation.Config; 39 40 import static com.google.common.truth.Truth.assertThat; 41 import static org.mockito.Mockito.anyBoolean; 42 import static org.mockito.Mockito.anyObject; 43 import static org.mockito.Mockito.doAnswer; 44 import static org.mockito.Mockito.eq; 45 import static org.mockito.Mockito.never; 46 import static org.mockito.Mockito.verify; 47 import static org.mockito.Mockito.when; 48 49 /** 50 * Tests for {@link EnterpriseInstalledPackagesPreferenceController}. 51 */ 52 @RunWith(SettingsRobolectricTestRunner.class) 53 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION) 54 public final class EnterpriseInstalledPackagesPreferenceControllerTest { 55 56 private static final String KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES 57 = "number_enterprise_installed_packages"; 58 59 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 60 private Context mContext; 61 private FakeFeatureFactory mFeatureFactory; 62 @Mock private PreferenceAvailabilityObserver mObserver; 63 64 private EnterpriseInstalledPackagesPreferenceController mController; 65 66 @Before setUp()67 public void setUp() { 68 MockitoAnnotations.initMocks(this); 69 FakeFeatureFactory.setupForTest(mContext); 70 mFeatureFactory = (FakeFeatureFactory) FakeFeatureFactory.getFactory(mContext); 71 mController = new EnterpriseInstalledPackagesPreferenceController(mContext, 72 null /* lifecycle */, true /* async */); 73 mController.setAvailabilityObserver(mObserver); 74 } 75 76 @Test testGetAvailabilityObserver()77 public void testGetAvailabilityObserver() { 78 assertThat(mController.getAvailabilityObserver()).isEqualTo(mObserver); 79 } 80 setNumberOfEnterpriseInstalledPackages(int number, boolean async)81 private void setNumberOfEnterpriseInstalledPackages(int number, boolean async) { 82 doAnswer(new Answer() { 83 public Object answer(InvocationOnMock invocation) { 84 ((ApplicationFeatureProvider.NumberOfAppsCallback) 85 invocation.getArguments()[1]).onNumberOfAppsResult(number); 86 return null; 87 }}).when(mFeatureFactory.applicationFeatureProvider) 88 .calculateNumberOfPolicyInstalledApps(eq(async), anyObject()); 89 } 90 91 @Test testUpdateState()92 public void testUpdateState() { 93 final Preference preference = new Preference(mContext, null, 0, 0); 94 preference.setVisible(true); 95 96 setNumberOfEnterpriseInstalledPackages(0, true /* async */); 97 mController.updateState(preference); 98 assertThat(preference.isVisible()).isFalse(); 99 verify(mObserver).onPreferenceAvailabilityUpdated(KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES, 100 false); 101 102 setNumberOfEnterpriseInstalledPackages(20, true /* async */); 103 when(mContext.getResources().getQuantityString( 104 R.plurals.enterprise_privacy_number_packages_lower_bound, 20, 20)) 105 .thenReturn("minimum 20 apps"); 106 mController.updateState(preference); 107 assertThat(preference.getSummary()).isEqualTo("minimum 20 apps"); 108 assertThat(preference.isVisible()).isTrue(); 109 verify(mObserver).onPreferenceAvailabilityUpdated(KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES, 110 true); 111 } 112 113 @Test testIsAvailableSync()114 public void testIsAvailableSync() { 115 final EnterpriseInstalledPackagesPreferenceController controller 116 = new EnterpriseInstalledPackagesPreferenceController(mContext, 117 null /* lifecycle */, false /* async */); 118 controller.setAvailabilityObserver(mObserver); 119 120 setNumberOfEnterpriseInstalledPackages(0, false /* async */); 121 assertThat(controller.isAvailable()).isFalse(); 122 verify(mObserver).onPreferenceAvailabilityUpdated( 123 KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES, false); 124 125 setNumberOfEnterpriseInstalledPackages(20, false /* async */); 126 assertThat(controller.isAvailable()).isTrue(); 127 verify(mObserver).onPreferenceAvailabilityUpdated( 128 KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES, true); 129 } 130 131 @Test testIsAvailableAsync()132 public void testIsAvailableAsync() { 133 setNumberOfEnterpriseInstalledPackages(0, true /* async */); 134 assertThat(mController.isAvailable()).isTrue(); 135 verify(mObserver, never()).onPreferenceAvailabilityUpdated( 136 eq(KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES), anyBoolean()); 137 138 setNumberOfEnterpriseInstalledPackages(20, true /* async */); 139 assertThat(mController.isAvailable()).isTrue(); 140 verify(mObserver, never()).onPreferenceAvailabilityUpdated( 141 eq(KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES), anyBoolean()); 142 } 143 144 @Test testHandlePreferenceTreeClick()145 public void testHandlePreferenceTreeClick() { 146 assertThat(mController.handlePreferenceTreeClick(new Preference(mContext, null, 0, 0))) 147 .isFalse(); 148 } 149 150 @Test testGetPreferenceKey()151 public void testGetPreferenceKey() { 152 assertThat(mController.getPreferenceKey()) 153 .isEqualTo(KEY_NUMBER_ENTERPRISE_INSTALLED_PACKAGES); 154 } 155 } 156