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 62 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 86 matching lines @@
       # Don't quote the ':' in drive letter ( say, C: ) on win.
       # Also, replace '\' with '/' as expected in a file:/// url.
       drive, rest = os.path.splitdrive(abs_path)
       quoted_path = drive.upper() + urllib.quote((rest.replace('\\', '/')))
       return 'file:///' + quoted_path
     else:
       quoted_path = urllib.quote(abs_path)
       return 'file://' + quoted_path
 
   @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 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 1177 matching lines @@
       u'tints': {u'buttons': [0.33000000000000002, 0.5, 0.46999999999999997]}}
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = {
       'command': 'GetThemeInfo',
     }
     return self._GetResultFromJSONRequest(cmd_dict)
 
+  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)
@@ skipping 289 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()