old_contents: #!/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 pyauto_functional  # Must be imported before pyauto
  import pyauto
  
  
  class InstantTest(pyauto.PyUITest):
    """TestCase for Omnibox Instant feature."""
  
    def setUp(self):
      pyauto.PyUITest.setUp(self)
      self.SetPrefs(pyauto.kInstantEnabled, True)
  
    def _DoneLoading(self):
      info = self.GetInstantInfo()
      return info.get('current') and not info.get('loading')
  
    def testInstantNavigation(self):
      """Test that instant navigates based on omnibox input."""
      self.SetOmniboxText('google.com')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue(location.startswith('http://www.google.com'))
  
      self.SetOmniboxText('bing.com')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue(location.startswith('http://www.bing.com'))
  
      # Initiate instant search (at default google.com).
      self.SetOmniboxText('instant')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue(location.startswith('http://www.google.com'))
  
  
  if __name__ == '__main__':
    pyauto_functional.Main()
  
new_contents: #!/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 pyauto_functional  # Must be imported before pyauto
  import pyauto
  
  
  class InstantTest(pyauto.PyUITest):
    """TestCase for Omnibox Instant feature."""
  
    def setUp(self):
      pyauto.PyUITest.setUp(self)
      self.SetPrefs(pyauto.kInstantEnabled, True)
  
    def _DoneLoading(self):
      info = self.GetInstantInfo()
      return info.get('current') and not info.get('loading')
  
    def testInstantNavigation(self):
      """Test that instant navigates based on omnibox input."""
      self.SetOmniboxText('google.com')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue('google.com' in location)
  
      self.SetOmniboxText('bing.com')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue('bing.com' in location)
  
      # Initiate instant search (at default google.com).
      self.SetOmniboxText('instant')
      self.assertTrue(self.WaitUntil(self._DoneLoading))
      location = self.GetInstantInfo()['location']
      self.assertTrue('google.com' in location)
  
  
  if __name__ == '__main__':
    pyauto_functional.Main()
  
subject: Fix instant pyauto test