src/tools/telemetry/telemetry/core/backends/chrome/android_browser_finder_unittest.py

   1 # Copyright 2013 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import unittest
   6
   7 from telemetry import test
   8 from telemetry.core import browser_options
   9 from telemetry.core.backends.chrome import android_browser_finder
  10 from telemetry.unittest import system_stub
  11
  12
  13 class LoggingStub(object):
  14   def __init__(self):
  15     self.warnings = []
  16
  17   def info(self, msg, *args):
  18     pass
  19
  20   def warn(self, msg, *args):
  21     self.warnings.append(msg % args)
  22
  23
  24 class AndroidBrowserFinderTest(unittest.TestCase):
  25   def setUp(self):
  26     self._stubs = system_stub.Override(android_browser_finder,
  27                                        ['adb_commands', 'os', 'subprocess'])
  28     android_browser_finder.adb_works = None  # Blow cache between runs.
  29
  30   def tearDown(self):
  31     self._stubs.Restore()
  32
  33   def test_no_adb(self):
  34     finder_options = browser_options.BrowserFinderOptions()
  35
  36     def NoAdb(*args, **kargs): # pylint: disable=W0613
  37       raise OSError('not found')
  38     self._stubs.subprocess.Popen = NoAdb
  39     browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
  40     self.assertEquals(0, len(browsers))
  41
  42   def test_adb_no_devices(self):
  43     finder_options = browser_options.BrowserFinderOptions()
  44
  45     browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
  46     self.assertEquals(0, len(browsers))
  47
  48   def test_adb_permissions_error(self):
  49     finder_options = browser_options.BrowserFinderOptions()
  50
  51     self._stubs.subprocess.Popen.communicate_result = (
  52         """List of devices attached
  53 ????????????\tno permissions""",
  54         """* daemon not running. starting it now on port 5037 *
  55 * daemon started successfully *
  56 """)
  57
  58     log_stub = LoggingStub()
  59     browsers = android_browser_finder.FindAllAvailableBrowsers(
  60       finder_options, log_stub)
  61     self.assertEquals(3, len(log_stub.warnings))
  62     self.assertEquals(0, len(browsers))
  63
  64   def test_adb_two_devices(self):
  65     finder_options = browser_options.BrowserFinderOptions()
  66
  67     self._stubs.adb_commands.attached_devices = ['015d14fec128220c',
  68                                                  '015d14fec128220d']
  69
  70     log_stub = LoggingStub()
  71     browsers = android_browser_finder.FindAllAvailableBrowsers(
  72       finder_options, log_stub)
  73     self.assertEquals(1, len(log_stub.warnings))
  74     self.assertEquals(0, len(browsers))
  75
  76   @test.Disabled('chromeos')
  77   def test_adb_one_device(self):
  78     finder_options = browser_options.BrowserFinderOptions()
  79
  80     self._stubs.adb_commands.attached_devices = ['015d14fec128220c']
  81
  82     def OnPM(args):
  83       assert args[0] == 'pm'
  84       assert args[1] == 'list'
  85       assert args[2] == 'packages'
  86       return ['package:org.chromium.content_shell_apk',
  87               'package.com.google.android.setupwizard']
  88
  89     self._stubs.adb_commands.shell_command_handlers['pm'] = OnPM
  90
  91     browsers = android_browser_finder.FindAllAvailableBrowsers(finder_options)
  92     self.assertEquals(1, len(browsers))