Index: tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py |
diff --git a/tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py b/tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py |
index f4ce39c08dcf12ebbaa3c84c7ed10ce68c7c7465..926ad1f5a6628f6135fd4253706e52af8d18939c 100644 |
--- a/tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py |
+++ b/tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py |
@@ -6,29 +6,26 @@ import unittest |
from telemetry import benchmark |
from telemetry.core import browser_options |
+from telemetry.core.platform import android_device |
+from telemetry.core.platform import android_platform_backend |
from telemetry.core.backends.chrome import android_browser_finder |
from telemetry.unittest import system_stub |
-class LoggingStub(object): |
- def __init__(self): |
- self.warnings = [] |
- |
- def info(self, msg, *args): |
- pass |
- |
- def warn(self, msg, *args): |
- self.warnings.append(msg % args) |
- |
- |
class AndroidBrowserFinderTest(unittest.TestCase): |
def setUp(self): |
self._stubs = system_stub.Override(android_browser_finder, |
- ['adb_commands', 'os', 'subprocess']) |
- self._log_stub = LoggingStub() |
+ ['adb_commands', 'os', 'subprocess', |
+ 'logging']) |
+ self._android_device_stub = system_stub.Override( |
+ android_device, ['adb_commands']) |
+ self._apb_stub = system_stub.Override( |
+ android_platform_backend, ['adb_commands']) |
def tearDown(self): |
self._stubs.Restore() |
+ self._android_device_stub.Restore() |
+ self._apb_stub.Restore() |
def test_no_adb(self): |
finder_options = browser_options.BrowserFinderOptions() |
@@ -36,15 +33,13 @@ class AndroidBrowserFinderTest(unittest.TestCase): |
def NoAdb(*args, **kargs): # pylint: disable=W0613 |
raise OSError('not found') |
self._stubs.subprocess.Popen = NoAdb |
- browsers = android_browser_finder.FindAllAvailableBrowsers( |
- finder_options, self._log_stub) |
+ browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options) |
self.assertEquals(0, len(browsers)) |
def test_adb_no_devices(self): |
finder_options = browser_options.BrowserFinderOptions() |
- browsers = android_browser_finder.FindAllAvailableBrowsers( |
- finder_options, self._log_stub) |
+ browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options) |
self.assertEquals(0, len(browsers)) |
def test_adb_permissions_error(self): |
@@ -56,28 +51,26 @@ class AndroidBrowserFinderTest(unittest.TestCase): |
"""* daemon not running. starting it now on port 5037 * |
* daemon started successfully * |
""") |
- |
- browsers = android_browser_finder.FindAllAvailableBrowsers( |
- finder_options, self._log_stub) |
- self.assertEquals(3, len(self._log_stub.warnings)) |
+ browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options) |
+ self.assertEquals(3, len(self._stubs.logging.warnings)) |
self.assertEquals(0, len(browsers)) |
def test_adb_two_devices(self): |
finder_options = browser_options.BrowserFinderOptions() |
- self._stubs.adb_commands.attached_devices = ['015d14fec128220c', |
- '015d14fec128220d'] |
+ self._android_device_stub.adb_commands.attached_devices = [ |
+ '015d14fec128220c', '015d14fec128220d'] |
- browsers = android_browser_finder.FindAllAvailableBrowsers( |
- finder_options, self._log_stub) |
- self.assertEquals(1, len(self._log_stub.warnings)) |
+ browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options) |
+ self.assertEquals(1, len(self._stubs.logging.warnings)) |
self.assertEquals(0, len(browsers)) |
@benchmark.Disabled('chromeos') |
def test_adb_one_device(self): |
finder_options = browser_options.BrowserFinderOptions() |
- self._stubs.adb_commands.attached_devices = ['015d14fec128220c'] |
+ self._android_device_stub.adb_commands.attached_devices = ( |
+ ['015d14fec128220c']) |
def OnPM(args): |
assert args[0] == 'pm' |
@@ -89,9 +82,8 @@ class AndroidBrowserFinderTest(unittest.TestCase): |
def OnLs(_): |
return ['/sys/devices/system/cpu/cpu0'] |
- self._stubs.adb_commands.shell_command_handlers['pm'] = OnPM |
- self._stubs.adb_commands.shell_command_handlers['ls'] = OnLs |
+ self._apb_stub.adb_commands.adb_device.shell_command_handlers['pm'] = OnPM |
+ self._apb_stub.adb_commands.adb_device.shell_command_handlers['ls'] = OnLs |
- browsers = android_browser_finder.FindAllAvailableBrowsers( |
- finder_options, self._log_stub) |
+ browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options) |
self.assertEquals(1, len(browsers)) |