Add pyauto tests for the New Tab page.

chrome/test/functional/test_utils.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.
 
+import copy
 import os
+import types
 
 import pyauto_functional
 import pyauto_utils
 
 
 """Commonly used functions for PyAuto tests."""
 
 def DownloadFileFromDownloadsDataDir(test, file_name):
   """Download a file from downloads data directory.
 
@@ skipping 17 matching lines @@
   """Delete a file from the downloads directory.
 
   Arg:
     test: derived from pyauto.PyUITest - base class for UI test cases
     file_name: name of file to remove
   """
   downloaded_pkg = os.path.join(test.GetDownloadDirectory().value(),
                                 file_name)
   pyauto_utils.RemovePath(downloaded_pkg)
   pyauto_utils.RemovePath(downloaded_pkg + '.crdownload')
+
+
+def StripUnmatchedKeys(item_to_strip, reference_item):
+  """Returns a copy of 'item_to_strip' where unmatched key-value pairs in
+  every dictionary are removed.
+
+  This will examine each dictionary in 'item_to_strip' recursively, and will
+  remove keys that are not found in the corresponding dictionary in
+  'reference_item'. This is useful for testing equality of a subset of data.
+
+  Items may contain dictionaries, lists, or primitives, but only corresponding
+  dictionaries will be stripped. A corresponding entry is one which is found
+  in the same index in the corresponding parent array or at the same key in the
+  corresponding parent dictionary.
+
+  Arg:
+    item_to_strip: item to copy and remove all unmatched key-value pairs
+    reference_item: item that serves as a reference for which keys-value pairs
+                    to strip from 'item_to_strip'
+
+  Returns:
+    a copy of 'item_to_strip' where all key-value pairs that do not have a
+    matching key in 'reference_item' are removed
+
+  Example:
+    item_to_strip = {'tabs': 3,
+                     'time': 5908}
+    reference_item = {'tabs': 2}
+    StripUnmatchedKeys(item_to_strip, reference_item) will return {'tabs': 3}
+  """
+  def StripList(list1, list2):
+    return_list = copy.deepcopy(list2)
+    for i in range(min(len(list1), len(list2))):
+      return_list[i] = StripUnmatchedKeys(list1[i], list2[i])
+    return return_list
+
+  def StripDict(dict1, dict2):
+    return_dict = {}
+    for key in dict1:
+      if key in dict2:
+        return_dict[key] = StripUnmatchedKeys(dict1[key], dict2[key])
+    return return_dict
+
+  item_to_strip_type = type(item_to_strip)
+  if item_to_strip_type is type(reference_item):
+    if item_to_strip_type is types.ListType:
+      return StripList(item_to_strip, reference_item)
+    elif item_to_strip_type is types.DictType:
+      return StripDict(item_to_strip, reference_item)
+  return copy.deepcopy(item_to_strip)