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 static com.android.settings.fuelgauge.BatteryBroadcastReceiver.BatteryUpdateType;
19 
20 import static com.google.common.truth.Truth.assertThat;
21 
22 import static org.mockito.ArgumentMatchers.any;
23 import static org.mockito.ArgumentMatchers.anyInt;
24 import static org.mockito.Mockito.doReturn;
25 import static org.mockito.Mockito.never;
26 import static org.mockito.Mockito.spy;
27 import static org.mockito.Mockito.times;
28 import static org.mockito.Mockito.verify;
29 
30 import android.content.Context;
31 import android.content.Intent;
32 import android.os.BatteryManager;
33 import android.os.PowerManager;
34 
35 import com.android.settings.Utils;
36 import com.android.settings.homepage.contextualcards.slices.BatteryFixSliceTest;
37 
38 import org.junit.Before;
39 import org.junit.Test;
40 import org.junit.runner.RunWith;
41 import org.mockito.Mock;
42 import org.mockito.MockitoAnnotations;
43 import org.robolectric.RobolectricTestRunner;
44 import org.robolectric.RuntimeEnvironment;
45 import org.robolectric.annotation.Config;
46 
47 @RunWith(RobolectricTestRunner.class)
48 public class BatteryBroadcastReceiverTest {
49 
50     private static final String BATTERY_INIT_LEVEL = "100%";
51     private static final String BATTERY_INIT_STATUS = "Not charging";
52     private static final int BATTERY_INTENT_LEVEL = 80;
53     private static final int BATTERY_INTENT_SCALE = 100;
54 
55     @Mock
56     private BatteryBroadcastReceiver.OnBatteryChangedListener mBatteryListener;
57     private BatteryBroadcastReceiver mBatteryBroadcastReceiver;
58     private Context mContext;
59     private Intent mChargingIntent;
60 
61     @Before
setUp()62     public void setUp() {
63         MockitoAnnotations.initMocks(this);
64         mContext = spy(RuntimeEnvironment.application);
65 
66         mBatteryBroadcastReceiver = new BatteryBroadcastReceiver(mContext);
67         mBatteryBroadcastReceiver.mBatteryLevel = BATTERY_INIT_LEVEL;
68         mBatteryBroadcastReceiver.mBatteryStatus = BATTERY_INIT_STATUS;
69         mBatteryBroadcastReceiver.mBatteryHealth = BatteryManager.BATTERY_HEALTH_UNKNOWN;
70         mBatteryBroadcastReceiver.setBatteryChangedListener(mBatteryListener);
71 
72         mChargingIntent = new Intent(Intent.ACTION_BATTERY_CHANGED);
73         mChargingIntent.putExtra(BatteryManager.EXTRA_LEVEL, BATTERY_INTENT_LEVEL);
74         mChargingIntent.putExtra(BatteryManager.EXTRA_SCALE, BATTERY_INTENT_SCALE);
75         mChargingIntent
76                 .putExtra(BatteryManager.EXTRA_STATUS, BatteryManager.BATTERY_STATUS_CHARGING);
77     }
78 
79     @Test
80     @Config(shadows = {
81             BatteryFixSliceTest.ShadowBatteryTipLoader.class
82     })
testOnReceive_batteryLevelChanged_dataUpdated()83     public void testOnReceive_batteryLevelChanged_dataUpdated() {
84         mBatteryBroadcastReceiver.onReceive(mContext, mChargingIntent);
85 
86         assertThat(mBatteryBroadcastReceiver.mBatteryLevel)
87                 .isEqualTo(Utils.getBatteryPercentage(mChargingIntent));
88         assertThat(mBatteryBroadcastReceiver.mBatteryStatus)
89                 .isEqualTo(Utils.getBatteryStatus(mContext, mChargingIntent));
90         verify(mBatteryListener).onBatteryChanged(BatteryUpdateType.BATTERY_LEVEL);
91     }
92 
93     @Test
94     @Config(shadows = {
95             BatteryFixSliceTest.ShadowBatteryTipLoader.class
96     })
testOnReceive_batteryHealthChanged_dataUpdated()97     public void testOnReceive_batteryHealthChanged_dataUpdated() {
98         mChargingIntent
99                 .putExtra(BatteryManager.EXTRA_HEALTH, BatteryManager.BATTERY_HEALTH_OVERHEAT);
100         mBatteryBroadcastReceiver.onReceive(mContext, mChargingIntent);
101 
102         assertThat(mBatteryBroadcastReceiver.mBatteryHealth)
103                 .isEqualTo(BatteryManager.BATTERY_HEALTH_OVERHEAT);
104         verify(mBatteryListener).onBatteryChanged(BatteryUpdateType.BATTERY_HEALTH);
105     }
106 
107     @Test
108     @Config(shadows = {
109             BatteryFixSliceTest.ShadowBatteryTipLoader.class
110     })
onReceive_batteryNotPresent_shouldShowHelpMessage()111     public void onReceive_batteryNotPresent_shouldShowHelpMessage() {
112         mChargingIntent.putExtra(BatteryManager.EXTRA_PRESENT, false);
113 
114         mBatteryBroadcastReceiver.onReceive(mContext, mChargingIntent);
115 
116         verify(mBatteryListener).onBatteryChanged(BatteryUpdateType.BATTERY_NOT_PRESENT);
117     }
118 
119     @Test
120     @Config(shadows = {
121             BatteryFixSliceTest.ShadowBatteryTipLoader.class
122     })
testOnReceive_powerSaveModeChanged_listenerInvoked()123     public void testOnReceive_powerSaveModeChanged_listenerInvoked() {
124         mBatteryBroadcastReceiver.onReceive(mContext,
125                 new Intent(PowerManager.ACTION_POWER_SAVE_MODE_CHANGED));
126 
127         verify(mBatteryListener).onBatteryChanged(BatteryUpdateType.BATTERY_SAVER);
128     }
129 
130     @Test
131     @Config(shadows = {
132             BatteryFixSliceTest.ShadowBatteryTipLoader.class
133     })
testOnReceive_batteryDataNotChanged_listenerNotInvoked()134     public void testOnReceive_batteryDataNotChanged_listenerNotInvoked() {
135         final String batteryLevel = Utils.getBatteryPercentage(mChargingIntent);
136         final String batteryStatus =
137                 Utils.getBatteryStatus(mContext, mChargingIntent);
138         mBatteryBroadcastReceiver.mBatteryLevel = batteryLevel;
139         mBatteryBroadcastReceiver.mBatteryStatus = batteryStatus;
140 
141         mBatteryBroadcastReceiver.onReceive(mContext, mChargingIntent);
142 
143         assertThat(mBatteryBroadcastReceiver.mBatteryLevel).isEqualTo(batteryLevel);
144         assertThat(mBatteryBroadcastReceiver.mBatteryStatus).isEqualTo(batteryStatus);
145         assertThat(mBatteryBroadcastReceiver.mBatteryHealth)
146                 .isEqualTo(BatteryManager.BATTERY_HEALTH_UNKNOWN);
147         verify(mBatteryListener, never()).onBatteryChanged(anyInt());
148     }
149 
150     @Test
151     @Config(shadows = {
152             BatteryFixSliceTest.ShadowBatteryTipLoader.class
153     })
testRegister_updateBatteryStatus()154     public void testRegister_updateBatteryStatus() {
155         doReturn(mChargingIntent).when(mContext).registerReceiver(any(), any());
156 
157         mBatteryBroadcastReceiver.register();
158         mBatteryBroadcastReceiver.register();
159 
160         assertThat(mBatteryBroadcastReceiver.mBatteryLevel)
161                 .isEqualTo(Utils.getBatteryPercentage(mChargingIntent));
162         assertThat(mBatteryBroadcastReceiver.mBatteryStatus)
163                 .isEqualTo(Utils.getBatteryStatus(mContext, mChargingIntent));
164         assertThat(mBatteryBroadcastReceiver.mBatteryHealth)
165                 .isEqualTo(BatteryManager.BATTERY_HEALTH_UNKNOWN);
166         // 2 times because register will force update the battery
167         verify(mBatteryListener, times(2)).onBatteryChanged(BatteryUpdateType.MANUAL);
168     }
169 }
170