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 OptimizedPreferenceControllerTest { 35 private static final int UID = 12345; 36 private static final String PACKAGE_NAME = "com.android.app"; 37 38 private OptimizedPreferenceController 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 OptimizedPreferenceController( 48 RuntimeEnvironment.application, UID, PACKAGE_NAME); 49 mPreference = new SelectorWithWidgetPreference(RuntimeEnvironment.application); 50 mController.mBatteryOptimizeUtils = mockBatteryOptimizeUtils; 51 } 52 53 @Test testUpdateState_invalidPackage_prefEnabled()54 public void testUpdateState_invalidPackage_prefEnabled() { 55 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true); 56 57 mController.updateState(mPreference); 58 59 assertThat(mPreference.isEnabled()).isTrue(); 60 assertThat(mPreference.isChecked()).isTrue(); 61 } 62 63 @Test testUpdateState_isSystemOrDefaultAppAndOptimizeStates_prefChecked()64 public void testUpdateState_isSystemOrDefaultAppAndOptimizeStates_prefChecked() { 65 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 66 when(mockBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true); 67 when(mockBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn( 68 BatteryOptimizeUtils.MODE_OPTIMIZED); 69 70 mController.updateState(mPreference); 71 72 assertThat(mPreference.isChecked()).isTrue(); 73 } 74 75 @Test testUpdateState_isSystemOrDefaultApp_prefUnchecked()76 public void testUpdateState_isSystemOrDefaultApp_prefUnchecked() { 77 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 78 when(mockBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true); 79 80 mController.updateState(mPreference); 81 82 assertThat(mPreference.isChecked()).isFalse(); 83 assertThat(mPreference.isEnabled()).isFalse(); 84 } 85 86 @Test testUpdateState_isOptimizedStates_prefChecked()87 public void testUpdateState_isOptimizedStates_prefChecked() { 88 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 89 when(mockBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn( 90 BatteryOptimizeUtils.MODE_OPTIMIZED); 91 92 mController.updateState(mPreference); 93 94 assertThat(mPreference.isChecked()).isTrue(); 95 } 96 97 @Test testUpdateState_prefUnchecked()98 public void testUpdateState_prefUnchecked() { 99 when(mockBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false); 100 101 mController.updateState(mPreference); 102 103 assertThat(mPreference.isChecked()).isFalse(); 104 } 105 106 @Test testHandlePreferenceTreeClick_samePrefKey_verifyAction()107 public void testHandlePreferenceTreeClick_samePrefKey_verifyAction() { 108 mPreference.setKey(mController.KEY_OPTIMIZED_PREF); 109 mController.handlePreferenceTreeClick(mPreference); 110 111 assertThat(mController.handlePreferenceTreeClick(mPreference)).isTrue(); 112 } 113 114 @Test testHandlePreferenceTreeClick_incorrectPrefKey_noAction()115 public void testHandlePreferenceTreeClick_incorrectPrefKey_noAction() { 116 assertThat(mController.handlePreferenceTreeClick(mPreference)).isFalse(); 117 } 118 } 119