164 lines
6.5 KiB
Python
Executable File
164 lines
6.5 KiB
Python
Executable File
#!/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()
|