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.datausage; 18 19 import static org.mockito.Matchers.any; 20 import static org.mockito.Matchers.anyBoolean; 21 import static org.mockito.Matchers.anyInt; 22 import static org.mockito.Matchers.anyString; 23 import static org.mockito.Mockito.RETURNS_DEEP_STUBS; 24 import static org.mockito.Mockito.doNothing; 25 import static org.mockito.Mockito.doReturn; 26 import static org.mockito.Mockito.mock; 27 import static org.mockito.Mockito.spy; 28 import static org.mockito.Mockito.verify; 29 import static org.mockito.Mockito.when; 30 31 import android.content.pm.PackageManager; 32 import android.net.NetworkPolicyManager; 33 import android.os.Bundle; 34 import android.support.v7.preference.PreferenceManager; 35 import android.support.v7.preference.PreferenceScreen; 36 import android.util.ArraySet; 37 import android.view.View; 38 39 import com.android.settings.testutils.FakeFeatureFactory; 40 import com.android.settings.testutils.SettingsRobolectricTestRunner; 41 import com.android.settings.testutils.shadow.ShadowEntityHeaderController; 42 import com.android.settings.testutils.shadow.ShadowRestrictedLockUtils; 43 import com.android.settings.widget.EntityHeaderController; 44 import com.android.settingslib.AppItem; 45 import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin; 46 import com.android.settingslib.RestrictedSwitchPreference; 47 import com.android.settingslib.wrapper.PackageManagerWrapper; 48 49 import org.junit.After; 50 import org.junit.Before; 51 import org.junit.Test; 52 import org.junit.runner.RunWith; 53 import org.mockito.Answers; 54 import org.mockito.Mock; 55 import org.mockito.MockitoAnnotations; 56 import org.robolectric.RuntimeEnvironment; 57 import org.robolectric.annotation.Config; 58 import org.robolectric.util.ReflectionHelpers; 59 60 @RunWith(SettingsRobolectricTestRunner.class) 61 @Config(shadows = {ShadowEntityHeaderController.class, ShadowRestrictedLockUtils.class}) 62 public class AppDataUsageTest { 63 64 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 65 private EntityHeaderController mHeaderController; 66 @Mock 67 private PackageManagerWrapper mPackageManagerWrapper; 68 69 private AppDataUsage mFragment; 70 71 @Before setUp()72 public void setUp() { 73 MockitoAnnotations.initMocks(this); 74 FakeFeatureFactory.setupForTest(); 75 } 76 77 @After tearDown()78 public void tearDown() { 79 ShadowEntityHeaderController.reset(); 80 } 81 82 @Test bindAppHeader_allWorkApps_shouldNotShowAppInfoLink()83 public void bindAppHeader_allWorkApps_shouldNotShowAppInfoLink() { 84 ShadowEntityHeaderController.setUseMock(mHeaderController); 85 when(mHeaderController.setRecyclerView(any(), any())).thenReturn(mHeaderController); 86 when(mHeaderController.setUid(anyInt())).thenReturn(mHeaderController); 87 88 mFragment = spy(new AppDataUsage()); 89 90 when(mFragment.getPreferenceManager()) 91 .thenReturn(mock(PreferenceManager.class, RETURNS_DEEP_STUBS)); 92 doReturn(mock(PreferenceScreen.class)).when(mFragment).getPreferenceScreen(); 93 ReflectionHelpers.setField(mFragment, "mAppItem", mock(AppItem.class)); 94 95 mFragment.onViewCreated(new View(RuntimeEnvironment.application), new Bundle()); 96 97 verify(mHeaderController).setHasAppInfoLink(false); 98 } 99 100 @Test bindAppHeader_workApp_shouldSetWorkAppUid()101 public void bindAppHeader_workApp_shouldSetWorkAppUid() throws 102 PackageManager.NameNotFoundException { 103 final int fakeUserId = 100; 104 105 mFragment = spy(new AppDataUsage()); 106 final ArraySet<String> packages = new ArraySet<>(); 107 packages.add("pkg"); 108 final AppItem appItem = new AppItem(123456789); 109 110 ReflectionHelpers.setField(mFragment, "mPackageManagerWrapper", mPackageManagerWrapper); 111 ReflectionHelpers.setField(mFragment, "mAppItem", appItem); 112 ReflectionHelpers.setField(mFragment, "mPackages", packages); 113 114 when(mPackageManagerWrapper.getPackageUidAsUser(anyString(), anyInt())) 115 .thenReturn(fakeUserId); 116 117 ShadowEntityHeaderController.setUseMock(mHeaderController); 118 when(mHeaderController.setRecyclerView(any(), any())).thenReturn(mHeaderController); 119 when(mHeaderController.setUid(fakeUserId)).thenReturn(mHeaderController); 120 when(mHeaderController.setHasAppInfoLink(anyBoolean())).thenReturn(mHeaderController); 121 122 when(mFragment.getPreferenceManager()) 123 .thenReturn(mock(PreferenceManager.class, RETURNS_DEEP_STUBS)); 124 doReturn(mock(PreferenceScreen.class)).when(mFragment).getPreferenceScreen(); 125 126 mFragment.onViewCreated(new View(RuntimeEnvironment.application), new Bundle()); 127 128 verify(mHeaderController).setHasAppInfoLink(true); 129 verify(mHeaderController).setUid(fakeUserId); 130 } 131 132 @Test changePreference_backgroundData_shouldUpdateUI()133 public void changePreference_backgroundData_shouldUpdateUI() { 134 mFragment = spy(new AppDataUsage()); 135 final AppItem appItem = new AppItem(123456789); 136 final RestrictedSwitchPreference pref = mock(RestrictedSwitchPreference.class); 137 final DataSaverBackend dataSaverBackend = mock(DataSaverBackend.class); 138 ReflectionHelpers.setField(mFragment, "mAppItem", appItem); 139 ReflectionHelpers.setField(mFragment, "mRestrictBackground", pref); 140 ReflectionHelpers.setField(mFragment, "mDataSaverBackend", dataSaverBackend); 141 142 doNothing().when(mFragment).updatePrefs(); 143 144 mFragment.onPreferenceChange(pref, true /* value */); 145 146 verify(mFragment).updatePrefs(); 147 } 148 149 @Test updatePrefs_restrictedByAdmin_shouldDisablePreference()150 public void updatePrefs_restrictedByAdmin_shouldDisablePreference() { 151 mFragment = spy(new AppDataUsage()); 152 final int testUid = 123123; 153 final AppItem appItem = new AppItem(testUid); 154 final RestrictedSwitchPreference restrictBackgroundPref 155 = mock(RestrictedSwitchPreference.class); 156 final RestrictedSwitchPreference unrestrictedDataPref 157 = mock(RestrictedSwitchPreference.class); 158 final DataSaverBackend dataSaverBackend = mock(DataSaverBackend.class); 159 final NetworkPolicyManager networkPolicyManager = mock(NetworkPolicyManager.class); 160 ReflectionHelpers.setField(mFragment, "mAppItem", appItem); 161 ReflectionHelpers.setField(mFragment, "mRestrictBackground", restrictBackgroundPref); 162 ReflectionHelpers.setField(mFragment, "mUnrestrictedData", unrestrictedDataPref); 163 ReflectionHelpers.setField(mFragment, "mDataSaverBackend", dataSaverBackend); 164 ReflectionHelpers.setField(mFragment.services, "mPolicyManager", networkPolicyManager); 165 166 ShadowRestrictedLockUtils.setRestricted(true); 167 doReturn(NetworkPolicyManager.POLICY_NONE).when(networkPolicyManager) 168 .getUidPolicy(testUid); 169 170 mFragment.updatePrefs(); 171 172 verify(restrictBackgroundPref).setDisabledByAdmin(any(EnforcedAdmin.class)); 173 verify(unrestrictedDataPref).setDisabledByAdmin(any(EnforcedAdmin.class)); 174 } 175 } 176