autotest: Pass suite_name to Handler and use it to determine a provision suite.
BUG=chromium:868528
TEST=None
Change-Id: I3ef91968d15aa23f6af1972ea12cb7f4173537a4
Reviewed-on: https://chromium-review.googlesource.com/1158847
Commit-Queue: Xixuan Wu <xixuan@chromium.org>
Tested-by: Xixuan Wu <xixuan@chromium.org>
Reviewed-by: Prathmesh Prabhu <pprabhu@chromium.org>
diff --git a/venv/skylab_suite/cros_suite.py b/venv/skylab_suite/cros_suite.py
index 641944c..a0e1f06 100644
--- a/venv/skylab_suite/cros_suite.py
+++ b/venv/skylab_suite/cros_suite.py
@@ -43,6 +43,7 @@
SuiteHandlerSpec = collections.namedtuple(
'SuiteHandlerSpec',
[
+ 'suite_name',
'wait',
'suite_id',
'timeout_mins',
@@ -88,6 +89,7 @@
"""
def __init__(self, specs):
+ self._suite_name = specs.suite_name
self._wait = specs.wait
self._timeout_mins = specs.timeout_mins
self._provision_num_required = specs.provision_num_required
@@ -109,7 +111,7 @@
def is_provision(self):
"""Return whether the suite handler is for provision suite."""
- return self._provision_num_required > 0
+ return self._suite_name == 'provision'
def set_suite_id(self, suite_id):
"""Set swarming task id for a suite.