[pyauto] Fix ActionTimeoutChanger

chrome/test/functional/test_pyauto.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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 time
 import unittest
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
 import pyauto_errors
 
 
 class PyAutoTest(pyauto.PyUITest):
   """Test functionality of the PyAuto framework."""
 
@@ skipping 27 matching lines @@
         lambda: self.FindInPage('some text', windex=1))  # invalid window
 
   def testJSONInterfaceTimeout(self):
     """Verify that an exception is raised when the JSON interface times out."""
     self.ClearEventQueue()
     self.AddDomEventObserver('foo')
     self.assertRaises(
         pyauto_errors.JSONInterfaceError,
         lambda: self.GetNextEvent(timeout=2000))  # event queue is empty
 
+  def testActionTimeoutChanger(self):
+    """Verify that ActionTimeoutChanger works."""
+    new_timeout = 1000  # 1 sec
+    changer = pyauto.PyUITest.ActionTimeoutChanger(self, new_timeout)
+    self.assertEqual(self.action_timeout_ms(), new_timeout)
+
+    # Verify the amount of time taken for automation timeout
+    then = time.time()
+    self.assertRaises(
+        pyauto_errors.JSONInterfaceError,
+        lambda: self.ExecuteJavascript('invalid js should timeout'))
+    elapsed = time.time() - then
+    self.assertTrue(elapsed < new_timeout / 1000.0 + 2,  # margin of 2 secs
+        msg='ActionTimeoutChanger did not work. '
+            'Automation timeout took %f secs' % elapsed)
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()