Index: tools/telemetry/telemetry/unittest/system_stub.py |
diff --git a/tools/telemetry/telemetry/unittest/system_stub.py b/tools/telemetry/telemetry/unittest/system_stub.py |
index b647f5fb5930d35298b01610f1af91cb2a437043..922d37f266dbe097acf45e1281215069cdb38a0c 100644 |
--- a/tools/telemetry/telemetry/unittest/system_stub.py |
+++ b/tools/telemetry/telemetry/unittest/system_stub.py |
@@ -14,6 +14,8 @@ import sys as real_sys |
class Override(object): |
def __init__(self, base_module, module_list): |
stubs = {'adb_commands': AdbCommandsModuleStub, |
+ 'perf_control': PerfControlModuleStub, |
+ 'thermal_throttle': ThermalThrottleModuleStub, |
'os': OsModuleStub, |
'subprocess': SubprocessModuleStub, |
'sys': SysModuleStub, |
@@ -79,6 +81,24 @@ class AdbCommandsModuleStub(object): |
def HasForwarder(_=None): |
return True |
+class PerfControlModuleStub(object): |
+ class PerfControlStub(object): |
+ def __init__(self, adb): |
+ pass |
+ |
+ def __init__(self): |
+ self.PerfControl = PerfControlModuleStub.PerfControlStub |
+ |
+ |
+class ThermalThrottleModuleStub(object): |
+ class ThermalThrottleStub(object): |
+ def __init__(self, adb): |
+ pass |
+ |
+ def __init__(self): |
+ self.ThermalThrottle = ThermalThrottleModuleStub.ThermalThrottleStub |
+ |
+ |
class OsModuleStub(object): |
class OsPathModuleStub(object): |
def __init__(self, sys_module): |