Index: tools/telemetry/telemetry/unittest_util/system_stub.py |
diff --git a/tools/telemetry/telemetry/unittest_util/system_stub.py b/tools/telemetry/telemetry/unittest_util/system_stub.py |
index cb7c20dd795d96b5692516f52a01876c89252a7f..27f3956dc22ad9cf3b094f4a00405fde16217eb7 100644 |
--- a/tools/telemetry/telemetry/unittest_util/system_stub.py |
+++ b/tools/telemetry/telemetry/unittest_util/system_stub.py |
@@ -9,14 +9,12 @@ This test allows one to test code that itself uses os, sys, and subprocess. |
import os |
import re |
-import shlex |
import sys |
class Override(object): |
def __init__(self, base_module, module_list): |
- stubs = {'adb_commands': AdbCommandsModuleStub, |
- 'cloud_storage': CloudStorageModuleStub, |
+ stubs = {'cloud_storage': CloudStorageModuleStub, |
'open': OpenFunctionStub, |
'os': OsModuleStub, |
'perf_control': PerfControlModuleStub, |
@@ -60,96 +58,6 @@ class Override(object): |
self._overrides = {} |
-class AdbDevice(object): |
- |
- def __init__(self): |
- self.has_root = False |
- self.needs_su = False |
- self.shell_command_handlers = {} |
- self.mock_content = [] |
- self.system_properties = {} |
- if self.system_properties.get('ro.product.cpu.abi') == None: |
- self.system_properties['ro.product.cpu.abi'] = 'armeabi-v7a' |
- |
- def HasRoot(self): |
- return self.has_root |
- |
- def NeedsSU(self): |
- return self.needs_su |
- |
- def RunShellCommand(self, args, **_kwargs): |
- if isinstance(args, basestring): |
- args = shlex.split(args) |
- handler = self.shell_command_handlers[args[0]] |
- return handler(args) |
- |
- def FileExists(self, _): |
- return False |
- |
- def ReadFile(self, device_path, as_root=False): # pylint: disable=W0613 |
- return self.mock_content |
- |
- def GetProp(self, property_name): |
- return self.system_properties[property_name] |
- |
- def SetProp(self, property_name, property_value): |
- self.system_properties[property_name] = property_value |
- |
- |
-class AdbCommandsModuleStub(object): |
- |
- class AdbCommandsStub(object): |
- |
- def __init__(self, module, device): |
- self._module = module |
- self._device = device |
- self.is_root_enabled = True |
- self._adb_device = module.adb_device |
- |
- def IsRootEnabled(self): |
- return self.is_root_enabled |
- |
- def RestartAdbdOnDevice(self): |
- pass |
- |
- def IsUserBuild(self): |
- return False |
- |
- def WaitForDevicePm(self): |
- pass |
- |
- def device(self): |
- return self._adb_device |
- |
- def device_serial(self): |
- return self._device |
- |
- def __init__(self): |
- self.attached_devices = [] |
- self.apk_package_name = None |
- self.adb_device = AdbDevice() |
- |
- def AdbCommandsStubConstructor(device=None): |
- return AdbCommandsModuleStub.AdbCommandsStub(self, device) |
- self.AdbCommands = AdbCommandsStubConstructor |
- |
- @staticmethod |
- def IsAndroidSupported(): |
- return True |
- |
- def GetPackageName(self, _): |
- return self.apk_package_name |
- |
- def GetAttachedDevices(self): |
- return self.attached_devices |
- |
- def SetupPrebuiltTools(self, _): |
- return True |
- |
- def CleanupLeftoverProcesses(self): |
- pass |
- |
- |
class CloudStorageModuleStub(object): |
PUBLIC_BUCKET = 'chromium-telemetry' |
PARTNER_BUCKET = 'chrome-partner-telemetry' |