Home | History | Annotate | Download | only in fuelgauge 
      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 
     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 android.content.Context;
     24 import android.content.pm.PackageManager;
     25 import android.os.Process;
     26 
     27 import com.android.internal.os.BatterySipper;
     28 import com.android.settings.testutils.SettingsRobolectricTestRunner;
     29 
     30 import org.junit.Before;
     31 import org.junit.Test;
     32 import org.junit.runner.RunWith;
     33 import org.mockito.Answers;
     34 import org.mockito.Mock;
     35 import org.mockito.MockitoAnnotations;
     36 
     37 @RunWith(SettingsRobolectricTestRunner.class)
     38 public class PowerUsageFeatureProviderImplTest {
     39 
     40     private static final int UID_OTHER = Process.FIRST_APPLICATION_UID + 2;
     41     private static final int UID_CALENDAR = Process.FIRST_APPLICATION_UID + 3;
     42     private static final int UID_MEDIA = Process.FIRST_APPLICATION_UID + 4;
     43     private static final int UID_SYSTEMUI = Process.FIRST_APPLICATION_UID + 5;
     44     private static final String[] PACKAGES_CALENDAR = {"com.android.providers.calendar"};
     45     private static final String[] PACKAGES_MEDIA = {"com.android.providers.media"};
     46     private static final String[] PACKAGES_SYSTEMUI = {"com.android.systemui"};
     47 
     48     @Mock(answer = Answers.RETURNS_DEEP_STUBS)
     49     private Context mContext;
     50     @Mock
     51     private BatterySipper mBatterySipper;
     52     @Mock
     53     private PackageManager mPackageManager;
     54     private PowerUsageFeatureProviderImpl mPowerFeatureProvider;
     55 
     56     @Before
     57     public void setUp() {
     58         MockitoAnnotations.initMocks(this);
     59 
     60         when(mContext.getApplicationContext()).thenReturn(mContext);
     61         mPowerFeatureProvider = new PowerUsageFeatureProviderImpl(mContext);
     62         when(mPackageManager.getPackagesForUid(UID_CALENDAR)).thenReturn(PACKAGES_CALENDAR);
     63         when(mPackageManager.getPackagesForUid(UID_MEDIA)).thenReturn(PACKAGES_MEDIA);
     64         when(mPackageManager.getPackagesForUid(UID_SYSTEMUI)).thenReturn(PACKAGES_SYSTEMUI);
     65         mPowerFeatureProvider.mPackageManager = mPackageManager;
     66         mBatterySipper.uidObj = new FakeUid(UID_OTHER);
     67     }
     68 
     69     @Test
     70     public void testIsTypeSystem_uidRoot_returnTrue() {
     71         mBatterySipper.drainType = BatterySipper.DrainType.APP;
     72         when(mBatterySipper.getUid()).thenReturn(Process.ROOT_UID);
     73 
     74         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
     75     }
     76 
     77     @Test
     78     public void testIsTypeSystem_uidSystem_returnTrue() {
     79         mBatterySipper.drainType = BatterySipper.DrainType.APP;
     80         when(mBatterySipper.getUid()).thenReturn(Process.SYSTEM_UID);
     81 
     82         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
     83     }
     84 
     85     @Test
     86     public void testIsTypeSystem_uidMedia_returnTrue() {
     87         mBatterySipper.drainType = BatterySipper.DrainType.APP;
     88         when(mBatterySipper.getUid()).thenReturn(Process.MEDIA_UID);
     89 
     90         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
     91     }
     92 
     93     @Test
     94     public void testIsTypeSystem_appCalendar_returnTrue() {
     95         mBatterySipper.drainType = BatterySipper.DrainType.APP;
     96         when(mBatterySipper.getUid()).thenReturn(UID_CALENDAR);
     97 
     98         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
     99     }
    100 
    101     @Test
    102     public void testIsTypeSystem_appMedia_returnTrue() {
    103         mBatterySipper.drainType = BatterySipper.DrainType.APP;
    104         when(mBatterySipper.getUid()).thenReturn(UID_MEDIA);
    105 
    106         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
    107     }
    108 
    109     @Test
    110     public void testIsTypeSystem_appSystemUi_returnTrue() {
    111         mBatterySipper.drainType = BatterySipper.DrainType.APP;
    112         when(mBatterySipper.getUid()).thenReturn(UID_SYSTEMUI);
    113 
    114         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isTrue();
    115     }
    116 
    117     @Test
    118     public void testIsTypeSystem_uidOther_returnFalse() {
    119         mBatterySipper.drainType = BatterySipper.DrainType.APP;
    120         when(mBatterySipper.getUid()).thenReturn(UID_OTHER);
    121 
    122         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isFalse();
    123     }
    124 
    125     @Test
    126     public void testIsTypeSystem_uidObjNull_returnFalse() {
    127         mBatterySipper.drainType = BatterySipper.DrainType.APP;
    128         mBatterySipper.uidObj = null;
    129 
    130         assertThat(mPowerFeatureProvider.isTypeSystem(mBatterySipper)).isFalse();
    131     }
    132 
    133     @Test
    134     public void testIsAdvancedUiEnabled_returnTrue() {
    135         assertThat(mPowerFeatureProvider.isAdvancedUiEnabled()).isTrue();
    136     }
    137 
    138     @Test
    139     public void testIsPowerAccountingToggleEnabled_returnTrue() {
    140         assertThat(mPowerFeatureProvider.isPowerAccountingToggleEnabled()).isTrue();
    141     }
    142 
    143     @Test
    144     public void testIsSmartBatterySupported_smartBatterySupported_returnTrue() {
    145         when(mContext.getResources().getBoolean(
    146                 com.android.internal.R.bool.config_smart_battery_available)).thenReturn(true);
    147 
    148         assertThat(mPowerFeatureProvider.isSmartBatterySupported()).isTrue();
    149     }
    150 
    151     @Test
    152     public void testIsSmartBatterySupported_smartBatteryNotSupported_returnFalse() {
    153         when(mContext.getResources().getBoolean(
    154                 com.android.internal.R.bool.config_smart_battery_available)).thenReturn(false);
    155 
    156         assertThat(mPowerFeatureProvider.isSmartBatterySupported()).isFalse();
    157     }
    158 }
    159