| 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 926ad1f5a6628f6135fd4253706e52af8d18939c..f4ce39c08dcf12ebbaa3c84c7ed10ce68c7c7465 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,26 +6,29 @@
|
|
|
| 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',
|
| - 'logging'])
|
| - self._android_device_stub = system_stub.Override(
|
| - android_device, ['adb_commands'])
|
| - self._apb_stub = system_stub.Override(
|
| - android_platform_backend, ['adb_commands'])
|
| + ['adb_commands', 'os', 'subprocess'])
|
| + self._log_stub = LoggingStub()
|
|
|
| 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()
|
| @@ -33,13 +36,15 @@
|
| def NoAdb(*args, **kargs): # pylint: disable=W0613
|
| raise OSError('not found')
|
| self._stubs.subprocess.Popen = NoAdb
|
| - browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
|
| + browsers = android_browser_finder.FindAllAvailableBrowsers(
|
| + finder_options, self._log_stub)
|
| self.assertEquals(0, len(browsers))
|
|
|
| def test_adb_no_devices(self):
|
| finder_options = browser_options.BrowserFinderOptions()
|
|
|
| - browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
|
| + browsers = android_browser_finder.FindAllAvailableBrowsers(
|
| + finder_options, self._log_stub)
|
| self.assertEquals(0, len(browsers))
|
|
|
| def test_adb_permissions_error(self):
|
| @@ -51,26 +56,28 @@
|
| """* daemon not running. starting it now on port 5037 *
|
| * daemon started successfully *
|
| """)
|
| - browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
|
| - self.assertEquals(3, len(self._stubs.logging.warnings))
|
| +
|
| + browsers = android_browser_finder.FindAllAvailableBrowsers(
|
| + finder_options, self._log_stub)
|
| + self.assertEquals(3, len(self._log_stub.warnings))
|
| self.assertEquals(0, len(browsers))
|
|
|
| def test_adb_two_devices(self):
|
| finder_options = browser_options.BrowserFinderOptions()
|
|
|
| - self._android_device_stub.adb_commands.attached_devices = [
|
| - '015d14fec128220c', '015d14fec128220d']
|
| + self._stubs.adb_commands.attached_devices = ['015d14fec128220c',
|
| + '015d14fec128220d']
|
|
|
| - browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
|
| - self.assertEquals(1, len(self._stubs.logging.warnings))
|
| + browsers = android_browser_finder.FindAllAvailableBrowsers(
|
| + finder_options, self._log_stub)
|
| + self.assertEquals(1, len(self._log_stub.warnings))
|
| self.assertEquals(0, len(browsers))
|
|
|
| @benchmark.Disabled('chromeos')
|
| def test_adb_one_device(self):
|
| finder_options = browser_options.BrowserFinderOptions()
|
|
|
| - self._android_device_stub.adb_commands.attached_devices = (
|
| - ['015d14fec128220c'])
|
| + self._stubs.adb_commands.attached_devices = ['015d14fec128220c']
|
|
|
| def OnPM(args):
|
| assert args[0] == 'pm'
|
| @@ -82,8 +89,9 @@
|
| def OnLs(_):
|
| return ['/sys/devices/system/cpu/cpu0']
|
|
|
| - self._apb_stub.adb_commands.adb_device.shell_command_handlers['pm'] = OnPM
|
| - self._apb_stub.adb_commands.adb_device.shell_command_handlers['ls'] = OnLs
|
| + self._stubs.adb_commands.shell_command_handlers['pm'] = OnPM
|
| + self._stubs.adb_commands.shell_command_handlers['ls'] = OnLs
|
|
|
| - browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
|
| + browsers = android_browser_finder.FindAllAvailableBrowsers(
|
| + finder_options, self._log_stub)
|
| self.assertEquals(1, len(browsers))
|
|
|