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 package com.android.settings.development; 17 18 import static com.google.common.truth.Truth.assertThat; 19 import static org.mockito.Matchers.any; 20 import static org.mockito.Matchers.anyInt; 21 import static org.mockito.Matchers.anyString; 22 import static org.mockito.Mockito.mock; 23 import static org.mockito.Mockito.spy; 24 import static org.mockito.Mockito.verify; 25 import static org.mockito.Mockito.when; 26 27 import android.content.Context; 28 import android.content.Intent; 29 import android.content.pm.ResolveInfo; 30 import android.provider.Settings; 31 import android.provider.Settings.Global; 32 import android.support.v7.preference.PreferenceScreen; 33 34 import com.android.settings.testutils.SettingsRobolectricTestRunner; 35 import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin; 36 import com.android.settingslib.RestrictedSwitchPreference; 37 import com.android.settingslib.wrapper.PackageManagerWrapper; 38 39 import org.junit.Before; 40 import org.junit.Test; 41 import org.junit.runner.RunWith; 42 import org.mockito.Mock; 43 import org.mockito.MockitoAnnotations; 44 import org.robolectric.RuntimeEnvironment; 45 import org.robolectric.util.ReflectionHelpers; 46 47 import java.util.Collections; 48 import java.util.List; 49 50 @RunWith(SettingsRobolectricTestRunner.class) 51 public class VerifyAppsOverUsbPreferenceControllerTest { 52 53 @Mock 54 private PackageManagerWrapper mPackageManager; 55 @Mock 56 private PreferenceScreen mScreen; 57 @Mock 58 private RestrictedSwitchPreference mPreference; 59 60 @Mock 61 private VerifyAppsOverUsbPreferenceController.RestrictedLockUtilsDelegate 62 mRestrictedLockUtilsDelegate; 63 64 private Context mContext; 65 private VerifyAppsOverUsbPreferenceController mController; 66 67 /** Convenience class for setting global int settings. */ 68 class GlobalSetter { set(String setting, int value)69 public GlobalSetter set(String setting, int value) { 70 Global.putInt(mContext.getContentResolver(), setting, value); 71 return this; 72 } 73 } 74 75 private final GlobalSetter mGlobals = new GlobalSetter(); 76 77 @Before setUp()78 public void setUp() { 79 MockitoAnnotations.initMocks(this); 80 mContext = RuntimeEnvironment.application; 81 when(mScreen.findPreference(anyString())).thenReturn(mPreference); 82 mController = new VerifyAppsOverUsbPreferenceController(mContext); 83 ReflectionHelpers 84 .setField(mController, "mRestrictedLockUtils", mRestrictedLockUtilsDelegate); 85 ReflectionHelpers.setField(mController, "mPackageManager", mPackageManager); 86 mController.displayPreference(mScreen); 87 } 88 setupVerifyBroadcastReceivers(boolean nonEmpty)89 private void setupVerifyBroadcastReceivers(boolean nonEmpty) { 90 final List<ResolveInfo> resolveInfos = nonEmpty 91 ? Collections.singletonList(mock(ResolveInfo.class)) 92 : Collections.emptyList(); 93 when(mPackageManager.queryBroadcastReceivers((Intent) any(), anyInt())) 94 .thenReturn(resolveInfos); 95 } 96 setupEnforcedAdmin(EnforcedAdmin result)97 private void setupEnforcedAdmin(EnforcedAdmin result) { 98 when(mRestrictedLockUtilsDelegate 99 .checkIfRestrictionEnforced(any(), anyString(), anyInt())).thenReturn(result); 100 } 101 102 @Test updateState_settingEnabled_preferenceShouldBeChecked()103 public void updateState_settingEnabled_preferenceShouldBeChecked() { 104 setupVerifyBroadcastReceivers(true); 105 setupEnforcedAdmin(null); 106 mGlobals.set(Global.ADB_ENABLED, 1 /* setting enabled */) 107 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 1 /* setting enabled */); 108 mController.updateState(mPreference); 109 verify(mPreference).setChecked(true); 110 } 111 112 @Test updateState_settingDisabled_preferenceShouldNotBeChecked()113 public void updateState_settingDisabled_preferenceShouldNotBeChecked() { 114 setupVerifyBroadcastReceivers(true); 115 setupEnforcedAdmin(null); 116 mGlobals.set(Global.ADB_ENABLED, 1 /* setting enabled */) 117 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 0 /* setting disabled */); 118 mController.updateState(mPreference); 119 verify(mPreference).setChecked(false); 120 } 121 122 @Test updateState_adbDisabled_preferenceShouldNotBeChecked()123 public void updateState_adbDisabled_preferenceShouldNotBeChecked() { 124 setupVerifyBroadcastReceivers(true); 125 setupEnforcedAdmin(null); 126 mGlobals.set(Global.ADB_ENABLED, 0 /* setting disabled */) 127 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 1 /* setting enabled */); 128 mController.updateState(mPreference); 129 verify(mPreference).setChecked(false); 130 } 131 132 @Test updateState_verifierOff_preferenceShouldNotBeChecked()133 public void updateState_verifierOff_preferenceShouldNotBeChecked() { 134 setupVerifyBroadcastReceivers(true); 135 setupEnforcedAdmin(null); 136 mGlobals.set(Global.ADB_ENABLED, 1 /* setting enabled */) 137 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 1 /* setting enabled */) 138 .set(Global.PACKAGE_VERIFIER_ENABLE, 0 /* setting disabled */); 139 mController.updateState(mPreference); 140 verify(mPreference).setChecked(false); 141 } 142 143 @Test updateState_noBroadcastReceivers_preferenceShouldNotBeChecked()144 public void updateState_noBroadcastReceivers_preferenceShouldNotBeChecked() { 145 setupVerifyBroadcastReceivers(false); 146 setupEnforcedAdmin(null); 147 mGlobals.set(Global.ADB_ENABLED, 1 /* setting enabled */) 148 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 1 /* setting enabled */); 149 mController.updateState(mPreference); 150 verify(mPreference).setChecked(false); 151 } 152 153 @Test updateState_restrictedByAdmin_preferenceShouldBeDisabled()154 public void updateState_restrictedByAdmin_preferenceShouldBeDisabled() { 155 setupVerifyBroadcastReceivers(true); 156 final EnforcedAdmin admin = new EnforcedAdmin(); 157 setupEnforcedAdmin(admin); 158 mGlobals.set(Global.ADB_ENABLED, 1 /* setting enabled */) 159 .set(Global.PACKAGE_VERIFIER_INCLUDE_ADB, 1 /* setting enabled */); 160 mController.updateState(mPreference); 161 verify(mPreference).setDisabledByAdmin(admin); 162 } 163 164 @Test isAvailable_verifierNotVisible_shouldReturnFalse()165 public void isAvailable_verifierNotVisible_shouldReturnFalse() { 166 setupVerifyBroadcastReceivers(true); 167 mGlobals.set(Global.PACKAGE_VERIFIER_SETTING_VISIBLE, 0 /* setting disabled */); 168 169 assertThat(mController.isAvailable()).isFalse(); 170 } 171 172 @Test isAvailable_verifierVisible_shouldReturnTrue()173 public void isAvailable_verifierVisible_shouldReturnTrue() { 174 setupVerifyBroadcastReceivers(true); 175 mGlobals.set(Global.PACKAGE_VERIFIER_SETTING_VISIBLE, 1 /* setting enabled */); 176 177 assertThat(mController.isAvailable()).isTrue(); 178 } 179 180 @Test onPreferenceChange_settingEnabled_shouldEnableUsbVerify()181 public void onPreferenceChange_settingEnabled_shouldEnableUsbVerify() { 182 mController.onPreferenceChange(mPreference, true /* new value */); 183 184 final int mode = Settings.Global.getInt(mContext.getContentResolver(), 185 android.provider.Settings.Global.PACKAGE_VERIFIER_INCLUDE_ADB, -1 /* default */); 186 187 assertThat(mode).isEqualTo(VerifyAppsOverUsbPreferenceController.SETTING_VALUE_ON); 188 } 189 190 @Test onPreferenceChange_settingDisabled_shouldDisableUsbVerify()191 public void onPreferenceChange_settingDisabled_shouldDisableUsbVerify() { 192 mController.onPreferenceChange(mPreference, false /* new value */); 193 194 final int mode = Settings.Global.getInt(mContext.getContentResolver(), 195 android.provider.Settings.Global.PACKAGE_VERIFIER_INCLUDE_ADB, -1 /* default */); 196 197 assertThat(mode).isEqualTo(VerifyAppsOverUsbPreferenceController.SETTING_VALUE_OFF); 198 } 199 }