MockBatteryStatsImpl.java revision 3d422c37b7f2caf364febc556e17b43c40c56be6
1/* 2 * Copyright (C) 2016 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 17package com.android.internal.os; 18 19import android.os.Handler; 20import android.os.Looper; 21import android.util.SparseIntArray; 22 23import java.util.ArrayList; 24import java.util.concurrent.Future; 25 26/** 27 * Mocks a BatteryStatsImpl object. 28 */ 29public class MockBatteryStatsImpl extends BatteryStatsImpl { 30 public BatteryStatsImpl.Clocks clocks; 31 public boolean mForceOnBattery; 32 33 MockBatteryStatsImpl(Clocks clocks) { 34 super(clocks); 35 this.clocks = mClocks; 36 mScreenOnTimer = new BatteryStatsImpl.StopwatchTimer(clocks, null, -1, null, 37 mOnBatteryTimeBase); 38 mScreenDozeTimer = new BatteryStatsImpl.StopwatchTimer(clocks, null, -1, null, 39 mOnBatteryTimeBase); 40 mBluetoothScanTimer = new StopwatchTimer(mClocks, null, -14, null, mOnBatteryTimeBase); 41 setExternalStatsSyncLocked(new DummyExternalStatsSync()); 42 43 // A no-op handler. 44 mHandler = new Handler(Looper.getMainLooper()) {}; 45 } 46 47 MockBatteryStatsImpl() { 48 this(new MockClocks()); 49 } 50 51 public TimeBase getOnBatteryTimeBase() { 52 return mOnBatteryTimeBase; 53 } 54 55 public TimeBase getOnBatteryScreenOffTimeBase() { 56 return mOnBatteryScreenOffTimeBase; 57 } 58 59 public int getScreenState() { 60 return mScreenState; 61 } 62 63 public boolean isOnBattery() { 64 return mForceOnBattery ? true : super.isOnBattery(); 65 } 66 67 public void forceRecordAllHistory() { 68 mHaveBatteryLevel = true; 69 mRecordingHistory = true; 70 mRecordAllHistory = true; 71 } 72 73 public TimeBase getOnBatteryBackgroundTimeBase(int uid) { 74 return getUidStatsLocked(uid).mOnBatteryBackgroundTimeBase; 75 } 76 77 public TimeBase getOnBatteryScreenOffBackgroundTimeBase(int uid) { 78 return getUidStatsLocked(uid).mOnBatteryScreenOffBackgroundTimeBase; 79 } 80 81 public MockBatteryStatsImpl setPowerProfile(PowerProfile powerProfile) { 82 mPowerProfile = powerProfile; 83 return this; 84 } 85 86 public MockBatteryStatsImpl setKernelUidCpuFreqTimeReader(KernelUidCpuFreqTimeReader reader) { 87 mKernelUidCpuFreqTimeReader = reader; 88 return this; 89 } 90 91 public MockBatteryStatsImpl setKernelUidCpuActiveTimeReader(KernelUidCpuActiveTimeReader reader) { 92 mKernelUidCpuActiveTimeReader = reader; 93 return this; 94 } 95 96 public MockBatteryStatsImpl setKernelUidCpuClusterTimeReader(KernelUidCpuClusterTimeReader reader) { 97 mKernelUidCpuClusterTimeReader = reader; 98 return this; 99 } 100 101 public MockBatteryStatsImpl setKernelUidCpuTimeReader(KernelUidCpuTimeReader reader) { 102 mKernelUidCpuTimeReader = reader; 103 return this; 104 } 105 106 public MockBatteryStatsImpl setKernelSingleUidTimeReader(KernelSingleUidTimeReader reader) { 107 mKernelSingleUidTimeReader = reader; 108 return this; 109 } 110 111 public MockBatteryStatsImpl setKernelCpuSpeedReaders(KernelCpuSpeedReader[] readers) { 112 mKernelCpuSpeedReaders = readers; 113 return this; 114 } 115 116 public MockBatteryStatsImpl setUserInfoProvider(UserInfoProvider provider) { 117 mUserInfoProvider = provider; 118 return this; 119 } 120 121 public MockBatteryStatsImpl setPartialTimers(ArrayList<StopwatchTimer> partialTimers) { 122 mPartialTimers = partialTimers; 123 return this; 124 } 125 126 public MockBatteryStatsImpl setLastPartialTimers(ArrayList<StopwatchTimer> lastPartialTimers) { 127 mLastPartialTimers = lastPartialTimers; 128 return this; 129 } 130 131 public MockBatteryStatsImpl setOnBatteryInternal(boolean onBatteryInternal) { 132 mOnBatteryInternal = onBatteryInternal; 133 return this; 134 } 135 136 public SparseIntArray getPendingUids() { 137 return mPendingUids; 138 } 139 140 private class DummyExternalStatsSync implements ExternalStatsSync { 141 @Override 142 public Future<?> scheduleSync(String reason, int flags) { 143 return null; 144 } 145 146 @Override 147 public Future<?> scheduleCpuSyncDueToRemovedUid(int uid) { 148 return null; 149 } 150 151 @Override 152 public Future<?> scheduleCpuSyncDueToSettingChange() { 153 return null; 154 } 155 156 @Override 157 public Future<?> scheduleReadProcStateCpuTimes( 158 boolean onBattery, boolean onBatteryScreenOff) { 159 return null; 160 } 161 162 @Override 163 public Future<?> scheduleCopyFromAllUidsCpuTimes( 164 boolean onBattery, boolean onBatteryScreenOff) { 165 return null; 166 } 167 168 } 169} 170 171