Add 2 tests for history.

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 265 matching lines @@
     if ret_dict.has_key('error'):
       raise JSONInterfaceError(ret_dict['error'])
 
   def WaitForAllDownloadsToComplete(self):
     """Wait for all downloads to complete."""
     # Implementation detail: uses the generic "JSON command" model
     # (experimental)
     self._SendJSONRequest(0, json.dumps({'command':
                                          'WaitForAllDownloadsToComplete'}))
 
+  def DownloadAndWaitForStart(self, file_url):
+    """Trigger download for the given url and wait for downloads to start.
+
+    It waits for download by looking at the download info from Chrome, so
+    anything which isn't registered by the history service won't be noticed.
+    This is not thread-safe, but it's fine to call this method to start
+    downloading multiple files in parallel. That is after starting a
+    download, it's fine to start another one even if the first one hasn't
+    completed.
+    """
+    num_downloads = len(self.GetDownloadsInfo().Downloads())
+    self.NavigateToURL(file_url)  # Trigger download.
+    # It might take a while for the download to kick in, hold on until then.
+    self.assertTrue(self.WaitUntil(
+        lambda: len(self.GetDownloadsInfo().Downloads()) == num_downloads + 1))
+
   def GetHistoryInfo(self, search_text=''):
     """Return info about browsing history.
 
     Args:
       search_text: the string to search in history.  Defaults to empty string
                    which means that all history would be returned. This is
                    functionally equivalent to searching for a text in the
                    chrome://history UI. So partial matches work too.
                    When non-empty, the history items returned will contain a
                    "snippet" field corresponding to the snippet visible in
@@ skipping 174 matching lines @@
     if self._options.verbose:
       verbosity = 2
     result = unittest.TextTestRunner(verbosity=verbosity).run(pyauto_suite)
     del loaded_tests  # Need to destroy test cases before the suite
     del pyauto_suite
     sys.exit(not result.wasSuccessful())
 
 
 if __name__ == '__main__':
   Main()