Add pyauto hook for getting and manipulating the data underneath the NTP.

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 64 matching lines @@
   print >>sys.stderr, "Could not locate built libraries. Did you build?"
   # Mac requires python2.5 even when not the default 'python' (e.g. 10.6)
   if 'darwin' == sys.platform and sys.version_info[:2] != (2,5):
     print  >>sys.stderr, "*\n* Perhaps use 'python2.5', not 'python' ?\n*"
   raise
 
 # Should go after sys.path is set appropriately
 import bookmark_model
 import download_info
 import history_info
+import ntp_model
 import omnibox_info
 import plugins_info
 import prefs_info
 from pyauto_errors import JSONInterfaceError
 import simplejson as json  # found in third_party
 
 
 class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
   """Base class for UI Test Cases in Python.
 
@@ skipping 95 matching lines @@
   @staticmethod
   def GetFileURLForDataPath(relative_path):
     """Get file:// url for the given path relative to the chrome test data dir.
 
     Also quotes the url using urllib.quote().
     """
     return PyUITest.GetFileURLForPath(
         os.path.join(PyUITest.DataDir(), relative_path))
 
   @staticmethod
+  def GetFileURLForDataPath(relative_path):

[Nirnimesh, 2010/11/16 22:25:05]

this function is duplicate

+    """Get file:// url for the given path relative to the chrome test data dir.
+
+    Also quotes the url using urllib.quote().
+    """
+    return PyUITest.GetFileURLForPath(
+        os.path.join(PyUITest.DataDir(), relative_path))
+
+  @staticmethod
   def IsMac():
     """Are we on Mac?"""
     return 'darwin' == sys.platform
 
   @staticmethod
   def IsLinux():
     """Are we on Linux? ChromeOS is linux too."""
     return 'linux2' == sys.platform
 
   @staticmethod
@@ skipping 1559 matching lines @@
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = {
       'command': 'DisableSyncForDatatypes',
       'datatypes': datatypes,
     }
     return self._GetResultFromJSONRequest(cmd_dict)['success']
 
+  def GetNTPModel(self):
+    """Returns the NTPModel."""
+    return ntp_model.NTPModel(self)
+
+
 class PyUITestSuite(pyautolib.PyUITestSuiteBase, unittest.TestSuite):
   """Base TestSuite for PyAuto UI tests."""
 
   def __init__(self, args):
     pyautolib.PyUITestSuiteBase.__init__(self, args)
 
     # Figure out path to chromium binaries
     browser_dir = os.path.normpath(os.path.dirname(pyautolib.__file__))
     logging.debug('Loading pyauto libs from %s', browser_dir)
     self.Initialize(pyautolib.FilePath(browser_dir))
@@ skipping 288 matching lines @@
     if self._options.verbose:
       verbosity = 2
     result = PyAutoTextTestRuner(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()