Fixing omnibox page content search test

functional/omnibox.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 glob
 import os
 import re
 import shutil
 import tempfile
@@ skipping 260 matching lines @@
     # Check if the partial URL would get the bookmark.
     split_url = urlparse.urlsplit(url)
     partial_url = self._GetOmniboxMatchesFor(split_url.scheme, windex=windex)
     self._VerifyHasBookmarkResult(partial_url)
     # Check if the partial title would get the bookmark.
     split_title = title.split()
     search_term = split_title[len(split_title) - 1]
     partial_title = self._GetOmniboxMatchesFor(search_term, windex=windex)
     self._VerifyHasBookmarkResult(partial_title)
 
-  def _GotNewMatches(self, old_matches_len, search_text):
-    """Determines if omnibox has any new matches"""
+  def _GotContentHistory(self, search_text, url):
+    """Determines if omnibox returns a previously visited page for given
+       search text
+    """
     # Omnibox doesn't change results if searching the same text repeatedly.
     # So setting '' in omnibox before the next repeated search.
     self.SetOmniboxText('')
-    new_matches = self._GetOmniboxMatchesFor(search_text)
-    if len(new_matches) > old_matches_len:
-      return True
-    return False
+    matches = self._GetOmniboxMatchesFor(search_text)
+    matches_description = [x for x in matches if x['destination_url'] == url]
+    return 1 == len(matches_description)
 
   def testContentHistory(self):
     """Verify omnibox results when entering page content
 
     Test verifies that visited page shows up in omnibox on entering page
     content.
     """
-    search_text = 'British throne'
-    old_matches = self._GetOmniboxMatchesFor(search_text)
     url = self.GetFileURLForPath(
         os.path.join(self.DataDir(), 'find_in_page', 'largepage.html'))
-    self.AppendTab(pyauto.GURL(url))
-    self.assertTrue(self.WaitUntil(lambda: self._GotNewMatches(len(old_matches),
-                                   search_text), timeout=1))
-    matches = self._GetOmniboxMatchesFor(search_text)
-    matches_description = [x for x in matches if x['destination_url'] == url]
-    self.assertEqual(1, len(matches_description))
+    self.NavigateToURL(url)
+    self.assertTrue(self.WaitUntil(
+        lambda: self._GotContentHistory('British throne', url)))
 
   def _GotHistoryPageOption(self, search_text):
     """Determines if omnibox returns an 'open history page' option for given
        search text"""
     # Omnibox doesn't change results if searching the same text repeatedly.
     # So setting '' in omnibox before the next repeated search.
     self.SetOmniboxText('')
     matches = self._GetOmniboxMatchesFor(search_text)
     matches_description = [x for x in matches if x['type'] ==
                            'open-history-page']
@@ skipping 38 matching lines @@
     self.NavigateToURL(url, 1, 0)
     self._CheckBookmarkResultForVariousInputs(url, title, windex=1)
     self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
     self.assertEqual(3, self.GetBrowserWindowCount())
     self.NavigateToURL(url, 2, 0)
     self._CheckBookmarkResultForVariousInputs(url, title, windex=2)
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()