| #!/usr/bin/python2 |
| # |
| # Copyright (c) 2013 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. |
| |
| import datetime |
| import mock |
| import mox |
| import unittest |
| |
| import common |
| |
| from autotest_lib.server.cros.dynamic_suite import reporting_utils |
| |
| |
| class TestMergeBugTemplate(mox.MoxTestBase): |
| """Test bug can be properly merged and validated.""" |
| def test_validate_success(self): |
| """Test a valid bug can be verified successfully.""" |
| bug_template= {} |
| bug_template['owner'] = 'someone@company.com' |
| reporting_utils.BugTemplate.validate_bug_template(bug_template) |
| |
| |
| def test_validate_success(self): |
| """Test a valid bug can be verified successfully.""" |
| # Bug template must be a dictionary. |
| bug_template = ['test'] |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| # Bug template must contain value for essential attribute, e.g., owner. |
| bug_template= {'no-owner': 'user1'} |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| # Bug template must contain value for essential attribute, e.g., owner. |
| bug_template= {'owner': 'invalid_email_address'} |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| # Check unexpected attributes. |
| bug_template= {} |
| bug_template['random tag'] = 'test' |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| # Value for cc must be a list |
| bug_template= {} |
| bug_template['cc'] = 'test' |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| # Value for labels must be a list |
| bug_template= {} |
| bug_template['labels'] = 'test' |
| self.assertRaises(reporting_utils.InvalidBugTemplateException, |
| reporting_utils.BugTemplate.validate_bug_template, |
| bug_template) |
| |
| |
| def test_merge_success(self): |
| """Test test and suite bug templates can be merged successfully.""" |
| test_bug_template = { |
| 'labels': ['l1'], |
| 'owner': 'user1@chromium.org', |
| 'status': 'Assigned', |
| 'title': None, |
| 'cc': ['cc1@chromium.org', 'cc2@chromium.org'] |
| } |
| suite_bug_template = { |
| 'labels': ['l2'], |
| 'owner': 'user2@chromium.org', |
| 'status': 'Fixed', |
| 'summary': 'This is a short summary for suite bug', |
| 'title': 'Title for suite bug', |
| 'cc': ['cc2@chromium.org', 'cc3@chromium.org'] |
| } |
| bug_template = reporting_utils.BugTemplate(suite_bug_template) |
| merged_bug_template = bug_template.finalize_bug_template( |
| test_bug_template) |
| self.assertEqual(merged_bug_template['owner'], |
| test_bug_template['owner'], |
| 'Value in test bug template should prevail.') |
| |
| self.assertEqual(merged_bug_template['title'], |
| suite_bug_template['title'], |
| 'If an attribute has value None in test bug template, ' |
| 'use the value given in suite bug template.') |
| |
| self.assertEqual(merged_bug_template['summary'], |
| suite_bug_template['summary'], |
| 'If an attribute does not exist in test bug template, ' |
| 'but exists in suite bug template, it should be ' |
| 'included in the merged template.') |
| |
| self.assertEqual(merged_bug_template['cc'], |
| test_bug_template['cc'] + suite_bug_template['cc'], |
| 'List values for an attribute should be merged.') |
| |
| self.assertEqual(merged_bug_template['labels'], |
| test_bug_template['labels'] + |
| suite_bug_template['labels'], |
| 'List values for an attribute should be merged.') |
| |
| test_bug_template['owner'] = '' |
| test_bug_template['cc'] = [''] |
| suite_bug_template['owner'] = '' |
| suite_bug_template['cc'] = [''] |
| bug_template = reporting_utils.BugTemplate(suite_bug_template) |
| merged_bug_template = bug_template.finalize_bug_template( |
| test_bug_template) |
| self.assertFalse('owner' in merged_bug_template, |
| 'owner should be removed from the merged template.') |
| self.assertFalse('cc' in merged_bug_template, |
| 'cc should be removed from the merged template.') |
| |
| |
| class NewDatetime(datetime.datetime): |
| """Fake datetime.datetime class. |
| |
| datetime.datetime cannot be patched directly since it's a built-in |
| type that is immutable. NewDatetime is created here for patching |
| datetime.datetime.now. |
| """ |
| @classmethod |
| def utcnow(cls): |
| """Fake datetime.datetime.utcnow().""" |
| pass |
| |
| |
| class TestLinks(unittest.TestCase): |
| """Test links of test can be properly generated.""" |
| |
| def setUp(self): |
| old_datetime = datetime.datetime |
| datetime.datetime = NewDatetime |
| mock_now = mock.patch('datetime.datetime.utcnow') |
| self._mock_now = mock_now.start() |
| self.addCleanup(mock_now.stop) |
| self.addCleanup(setattr, datetime, 'datetime', old_datetime) |
| |
| |
| def test_generate_test_history_link(self): |
| """Test a link of test history can be generated.""" |
| self._mock_now.return_value = datetime.datetime(2018, 3, 29) |
| link = reporting_utils.link_test_history('jetstream_PrioritizedDevice') |
| expected_link = ('https://stainless.corp.google.com/search?' |
| 'test=^jetstream\_PrioritizedDevice$&' |
| 'first_date=2018-03-01&' |
| 'last_date=2018-03-29&' |
| 'row=model&col=build&view=matrix') |
| self.assertEqual(link, expected_link) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
