#!/usr/bin/python
|
|
"""Unit tests for the perf_uploader.py module.
|
|
"""
|
|
import json, unittest
|
|
import common
|
from autotest_lib.tko import models as tko_models
|
from autotest_lib.tko.perf_upload import perf_uploader
|
|
|
class test_aggregate_iterations(unittest.TestCase):
|
"""Tests for the aggregate_iterations function."""
|
|
_PERF_ITERATION_DATA = {
|
'1': [
|
{
|
'description': 'metric1',
|
'value': 1,
|
'stddev': 0.0,
|
'units': 'units1',
|
'higher_is_better': True,
|
'graph': None
|
},
|
{
|
'description': 'metric2',
|
'value': 10,
|
'stddev': 0.0,
|
'units': 'units2',
|
'higher_is_better': True,
|
'graph': 'graph1',
|
},
|
{
|
'description': 'metric2',
|
'value': 100,
|
'stddev': 1.7,
|
'units': 'units3',
|
'higher_is_better': False,
|
'graph': 'graph2',
|
}
|
],
|
'2': [
|
{
|
'description': 'metric1',
|
'value': 2,
|
'stddev': 0.0,
|
'units': 'units1',
|
'higher_is_better': True,
|
'graph': None,
|
},
|
{
|
'description': 'metric2',
|
'value': 20,
|
'stddev': 0.0,
|
'units': 'units2',
|
'higher_is_better': True,
|
'graph': 'graph1',
|
},
|
{
|
'description': 'metric2',
|
'value': 200,
|
'stddev': 21.2,
|
'units': 'units3',
|
'higher_is_better': False,
|
'graph': 'graph2',
|
}
|
],
|
}
|
|
|
def setUp(self):
|
"""Sets up for each test case."""
|
self._perf_values = []
|
for iter_num, iter_data in self._PERF_ITERATION_DATA.iteritems():
|
self._perf_values.append(
|
tko_models.perf_value_iteration(iter_num, iter_data))
|
|
|
|
class test_json_config_file_sanity(unittest.TestCase):
|
"""Sanity tests for the JSON-formatted presentation config file."""
|
|
def test_parse_json(self):
|
"""Verifies _parse_config_file function."""
|
perf_uploader._parse_config_file(
|
perf_uploader._PRESENTATION_CONFIG_FILE)
|
|
|
def test_proper_json(self):
|
"""Verifies the file can be parsed as proper JSON."""
|
try:
|
with open(perf_uploader._PRESENTATION_CONFIG_FILE, 'r') as fp:
|
json.load(fp)
|
except:
|
self.fail('Presentation config file could not be parsed as JSON.')
|
|
|
def test_unique_test_names(self):
|
"""Verifies that each test name appears only once in the JSON file."""
|
json_obj = []
|
try:
|
with open(perf_uploader._PRESENTATION_CONFIG_FILE, 'r') as fp:
|
json_obj = json.load(fp)
|
except:
|
self.fail('Presentation config file could not be parsed as JSON.')
|
|
name_set = set([x['autotest_name'] for x in json_obj])
|
self.assertEqual(len(name_set), len(json_obj),
|
msg='Autotest names not unique in the JSON file.')
|
|
|
def test_required_master_name(self):
|
"""Verifies that master name must be specified."""
|
json_obj = []
|
try:
|
with open(perf_uploader._PRESENTATION_CONFIG_FILE, 'r') as fp:
|
json_obj = json.load(fp)
|
except:
|
self.fail('Presentation config file could not be parsed as JSON.')
|
|
for entry in json_obj:
|
if not 'master_name' in entry:
|
self.fail('Missing master field for test %s.' %
|
entry['autotest_name'])
|
|
|
class test_gather_presentation_info(unittest.TestCase):
|
"""Tests for the gather_presentation_info function."""
|
|
_PRESENT_INFO = {
|
'test_name': {
|
'master_name': 'new_master_name',
|
'dashboard_test_name': 'new_test_name',
|
}
|
}
|
|
_PRESENT_INFO_MISSING_MASTER = {
|
'test_name': {
|
'dashboard_test_name': 'new_test_name',
|
}
|
}
|
|
_PRESENT_INFO_REGEX = {
|
'test_name.*': {
|
'master_name': 'new_master_name',
|
'dashboard_test_name': 'new_test_name',
|
}
|
}
|
|
def test_test_name_regex_specified(self):
|
"""Verifies gathers presentation info for regex search correctly"""
|
for test_name in ['test_name.arm.7.1', 'test_name.x86.7.1']:
|
result = perf_uploader._gather_presentation_info(
|
self._PRESENT_INFO, 'test_name_P')
|
self.assertTrue(
|
all([key in result for key in
|
['test_name', 'master_name']]),
|
msg='Unexpected keys in resulting dictionary: %s' % result)
|
self.assertEqual(result['master_name'], 'new_master_name',
|
msg='Unexpected "master_name" value: %s' %
|
result['master_name'])
|
self.assertEqual(result['test_name'], 'new_test_name',
|
msg='Unexpected "test_name" value: %s' %
|
result['test_name'])
|
|
def test_test_name_specified(self):
|
"""Verifies gathers presentation info correctly."""
|
result = perf_uploader._gather_presentation_info(
|
self._PRESENT_INFO, 'test_name')
|
self.assertTrue(
|
all([key in result for key in
|
['test_name', 'master_name']]),
|
msg='Unexpected keys in resulting dictionary: %s' % result)
|
self.assertEqual(result['master_name'], 'new_master_name',
|
msg='Unexpected "master_name" value: %s' %
|
result['master_name'])
|
self.assertEqual(result['test_name'], 'new_test_name',
|
msg='Unexpected "test_name" value: %s' %
|
result['test_name'])
|
|
|
def test_test_name_not_specified(self):
|
"""Verifies exception raised if test is not there."""
|
self.assertRaises(
|
perf_uploader.PerfUploadingError,
|
perf_uploader._gather_presentation_info,
|
self._PRESENT_INFO, 'other_test_name')
|
|
|
def test_master_not_specified(self):
|
"""Verifies exception raised if master is not there."""
|
self.assertRaises(
|
perf_uploader.PerfUploadingError,
|
perf_uploader._gather_presentation_info,
|
self._PRESENT_INFO_MISSING_MASTER, 'test_name')
|
|
|
class test_get_id_from_version(unittest.TestCase):
|
"""Tests for the _get_id_from_version function."""
|
|
def test_correctly_formatted_versions(self):
|
"""Verifies that the expected ID is returned when input is OK."""
|
chrome_version = '27.0.1452.2'
|
cros_version = '27.3906.0.0'
|
# 1452.2 + 3906.0.0
|
# --> 01452 + 002 + 03906 + 000 + 00
|
# --> 14520020390600000
|
self.assertEqual(
|
14520020390600000,
|
perf_uploader._get_id_from_version(
|
chrome_version, cros_version))
|
|
chrome_version = '25.10.1000.0'
|
cros_version = '25.1200.0.0'
|
# 1000.0 + 1200.0.0
|
# --> 01000 + 000 + 01200 + 000 + 00
|
# --> 10000000120000000
|
self.assertEqual(
|
10000000120000000,
|
perf_uploader._get_id_from_version(
|
chrome_version, cros_version))
|
|
def test_returns_none_when_given_invalid_input(self):
|
"""Checks the return value when invalid input is given."""
|
chrome_version = '27.0'
|
cros_version = '27.3906.0.0'
|
self.assertIsNone(perf_uploader._get_id_from_version(
|
chrome_version, cros_version))
|
|
|
class test_get_version_numbers(unittest.TestCase):
|
"""Tests for the _get_version_numbers function."""
|
|
def test_with_valid_versions(self):
|
"""Checks the version numbers used when data is formatted as expected."""
|
self.assertEqual(
|
('34.5678.9.0', '34.5.678.9'),
|
perf_uploader._get_version_numbers(
|
{
|
'CHROME_VERSION': '34.5.678.9',
|
'CHROMEOS_RELEASE_VERSION': '5678.9.0',
|
}))
|
|
def test_with_missing_version_raises_error(self):
|
"""Checks that an error is raised when a version is missing."""
|
with self.assertRaises(perf_uploader.PerfUploadingError):
|
perf_uploader._get_version_numbers(
|
{
|
'CHROMEOS_RELEASE_VERSION': '5678.9.0',
|
})
|
|
def test_with_unexpected_version_format_raises_error(self):
|
"""Checks that an error is raised when there's a rN suffix."""
|
with self.assertRaises(perf_uploader.PerfUploadingError):
|
perf_uploader._get_version_numbers(
|
{
|
'CHROME_VERSION': '34.5.678.9',
|
'CHROMEOS_RELEASE_VERSION': '5678.9.0r1',
|
})
|
|
def test_with_valid_release_milestone(self):
|
"""Checks the version numbers used when data is formatted as expected."""
|
self.assertEqual(
|
('54.5678.9.0', '34.5.678.9'),
|
perf_uploader._get_version_numbers(
|
{
|
'CHROME_VERSION': '34.5.678.9',
|
'CHROMEOS_RELEASE_VERSION': '5678.9.0',
|
'CHROMEOS_RELEASE_CHROME_MILESTONE': '54',
|
}))
|
|
|
class test_format_for_upload(unittest.TestCase):
|
"""Tests for the format_for_upload function."""
|
|
_PERF_DATA = {
|
"charts": {
|
"metric1": {
|
"summary": {
|
"improvement_direction": "down",
|
"type": "scalar",
|
"units": "msec",
|
"value": 2.7,
|
}
|
},
|
"metric2": {
|
"summary": {
|
"improvement_direction": "up",
|
"type": "scalar",
|
"units": "frames_per_sec",
|
"value": 101.35,
|
}
|
}
|
},
|
}
|
_PRESENT_INFO = {
|
'master_name': 'new_master_name',
|
'test_name': 'new_test_name',
|
}
|
|
def setUp(self):
|
self._perf_data = self._PERF_DATA
|
|
def _verify_result_string(self, actual_result, expected_result):
|
"""Verifies a JSON string matches the expected result.
|
|
This function compares JSON objects rather than strings, because of
|
possible floating-point values that need to be compared using
|
assertAlmostEqual().
|
|
@param actual_result: The candidate JSON string.
|
@param expected_result: The reference JSON string that the candidate
|
must match.
|
|
"""
|
actual = json.loads(actual_result)
|
expected = json.loads(expected_result)
|
|
def ordered(obj):
|
if isinstance(obj, dict):
|
return sorted((k, ordered(v)) for k, v in obj.items())
|
if isinstance(obj, list):
|
return sorted(ordered(x) for x in obj)
|
else:
|
return obj
|
fail_msg = 'Unexpected result string: %s' % actual_result
|
self.assertEqual(ordered(expected), ordered(actual), msg=fail_msg)
|
|
|
def test_format_for_upload(self):
|
"""Verifies format_for_upload generates correct json data."""
|
result = perf_uploader._format_for_upload(
|
'platform', '25.1200.0.0', '25.10.1000.0', 'WINKY E2A-F2K-Q35',
|
'i7', 'test_machine', self._perf_data, self._PRESENT_INFO,
|
'52926644-username/hostname')
|
expected_result_string = (
|
'{"versions": {'
|
'"cros_version": "25.1200.0.0",'
|
'"chrome_version": "25.10.1000.0"'
|
'},'
|
'"point_id": 10000000120000000,'
|
'"bot": "cros-platform-i7",'
|
'"chart_data": {'
|
'"charts": {'
|
'"metric2": {'
|
'"summary": {'
|
'"units": "frames_per_sec",'
|
'"type": "scalar",'
|
'"value": 101.35,'
|
'"improvement_direction": "up"'
|
'}'
|
'},'
|
'"metric1": {'
|
'"summary": {'
|
'"units": "msec",'
|
'"type": "scalar",'
|
'"value": 2.7,'
|
'"improvement_direction": "down"}'
|
'}'
|
'}'
|
'},'
|
'"master": "new_master_name",'
|
'"supplemental": {'
|
'"hardware_identifier": "WINKY E2A-F2K-Q35",'
|
'"jobname": "52926644-username/hostname",'
|
'"hardware_hostname": "test_machine",'
|
'"default_rev": "r_cros_version",'
|
'"variant_name": "i7"}'
|
'}')
|
self._verify_result_string(result['data'], expected_result_string)
|
|
|
if __name__ == '__main__':
|
unittest.main()
|
