| 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))
|
|
|