autotest: verify all the DUTs at the beginning of testing push
Verify all the DUTs at the beginning of testing push to ensure that
every DUT is in a good state.
BUG=None
TEST=unittest and also test in testing server.
Change-Id: I77fc502886ae3f020866fd7761fd7790b094e1fe
Reviewed-on: https://chromium-review.googlesource.com/418490
Reviewed-by: Shuqian Zhao <shuqianz@chromium.org>
Reviewed-by: Prathmesh Prabhu <pprabhu@chromium.org>
Tested-by: Shuqian Zhao <shuqianz@chromium.org>
Commit-Queue: Shuqian Zhao <shuqianz@chromium.org>
diff --git a/site_utils/test_push_unittest.py b/site_utils/test_push_unittest.py
index 5b100dc..a517e96 100755
--- a/site_utils/test_push_unittest.py
+++ b/site_utils/test_push_unittest.py
@@ -13,11 +13,13 @@
from autotest_lib.server import site_utils
from autotest_lib.server.cros.dynamic_suite import reporting
# Mock the retry.retry used in the test_push before import it.
-def mock_retry(ExceptionToCheck, timeout_min, exception_to_raise=None):
+def mock_retry(ExceptionToCheck, timeout_min, delay_sec,
+ exception_to_raise=None):
"""A mock retry decorator to use in place of the actual one for testing.
@param ExceptionToCheck: the exception to check.
@param timeout_mins: Amount of time in mins to wait before timing out.
+ @param delay_sec: Amount of time in secs to wait before retry.
@param exception_to_raise: Ignored
"""