autotest: Allow dump_suite_report to specife AFE host.
BUG=chromium:641093
TEST=dump_suite_report.py
Change-Id: Ibecc536b5026e5e892cd701b5cd3854d1ad35097
Reviewed-on: https://chromium-review.googlesource.com/446578
Commit-Ready: David Riley <davidriley@chromium.org>
Tested-by: David Riley <davidriley@chromium.org>
Reviewed-by: Shuqian Zhao <shuqianz@chromium.org>
diff --git a/site_utils/dump_suite_report.py b/site_utils/dump_suite_report.py
index 24176f7..9a6da17 100755
--- a/site_utils/dump_suite_report.py
+++ b/site_utils/dump_suite_report.py
@@ -10,6 +10,7 @@
import common
import sys
+from autotest_lib.server import frontend
from autotest_lib.server.lib import suite_report
from chromite.lib import commandline
from chromite.lib import cros_logging as logging
@@ -21,6 +22,8 @@
help='Suite job ids to dump')
parser.add_argument('--output', '-o', type=str, action='store',
help='Path to write JSON file to')
+ parser.add_argument('--afe', type=str, action='store',
+ help='AFE server to connect to')
return parser
@@ -33,11 +36,16 @@
parser = GetParser()
options = parser.parse_args(argv[1:])
+ afe = None
+ if options.afe:
+ afe = frontend.AFE(server=options.afe)
+
# Look up and generate entries for all jobs.
entries = []
for suite_job_id in options.job_ids:
logging.debug('Suite job %s:' % suite_job_id)
- suite_entries = suite_report.generate_suite_report(suite_job_id)
+ suite_entries = suite_report.generate_suite_report(suite_job_id,
+ afe=afe)
logging.debug('... generated %d entries' % len(suite_entries))
entries.extend(suite_entries)