| # Copyright 2018 The ChromiumOS Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Tast-specific logics of TKO parser.""" |
| |
| import json |
| import os |
| |
| import common |
| |
| from autotest_lib.client.common_lib import utils |
| |
| # Name of the Autotest test case that runs Tast tests. |
| _TAST_AUTOTEST_NAME = 'tast' |
| |
| # Prefix added to Tast test names when writing their results to TKO. |
| # This should match with _TEST_NAME_PREFIX in server/site_tests/tast/tast.py. |
| _TAST_TEST_NAME_PREFIX = 'tast.' |
| |
| |
| def is_tast_test(test_name): |
| """Checks if a test is a Tast test.""" |
| return test_name.startswith(_TAST_TEST_NAME_PREFIX) |
| |
| |
| def load_tast_test_aux_results(job, test_name): |
| """Loads auxiliary results of a Tast test. |
| |
| @param job: A job object. |
| @param test_name: The name of the test. |
| @return (attributes, perf_values) where |
| attributes: A str-to-str dict of attribute keyvals |
| perf_values: A dict loaded from a chromeperf JSON |
| """ |
| assert is_tast_test(test_name) |
| |
| test_dir = os.path.join(job.dir, _TAST_AUTOTEST_NAME) |
| |
| case_name = test_name[len(_TAST_TEST_NAME_PREFIX):] |
| case_dir = os.path.join(test_dir, 'results', 'tests', case_name) |
| |
| # Load attribute keyvals. |
| attributes_path = os.path.join(test_dir, 'keyval') |
| if os.path.exists(attributes_path): |
| attributes = utils.read_keyval(attributes_path) |
| else: |
| attributes = {} |
| |
| # Load a chromeperf JSON. |
| perf_values_path = os.path.join(case_dir, 'results-chart.json') |
| if os.path.exists(perf_values_path): |
| with open(perf_values_path) as fp: |
| perf_values = json.load(fp) |
| else: |
| perf_values = {} |
| |
| return attributes, perf_values |