MockBatteryStatsImpl.java revision 695cf72f26638f051d356abcb4acdc0c1fe3918d
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 setKernelUidCpuTimeReader(KernelUidCpuTimeReader reader) { 92 mKernelUidCpuTimeReader = reader; 93 return this; 94 } 95 96 public MockBatteryStatsImpl setKernelSingleUidTimeReader(KernelSingleUidTimeReader reader) { 97 mKernelSingleUidTimeReader = reader; 98 return this; 99 } 100 101 public MockBatteryStatsImpl setKernelCpuSpeedReaders(KernelCpuSpeedReader[] readers) { 102 mKernelCpuSpeedReaders = readers; 103 return this; 104 } 105 106 public MockBatteryStatsImpl setUserInfoProvider(UserInfoProvider provider) { 107 mUserInfoProvider = provider; 108 return this; 109 } 110 111 public MockBatteryStatsImpl setPartialTimers(ArrayList<StopwatchTimer> partialTimers) { 112 mPartialTimers = partialTimers; 113 return this; 114 } 115 116 public MockBatteryStatsImpl setLastPartialTimers(ArrayList<StopwatchTimer> lastPartialTimers) { 117 mLastPartialTimers = lastPartialTimers; 118 return this; 119 } 120 121 public MockBatteryStatsImpl setOnBatteryInternal(boolean onBatteryInternal) { 122 mOnBatteryInternal = onBatteryInternal; 123 return this; 124 } 125 126 public SparseIntArray getPendingUids() { 127 return mPendingUids; 128 } 129 130 private class DummyExternalStatsSync implements ExternalStatsSync { 131 @Override 132 public Future<?> scheduleSync(String reason, int flags) { 133 return null; 134 } 135 136 @Override 137 public Future<?> scheduleCpuSyncDueToRemovedUid(int uid) { 138 return null; 139 } 140 141 @Override 142 public Future<?> scheduleReadProcStateCpuTimes() { 143 return null; 144 } 145 146 @Override 147 public Future<?> scheduleCopyFromAllUidsCpuTimes() { 148 return null; 149 } 150 151 } 152} 153 154