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.fuelgauge; 17 18 import android.content.Context; 19 import android.os.PowerManager; 20 21 import com.android.settings.testutils.SettingsRobolectricTestRunner; 22 import com.android.settings.TestConfig; 23 import com.android.settings.widget.MasterSwitchPreference; 24 import com.android.settingslib.core.lifecycle.Lifecycle; 25 26 import org.junit.Before; 27 import org.junit.Test; 28 import org.junit.runner.RunWith; 29 import org.mockito.Mock; 30 import org.mockito.MockitoAnnotations; 31 import org.robolectric.annotation.Config; 32 import org.robolectric.util.ReflectionHelpers; 33 34 import static org.mockito.Mockito.doNothing; 35 import static org.mockito.Mockito.spy; 36 import static org.mockito.Mockito.verify; 37 import static org.mockito.Mockito.when; 38 39 @RunWith(SettingsRobolectricTestRunner.class) 40 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION) 41 public class BatterySaverControllerTest { 42 @Mock 43 private MasterSwitchPreference mBatterySaverPref; 44 @Mock 45 private PowerManager mPowerManager; 46 @Mock 47 private Context mContext; 48 @Mock 49 private Lifecycle mLifecycle; 50 private BatterySaverController mBatterySaverController; 51 52 @Before setUp()53 public void setUp() { 54 MockitoAnnotations.initMocks(this); 55 56 mBatterySaverController = spy(new BatterySaverController(mContext, mLifecycle)); 57 ReflectionHelpers.setField(mBatterySaverController, "mPowerManager", mPowerManager); 58 ReflectionHelpers.setField(mBatterySaverController, "mBatterySaverPref", mBatterySaverPref); 59 doNothing().when(mBatterySaverController).refreshConditionManager(); 60 } 61 62 @Test testOnPreferenceChange_TurnOnBatterySaver_BatterySaverOn()63 public void testOnPreferenceChange_TurnOnBatterySaver_BatterySaverOn() { 64 testOnPreferenceChangeInner(true); 65 } 66 67 @Test testOnPreferenceChange_TurnOffBatterySaver_BatterySaverOff()68 public void testOnPreferenceChange_TurnOffBatterySaver_BatterySaverOff() { 69 testOnPreferenceChangeInner(false); 70 } 71 72 @Test testUpdateState_SaverModeOn_PreferenceChecked()73 public void testUpdateState_SaverModeOn_PreferenceChecked() { 74 testUpdateStateInner(true); 75 } 76 77 @Test testUpdateState_SaverModeOff_PreferenceUnChecked()78 public void testUpdateState_SaverModeOff_PreferenceUnChecked() { 79 testUpdateStateInner(false); 80 } 81 82 @Test testOnBatteryChanged_pluggedIn_setDisable()83 public void testOnBatteryChanged_pluggedIn_setDisable() { 84 mBatterySaverController.onBatteryChanged(true /* pluggedIn */); 85 86 verify(mBatterySaverPref).setSwitchEnabled(false); 87 } 88 89 @Test testOnBatteryChanged_notPluggedIn_setEnable()90 public void testOnBatteryChanged_notPluggedIn_setEnable() { 91 mBatterySaverController.onBatteryChanged(false /* pluggedIn */); 92 93 verify(mBatterySaverPref).setSwitchEnabled(true); 94 } 95 testOnPreferenceChangeInner(final boolean saverOn)96 private void testOnPreferenceChangeInner(final boolean saverOn) { 97 when(mPowerManager.setPowerSaveMode(saverOn)).thenReturn(true); 98 when(mPowerManager.isPowerSaveMode()).thenReturn(!saverOn); 99 100 mBatterySaverController.onPreferenceChange(mBatterySaverPref, saverOn); 101 verify(mPowerManager).setPowerSaveMode(saverOn); 102 } 103 testUpdateStateInner(final boolean saverOn)104 private void testUpdateStateInner(final boolean saverOn) { 105 when(mPowerManager.isPowerSaveMode()).thenReturn(saverOn); 106 107 mBatterySaverController.updateState(mBatterySaverPref); 108 verify(mBatterySaverPref).setChecked(saverOn); 109 } 110 } 111