commit 2bc238a157cc26b956064b142065fd026bb28354
author David James <davidjames@chromium.org> Tue Dec 03 15:21:36 2013 -0800
committer chrome-internal-fetch <chrome-internal-fetch@google.com> Thu Dec 05 05:28:12 2013 +0000
tree ffef67061a1c61b4297ab340414f4e0b5aed4ffb
parent 45f277e1b0ef33d59bc959c64f2f3726234c1d39

crostestutils: Move Timeout* functions to chromite.lib.timeout_util.

BUG=chromium:323675
TEST=pylint

CQ-DEPEND=CL:Ida5b3b5be04a3b891da0f0c06db832d3d01be972

Change-Id: Ife015dfd0f08eefbe8c9e73112dedca5fbe78f61
Reviewed-on: https://chromium-review.googlesource.com/178733
Reviewed-by: Aviv Keshet <akeshet@chromium.org>
Tested-by: David James <davidjames@chromium.org>
Commit-Queue: David James <davidjames@chromium.org>

au_test_harness/cros_au_test_harness.py

diff --git a/au_test_harness/cros_au_test_harness.py b/au_test_harness/cros_au_test_harness.py
index df6ab93..0745267 100755
--- a/au_test_harness/cros_au_test_harness.py
+++ b/au_test_harness/cros_au_test_harness.py
@@ -28,6 +28,7 @@
 from chromite.lib import cros_build_lib
 from chromite.lib import parallel
 from chromite.lib import sudo
+from chromite.lib import timeout_util
 from crostestutils.au_test_harness import au_test
 from crostestutils.au_test_harness import au_worker
 from crostestutils.lib import dev_server_wrapper
@@ -66,7 +67,7 @@
 
     If the test suite fails, raise a BackgroundFailure.
     """
-    with cros_build_lib.SubCommandTimeout(constants.MAX_TIMEOUT_SECONDS):
+    with timeout_util.Timeout(constants.MAX_TIMEOUT_SECONDS):
       test_result = super(_LessBacktracingTestRunner, self).run(*args, **kwargs)
       if test_result is None or not test_result.wasSuccessful():
         msg = 'Test harness failed. See logs for details.'

generate_test_payloads/cros_generate_test_payloads.py

diff --git a/generate_test_payloads/cros_generate_test_payloads.py b/generate_test_payloads/cros_generate_test_payloads.py
index faf782e..4703ef1 100755
--- a/generate_test_payloads/cros_generate_test_payloads.py
+++ b/generate_test_payloads/cros_generate_test_payloads.py
@@ -38,6 +38,7 @@
 from chromite.lib import osutils
 from chromite.lib import parallel
 from chromite.lib import sudo
+from chromite.lib import timeout_util
 from crostestutils.au_test_harness import cros_au_test_harness
 from crostestutils.generate_test_payloads import payload_generation_exception
 from crostestutils.lib import dev_server_wrapper
@@ -255,12 +256,12 @@
 
       logging.info(debug_message)
       try:
-        with cros_build_lib.SubCommandTimeout(constants.MAX_TIMEOUT_SECONDS):
+        with timeout_util.Timeout(constants.MAX_TIMEOUT_SECONDS):
           cros_build_lib.SudoRunCommand(command, log_stdout_to_file=log_file,
                                         combine_stdout_stderr=True,
                                         enter_chroot=True, print_cmd=False,
                                         cwd=constants.SOURCE_ROOT)
-      except (cros_build_lib.TimeoutError, cros_build_lib.RunCommandError):
+      except (timeout_util.TimeoutError, cros_build_lib.RunCommandError):
         # Print output first, then re-raise the exception.
         if os.path.isfile(log_file):
           logging.error(osutils.ReadFile(log_file))