autotest: enable metrics from dump_suite_report
BUG=None
TEST=None
Change-Id: I55f687022f20b2002801096783a6433163911cd2
Reviewed-on: https://chromium-review.googlesource.com/510862
Tested-by: Aviv Keshet <akeshet@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 e2f80c6..21ecba9 100755
--- a/site_utils/dump_suite_report.py
+++ b/site_utils/dump_suite_report.py
@@ -14,6 +14,7 @@
from autotest_lib.server.lib import suite_report
from chromite.lib import commandline
from chromite.lib import cros_logging as logging
+from chromite.lib import ts_mon_config
def GetParser():
"""Creates the argparse parser."""
@@ -36,25 +37,27 @@
parser = GetParser()
options = parser.parse_args(argv[1:])
- afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10,
- server=options.afe)
- tko = frontend_wrappers.RetryingTKO(timeout_min=5, delay_sec=10)
+ with ts_mon_config.SetupTsMonGlobalState('dump_suite_report'):
- # 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,
- afe=afe, tko=tko)
- logging.debug('... generated %d entries' % len(suite_entries))
- entries.extend(suite_entries)
+ afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10,
+ server=options.afe)
+ tko = frontend_wrappers.RetryingTKO(timeout_min=5, delay_sec=10)
- # Write all entries as JSON.
- if options.output:
- with open(options.output, 'w') as f:
- suite_report.dump_entries_as_json(entries, f)
- else:
- suite_report.dump_entries_as_json(entries, sys.stdout)
+ # 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,
+ afe=afe, tko=tko)
+ logging.debug('... generated %d entries' % len(suite_entries))
+ entries.extend(suite_entries)
+
+ # Write all entries as JSON.
+ if options.output:
+ with open(options.output, 'w') as f:
+ suite_report.dump_entries_as_json(entries, f)
+ else:
+ suite_report.dump_entries_as_json(entries, sys.stdout)
if __name__ == '__main__':