Fixing search_engines.SearchEnginesTest.testDiscoverSearchEngine

functional/search_engines.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 re
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
+import test_utils
 
 
 class SearchEnginesTest(pyauto.PyUITest):
   """TestCase for Search Engines."""
 
   _localhost_prefix = 'http://localhost:1000/'
 
   def _GetSearchEngineWithKeyword(self, keyword):
     """Get search engine info and return an element that matches keyword.
 
@@ skipping 22 matching lines @@
 
   def testDiscoverSearchEngine(self):
     """Test that chrome discovers youtube search engine after searching."""
     # Take a snapshot of current search engine info.
     info = self.GetSearchEngineInfo()
     youtube = self._GetSearchEngineWithKeyword('youtube.com')
     self.assertFalse(youtube)
     # Use omnibox to invoke search engine discovery.
     # Navigating using NavigateToURL does not currently invoke this logic.
     self.SetOmniboxText('http://www.youtube.com')
-    self.OmniboxAcceptInput()
+    # Due to slow navigation to youtube.com on Leopard test machines, waiting
+    # here 1 min (max).
+    test_utils.CallFunctionWithNewTimeout(self, 1 * 60 * 1000,
+                                          self.OmniboxAcceptInput)
     def InfoUpdated(old_info):
       new_info = self.GetSearchEngineInfo()
       if len(new_info) > len(old_info):
         return True
       return False
     self.WaitUntil(lambda: InfoUpdated(info))
     youtube = self._GetSearchEngineWithKeyword('youtube.com')
     self.assertTrue(youtube)
     self.assertTrue(re.search('youtube', youtube['short_name'],
                               re.IGNORECASE))
@@ skipping 113 matching lines @@
     self.assertEqual(self._localhost_prefix + '?edited=true&q=foobar',
                      self.GetActiveTabURL().spec())
     # Delete a search engine.
     self.assertTrue(self._GetSearchEngineWithKeyword('testspecial@!%^*#.com'))
     self.DeleteSearchEngine('testspecial@!%^*#.com')
     self.assertFalse(self._GetSearchEngineWithKeyword('testspecial@!%^*#.com'))
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()