1# Copyright 2015 The Chromium OS Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import mox 6import unittest 7 8import common 9from autotest_lib.frontend import setup_django_lite_environment 10from autotest_lib.server import frontend 11from autotest_lib.server import site_utils 12from autotest_lib.server.cros.dynamic_suite import tools 13from autotest_lib.server.cros.dynamic_suite import suite_common 14 15 16class SiteUtilsUnittests(mox.MoxTestBase): 17 """Test functions in site_utils.py""" 18 19 def testParseJobName(self): 20 """Test method parse_job_name. 21 """ 22 trybot_paladin_build = 'trybot-lumpy-paladin/R27-3837.0.0-b123' 23 trybot_release_build = 'trybot-lumpy-release/R27-3837.0.0-b456' 24 release_build = 'lumpy-release/R27-3837.0.0' 25 paladin_build = 'lumpy-paladin/R27-3878.0.0-rc7' 26 brillo_build = 'git_mnc-brillo-dev/lumpy-eng/1234' 27 chrome_pfq_build = 'lumpy-chrome-pfq/R27-3837.0.0' 28 chromium_pfq_build = 'lumpy-chromium-pfq/R27-3837.0.0' 29 30 builds = [trybot_paladin_build, trybot_release_build, release_build, 31 paladin_build, brillo_build, chrome_pfq_build, 32 chromium_pfq_build] 33 test_name = 'login_LoginSuccess' 34 board = 'lumpy' 35 suite = 'bvt' 36 for build in builds: 37 expected_info = {'board': board, 38 'suite': suite, 39 'build': build} 40 build_parts = build.split('/') 41 if len(build_parts) == 2: 42 expected_info['build_version'] = build_parts[1] 43 else: 44 expected_info['build_version'] = build_parts[2] 45 suite_job_name = ('%s-%s' % 46 (build, suite_common.canonicalize_suite_name(suite))) 47 info = site_utils.parse_job_name(suite_job_name) 48 self.assertEqual(info, expected_info, '%s failed to be parsed to ' 49 '%s' % (suite_job_name, expected_info)) 50 test_job_name = tools.create_job_name(build, suite, test_name) 51 info = site_utils.parse_job_name(test_job_name) 52 self.assertEqual(info, expected_info, '%s failed to be parsed to ' 53 '%s' % (test_job_name, expected_info)) 54 55 56 def testGetViewsFromTko(self): 57 """Test method get_test_views_from_tko 58 """ 59 test_results = [ 60 ('dummy_Pass', 'GOOD'), 61 ('dummy_Fail.RetrySuccess', 'GOOD'), 62 ('dummy_Fail.RetrySuccess', 'FAIL'), 63 ('dummy_Fail.Fail', 'FAIL'), 64 ('dummy_Fail.Fail', 'FAIL'), 65 ] 66 67 expected_test_views = { 68 'dummy_Pass': ['GOOD'], 69 'dummy_Fail.RetrySuccess': ['FAIL', 'GOOD'], 70 'dummy_Fail.Fail': ['FAIL', 'FAIL'], 71 } 72 73 self.mox.UnsetStubs() 74 tko = self.mox.CreateMock(frontend.TKO) 75 tko.run('get_detailed_test_views', afe_job_id=0).AndReturn( 76 [{'test_name':r[0], 'status':r[1]} for r in test_results]) 77 78 self.mox.ReplayAll() 79 test_views = site_utils.get_test_views_from_tko(0, tko) 80 self.mox.VerifyAll() 81 82 self.assertEqual(sorted(test_views.keys()), 83 sorted(expected_test_views.keys()), 84 'Test list %s does not match expected test list %s.' % 85 (sorted(test_views.keys()), 86 sorted(expected_test_views.keys()))) 87 88 for test_name, test_status_list in test_views.iteritems(): 89 self.assertEqual(sorted(test_status_list), 90 sorted(expected_test_views[test_name]), 91 'For test %s the status list %s does not match ' 92 'expected status list %s.' % 93 (test_name, 94 sorted(test_status_list), 95 sorted(expected_test_views[test_name]))) 96 97 98if __name__ == '__main__': 99 unittest.main() 100