1 /* 2 * Copyright (C) 2021 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.fuelgauge; 18 19 import static com.google.common.truth.Truth.assertThat; 20 21 import static org.mockito.Mockito.when; 22 23 import com.android.settingslib.widget.SelectorWithWidgetPreference; 24 25 import org.junit.Before; 26 import org.junit.Test; 27 import org.junit.runner.RunWith; 28 import org.mockito.Mock; 29 import org.mockito.MockitoAnnotations; 30 import org.robolectric.RobolectricTestRunner; 31 import org.robolectric.RuntimeEnvironment; 32 33 @RunWith(RobolectricTestRunner.class) 34 public class UnrestrictedPreferenceControllerTest { 35 private static final int UID = 12345; 36 private static final String PACKAGE_NAME = "com.android.app"; 37 38 private UnrestrictedPreferenceController mController; 39 private SelectorWithWidgetPreference mPreference; 40 41 @Mock BatteryOptimizeUtils mockBatteryOptimizeUtils; 42 43 @Before setUp()44 public void setUp() { 45 MockitoAnnotations.initMocks(this); 46 47 mController = new UnrestrictedPreferenceController( 48 RuntimeEnvironment.application, UID, PACKAGE_NAME); 49 mPreference = new SelectorWithWidgetPreference(RuntimeEnvironment.application); 50 mController.mBatteryOptimizeUtils = mockBatteryOptimizeUtils; 51 } 52 53 @Test testUpdateState_isValidPackage_prefEnabled()54 public void testUpdateState_isValidPackage_prefEnabled() { 55 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 56 57 mController.updateState(mPreference); 58 59 assertThat(mPreference.isEnabled()).isTrue(); 60 } 61 62 @Test testUpdateState_invalidPackage_prefDisabled()63 public void testUpdateState_invalidPackage_prefDisabled() { 64 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true); 65 66 mController.updateState(mPreference); 67 68 assertThat(mPreference.isEnabled()).isFalse(); 69 } 70 71 @Test testUpdateState_isSystemOrDefaultAppAndUnrestrictedStates_prefChecked()72 public void testUpdateState_isSystemOrDefaultAppAndUnrestrictedStates_prefChecked() { 73 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 74 when(mockBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true); 75 when(mockBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn( 76 BatteryOptimizeUtils.MODE_UNRESTRICTED); 77 78 mController.updateState(mPreference); 79 80 assertThat(mPreference.isChecked()).isTrue(); 81 } 82 83 @Test testUpdateState_isSystemOrDefaultApp_prefUnchecked()84 public void testUpdateState_isSystemOrDefaultApp_prefUnchecked() { 85 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 86 when(mockBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true); 87 88 mController.updateState(mPreference); 89 90 assertThat(mPreference.isChecked()).isFalse(); 91 assertThat(mPreference.isEnabled()).isFalse(); 92 } 93 94 @Test testUpdateState_isUnrestrictedStates_prefChecked()95 public void testUpdateState_isUnrestrictedStates_prefChecked() { 96 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 97 when(mockBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn( 98 BatteryOptimizeUtils.MODE_UNRESTRICTED); 99 100 mController.updateState(mPreference); 101 102 assertThat(mPreference.isChecked()).isTrue(); 103 } 104 105 @Test testUpdateState_prefUnchecked()106 public void testUpdateState_prefUnchecked() { 107 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 108 109 mController.updateState(mPreference); 110 111 assertThat(mPreference.isChecked()).isFalse(); 112 } 113 114 @Test testHandlePreferenceTreeClick_samePrefKey_verifyAction()115 public void testHandlePreferenceTreeClick_samePrefKey_verifyAction() { 116 mPreference.setKey(mController.KEY_UNRESTRICTED_PREF); 117 mController.handlePreferenceTreeClick(mPreference); 118 119 assertThat(mController.handlePreferenceTreeClick(mPreference)).isTrue(); 120 } 121 122 @Test testHandlePreferenceTreeClick_incorrectPrefKey_noAction()123 public void testHandlePreferenceTreeClick_incorrectPrefKey_noAction() { 124 assertThat(mController.handlePreferenceTreeClick(mPreference)).isFalse(); 125 } 126 } 127