PyAuto: Flags to forbid cleaning profile dir...

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/python
 
 # Copyright (c) 2010 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,
@@ skipping 50 matching lines @@
 
   Example:
 
     class MyTest(PyUITest):
 
       def testNavigation(self):
         self.NavigateToURL("http://www.google.com")
         self.assertTrue("Google" == self.GetActiveTabTitle())
   """
 
-  def __init__(self, methodName='runTest', extra_chrome_flags=None):
+  def __init__(self, methodName='runTest', **kwargs):
     """Initialize PyUITest.
 
     When redefining __init__ in a derived class, make sure that:
       o you make a call this __init__
       o __init__ takes methodName as a arg. this is mandated by unittest module
 
     Args:
       methodName: the default method name. Internal use by unittest module
-      extra_chrome_flags: additional flags to pass when launching chrome
+
+      (The rest of the args can be in any order. They can even be skipped in
+       which case the defaults will be used.)
+
+      extra_chrome_flags: additional flags to pass when launching chrome.
+                          Defaults to None
+      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.
+    extra_chrome_flags = kwargs.get('extra_chrome_flags')
+    clear_profile = kwargs.get('clear_profile', True)
+    homepage = kwargs.get('homepage', 'about:blank')
+
     args = sys.argv
     if extra_chrome_flags is not None:

[John Grabowski, 2010/03/17 01:18:56]

"if extra chrome flags:"
(no need for "is not None" unless the empty string is a valid value, which I
don't think it is)

        args.append('--extra-chrome-flags=%s' % extra_chrome_flags)
-    pyautolib.PyUITestSuite.__init__(self, args)
+    pyautolib.PyUITestSuite.__init__(self, args, clear_profile, homepage)
     # Figure out path to chromium binaries
     browser_dir = os.path.normpath(os.path.dirname(pyautolib.__file__))
     os.environ['PATH'] = browser_dir + os.pathsep + os.environ['PATH']
     self.Initialize(pyautolib.FilePath(browser_dir))
     unittest.TestCase.__init__(self, methodName)
 
   def __del__(self):
     pyautolib.PyUITestSuite.__del__(self)
 
   def run(self, result=None):
     """The main run method.
 
     We override this method to make calls to the setup steps in PyUITestSuite.
     """
     self.SetUp()     # Open a browser window
     unittest.TestCase.run(self, result)
     self.TearDown()  # Destroy the browser window
 
   def GetBookmarkModel(self):
     """Return the bookmark model as a BookmarkModel object.
 
     This is a snapshot of the bookmark model; it is not a proxy and
     does not get updated as the bookmark model changes.
     """
     return bookmark_model.BookmarkModel(self._GetBookmarksAsJSON())