Chrome updater test framework

pyautolib/pyauto.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
 run pydoc on this file.
@@ skipping 31 matching lines @@
 import string
 import subprocess
 import sys
 import tempfile
 import time
 import types
 import unittest
 import urllib
 
 import pyauto_paths
+from py_unittest_util import GTestTextTestRunner
 
 
 def _LocateBinDirs():
   """Setup a few dirs where we expect to find dependency libraries."""
   deps_dirs = [
       os.path.dirname(__file__),
       pyauto_paths.GetThirdPartyDir(),
       os.path.join(pyauto_paths.GetThirdPartyDir(), 'webdriver', 'pylib'),
   ]
   sys.path += map(os.path.normpath, pyauto_paths.GetBuildDirs() + deps_dirs)
@@ skipping 65 matching lines @@
     Args:
       methodName: the default method name. Internal use by unittest module
 
       (The rest of the args can be in any order. They can even be skipped in
        which case the defaults will be used.)
 
       clear_profile: If True, clean the profile dir before use. Defaults to True
       homepage: the home page. Defaults to "about:blank"
     """
     # Fetch provided keyword args, or fill in defaults.
+    browser_path = kwargs.get('browser_path', None)
     clear_profile = kwargs.get('clear_profile', True)
     homepage = kwargs.get('homepage', 'about:blank')
 
     pyautolib.PyUITestBase.__init__(self, clear_profile, homepage)
-    self.Initialize(pyautolib.FilePath(self.BrowserPath()))
+    if browser_path:
+      self.Initialize(pyautolib.FilePath(browser_path))
+    else:
+      self.Initialize(pyautolib.FilePath(self.BrowserPath()))
     unittest.TestCase.__init__(self, methodName)
 
     # Give all pyauto tests easy access to pprint.PrettyPrinter functions.
     self.pprint = pprint.pprint
     self.pformat = pprint.pformat
 
     # Set up remote proxies, if they were requested.
     self.remotes = []
     self.remote = None
     global _REMOTE_PROXY
@@ skipping 5576 matching lines @@
     logging.debug("Loading %d tests from %s", len(test_names), test_names)
     if self._options.list_tests:  # List tests and exit
       for name in test_names:
         print name
       sys.exit(0)
     pyauto_suite = PyUITestSuite(suite_args)
     loaded_tests = unittest.defaultTestLoader.loadTestsFromNames(test_names)
     pyauto_suite.addTests(loaded_tests)
     verbosity = 1
     if self._options.verbose:
-      verbosity = 2
-    result = PyAutoTextTestRunner(verbosity=verbosity).run(pyauto_suite)
+      verbosity = 2    
+    result = GTestTextTestRunner(verbosity=verbosity).run(pyauto_suite)
     del loaded_tests  # Need to destroy test cases before the suite
     del pyauto_suite
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()