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