blob: 223bf1351c5a8826a4255abe555b07c34efb828f [file] [log] [blame]
# Lint as: python2, python3
# Copyright 2015 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import mox
import unittest
import common
from autotest_lib.frontend import setup_django_lite_environment
from autotest_lib.server import frontend
from autotest_lib.server import site_utils
from autotest_lib.server.cros.dynamic_suite import tools
from autotest_lib.server.cros.dynamic_suite import suite_common
import six
class SiteUtilsUnittests(mox.MoxTestBase):
"""Test functions in site_utils.py"""
def testParseJobName(self):
"""Test method parse_job_name.
"""
trybot_paladin_build = 'trybot-lumpy-paladin/R27-3837.0.0-b123'
trybot_release_build = 'trybot-lumpy-release/R27-3837.0.0-b456'
release_build = 'lumpy-release/R27-3837.0.0'
paladin_build = 'lumpy-paladin/R27-3878.0.0-rc7'
brillo_build = 'git_mnc-brillo-dev/lumpy-eng/1234'
chrome_pfq_build = 'lumpy-chrome-pfq/R27-3837.0.0'
chromium_pfq_build = 'lumpy-chromium-pfq/R27-3837.0.0'
builds = [trybot_paladin_build, trybot_release_build, release_build,
paladin_build, brillo_build, chrome_pfq_build,
chromium_pfq_build]
test_name = 'login_LoginSuccess'
board = 'lumpy'
suite = 'bvt'
for build in builds:
expected_info = {'board': board,
'suite': suite,
'build': build}
build_parts = build.split('/')
if len(build_parts) == 2:
expected_info['build_version'] = build_parts[1]
else:
expected_info['build_version'] = build_parts[2]
suite_job_name = ('%s-%s' %
(build, suite_common.canonicalize_suite_name(suite)))
info = site_utils.parse_job_name(suite_job_name)
self.assertEqual(info, expected_info, '%s failed to be parsed to '
'%s' % (suite_job_name, expected_info))
test_job_name = tools.create_job_name(build, suite, test_name)
info = site_utils.parse_job_name(test_job_name)
self.assertEqual(info, expected_info, '%s failed to be parsed to '
'%s' % (test_job_name, expected_info))
def testGetViewsFromTko(self):
"""Test method get_test_views_from_tko
"""
test_results = [
('dummy_Pass', 'GOOD'),
('dummy_Fail.RetrySuccess', 'GOOD'),
('dummy_Fail.RetrySuccess', 'FAIL'),
('dummy_Fail.Fail', 'FAIL'),
('dummy_Fail.Fail', 'FAIL'),
]
expected_test_views = {
'dummy_Pass': ['GOOD'],
'dummy_Fail.RetrySuccess': ['FAIL', 'GOOD'],
'dummy_Fail.Fail': ['FAIL', 'FAIL'],
}
self.mox.UnsetStubs()
tko = self.mox.CreateMock(frontend.TKO)
tko.run('get_detailed_test_views', afe_job_id=0).AndReturn(
[{'test_name':r[0], 'status':r[1]} for r in test_results])
self.mox.ReplayAll()
test_views = site_utils.get_test_views_from_tko(0, tko)
self.mox.VerifyAll()
self.assertEqual(sorted(test_views.keys()),
sorted(expected_test_views.keys()),
'Test list %s does not match expected test list %s.' %
(sorted(test_views.keys()),
sorted(expected_test_views.keys())))
for test_name, test_status_list in six.iteritems(test_views):
self.assertEqual(sorted(test_status_list),
sorted(expected_test_views[test_name]),
'For test %s the status list %s does not match '
'expected status list %s.' %
(test_name,
sorted(test_status_list),
sorted(expected_test_views[test_name])))
if __name__ == '__main__':
unittest.main()