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 879eaf9c09a4adc68a6bb0ebdf83ec20bcd3bade..a3b50ead2ce81f8c031b976b9358c383286aff51 100644 |
--- a/tools/telemetry/telemetry/unittest/system_stub.py |
+++ b/tools/telemetry/telemetry/unittest/system_stub.py |
@@ -197,11 +197,14 @@ class OsModuleStub(object): |
X_OK = os.X_OK |
+ pathsep = os.pathsep |
+ |
def __init__(self, sys_module=sys): |
self.path = OsModuleStub.OsPathModuleStub(sys_module) |
self.environ = OsModuleStub.OsEnvironModuleStub() |
self.display = ':0' |
self.local_app_data = None |
+ self.sys_path = None |
self.program_files = None |
self.program_files_x86 = None |
self.devnull = os.devnull |
@@ -209,16 +212,20 @@ class OsModuleStub(object): |
def access(self, path, _): |
return path in self.path.files |
- def getenv(self, name): |
+ def getenv(self, name, value=None): |
if name == 'DISPLAY': |
- return self.display |
+ env = self.display |
elif name == 'LOCALAPPDATA': |
- return self.local_app_data |
+ env = self.local_app_data |
+ elif name == 'PATH': |
+ env = self.sys_path |
elif name == 'PROGRAMFILES': |
- return self.program_files |
+ env = self.program_files |
elif name == 'PROGRAMFILES(X86)': |
- return self.program_files_x86 |
- raise Exception('Unsupported getenv') |
+ env = self.program_files_x86 |
+ else: |
+ raise NotImplementedError('Unsupported getenv') |
+ return env if env else value |
class PerfControlModuleStub(object): |