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 10ec1ce1fea7de0edb8785198cbe2244ce04be56..1e9ddb522dbbcbcab07fdc3178544bbd23269751 100644 |
--- a/tools/telemetry/telemetry/unittest/system_stub.py |
+++ b/tools/telemetry/telemetry/unittest/system_stub.py |
@@ -20,6 +20,7 @@ class Override(object): |
'open': OpenFunctionStub, |
'os': OsModuleStub, |
'perf_control': PerfControlModuleStub, |
+ 'raw_input': RawInputFunctionStub, |
'subprocess': SubprocessModuleStub, |
'sys': SysModuleStub, |
'thermal_throttle': ThermalThrottleModuleStub, |
@@ -219,6 +220,14 @@ class PerfControlModuleStub(object): |
self.PerfControl = PerfControlModuleStub.PerfControlStub |
+class RawInputFunctionStub(object): |
+ def __init__(self): |
+ self.input = '' |
+ |
+ def __call__(self, name, *args, **kwargs): |
+ return self.input |
+ |
+ |
class SubprocessModuleStub(object): |
class PopenStub(object): |
def __init__(self): |