Files
autospec/tests/test_commitmessage.py
T
Matthew Johnson 02c63d802c Record imported public key in commit message
When the user chooses to import a public key during the autospec run
display information on that key in the commit message.

Signed-off-by: Matthew Johnson <matthew.johnson@intel.com>
2018-01-18 15:20:30 -08:00

239 lines
9.9 KiB
Python

import unittest
import unittest.mock as mock
import os
import tempfile
import commitmessage
class TestCommitmessage(unittest.TestCase):
def setUp(self):
commitmessage.tarball.name = 'testball'
commitmessage.tarball.version = '0.0.1'
commitmessage.config.old_version = '0.0.0'
self.workingdir = tempfile.TemporaryDirectory()
commitmessage.build.setup_workingdir(self.workingdir.name)
def tearDown(self):
self.workingdir.cleanup()
def test_is_header(self):
"""
Test is_header function with list of lines. First and last line, line
followed by a line containing '---', and lines after a blank '' line
should be recognized as headers. Last line recognized as header because
it is a relevant ending point.
"""
lines = ['line1', # True
'line2', # False
'', # False
'line4', # True
'line5', # True
'---', # False
'line7', # False
'line8'] # True
for idx, line in enumerate(lines):
print(line, idx)
if idx in [0, 3, 4, 7]:
self.assertTrue(commitmessage.is_header(lines, idx))
else:
self.assertFalse(commitmessage.is_header(lines, idx))
def test_find_in_line(self):
"""
Trivially tests commitmessage.find_in_line(). Just makes sure results
evaluate to the correct bool.
"""
self.assertTrue(commitmessage.find_in_line(r'Version', 'AVersionInThisLine'))
self.assertFalse(commitmessage.find_in_line(r'z', 'the quick brown fox jumps over the lady dog'))
def test_process_NEWS(self):
"""
Test process_NEWS() function with valid newsfile provided
"""
with tempfile.TemporaryDirectory() as tmpd:
commitmessage.build.download_path = tmpd
with open(os.path.join(tmpd, 'NEWS'), 'w') as newsfile:
newsfile.write(GOOD_NEWS)
# commitmessage returned will have an empty string as first and
# last items
expected_msg = [""] + GOOD_NEWS.split('\n')[3:13]
expected_cvs = set()
self.assertEqual(commitmessage.process_NEWS('NEWS'),
(expected_msg, expected_cvs))
def test_process_NEWS_bad_news(self):
"""
Test process_NEWS() function with irrelevant newsfile provided
"""
with tempfile.TemporaryDirectory() as tmpd:
commitmessage.build.download_path = tmpd
with open(os.path.join(tmpd, 'NEWS'), 'w') as newsfile:
# make GOOD_NEWS irrelevant by replacing current version
newsfile.write(GOOD_NEWS.replace('0.0.1', '0.0.0'))
self.assertEqual(commitmessage.process_NEWS('NEWS'), ([], set()))
def test_process_NEWS_good_cves(self):
"""
Test process_NEWS() function with valid newsfile and CVEs
"""
with tempfile.TemporaryDirectory() as tmpd:
commitmessage.build.download_path = tmpd
with open(os.path.join(tmpd, 'NEWS'), 'w') as newsfile:
# give GOOD_NEWS some CVEs
newsfile.write(GOOD_NEWS.replace('change2.1', 'CVE-2-1')
.replace('change2.2', 'CVE-2-2'))
# commitmessage returned will have an empty string as first and
# last items.
# replace change2.* strings with CVE strings
expected_msg = [""] + GOOD_NEWS.replace('change2.1', 'CVE-2-1')\
.replace('change2.2', 'CVE-2-2')\
.split('\n')[3:13]
expected_cvs = set(['CVE-2-1', 'CVE-2-2'])
self.assertEqual(commitmessage.process_NEWS('NEWS'),
(expected_msg, expected_cvs))
def test_process_NEWS_long(self):
"""
Test process_NEWS() function with valid newsfile provided, but relevant
block is longer than 15 lines, causing it to be truncated.
"""
with tempfile.TemporaryDirectory() as tmpd:
commitmessage.build.download_path = tmpd
long_news = GOOD_NEWS.replace('text explaining change2.2',
'1\n2\n3\n4\n5\n6\n7\n8\n9\n')
with open(os.path.join(tmpd, 'NEWS'), 'w') as newsfile:
newsfile.write(long_news)
# commitmessage returned will have an empty string as first and
# last items, extend the expected message with extra lines and
# truncate message.
expected_msg = [""] + GOOD_NEWS.split('\n')[3:11]
expected_msg.extend(['1', '2', '3', '4', '5', '6', '7', '',
'(NEWS truncated at 15 lines)', ''])
expected_cvs = set()
self.assertEqual(commitmessage.process_NEWS('NEWS'),
(expected_msg, expected_cvs))
def test_guess_commit_message(self):
"""
Test guess_commit_message() with mocked internal functions and both
commitmessage information and cves available from newsfile.
"""
process_NEWS_backup = commitmessage.process_NEWS
def mock_process_NEWS(newsfile):
return (['', 'commit', 'message', 'with', 'cves', ''],
set(['cve1', 'cve2']))
commitmessage.process_NEWS = mock_process_NEWS
open_name = 'util.open'
with mock.patch(open_name, create=True) as mock_open:
mock_open.return_value = mock.MagicMock()
commitmessage.guess_commit_message("")
# reset mocks before asserting so a failure doesn't cascade to
# other tests
commitmessage.process_NEWS = process_NEWS_backup
fh = mock_open.return_value.__enter__.return_value
fh.write.assert_called_with(
'testball: Autospec creation for update from version 0.0.0 to '
'version 0.0.1\n\n\ncommit\nmessage\nwith\ncves\n\n\ncommit\n'
'message\nwith\ncves\n\nCVEs fixed in this build:\ncve1\ncve2'
'\n\n')
def test_guess_commit_message_cve_config(self):
"""
Test guess_commit_message() with mocked internal functions and both
commitmessage information and cves available from newsfile. A cve is
also available from config, which changes the first line of the commmit
message.
"""
process_NEWS_backup = commitmessage.process_NEWS
def mock_process_NEWS(newsfile):
return (['', 'commit', 'message', 'with', 'cves', ''],
set(['cve1', 'cve2']))
commitmessage.process_NEWS = mock_process_NEWS
commitmessage.config.cves = set(['CVE-1234-5678'])
commitmessage.config.old_version = None # Allow cve title to be set
open_name = 'util.open'
with mock.patch(open_name, create=True) as mock_open:
mock_open.return_value = mock.MagicMock()
commitmessage.guess_commit_message("")
# reset mocks before asserting so a failure doesn't cascade to
# other tests
commitmessage.process_NEWS = process_NEWS_backup
commitmessage.config.cves = set()
fh = mock_open.return_value.__enter__.return_value
fh.write.assert_called_with(
'testball: Fix for CVE-1234-5678\n\n\ncommit\nmessage\nwith\n'
'cves\n\n\ncommit\nmessage\nwith\ncves\n\nCVEs fixed in this '
'build:\nCVE-1234-5678\ncve1\ncve2\n\n')
def test_guess_commit_message_imported_key(self):
"""
Test guess_commit_message() with mocked internal functions and both
commitmessage information and cves available from newsfile. A cve is
also available from config, which changes the first line of the commmit
message. Additionally there is imported key info that will be displayed
at the end of the message.
"""
process_NEWS_backup = commitmessage.process_NEWS
def mock_process_NEWS(newsfile):
return (['', 'commit', 'message', 'with', 'cves', ''],
set(['cve1', 'cve2']))
commitmessage.process_NEWS = mock_process_NEWS
commitmessage.config.cves = set(['CVE-1234-5678'])
commitmessage.config.old_version = None # Allow cve title to be set
open_name = 'util.open'
with mock.patch(open_name, create=True) as mock_open:
mock_open.return_value = mock.MagicMock()
commitmessage.guess_commit_message("keyinfo content")
# reset mocks before asserting so a failure doesn't cascade to
# other tests
commitmessage.process_NEWS = process_NEWS_backup
commitmessage.config.cves = set()
fh = mock_open.return_value.__enter__.return_value
fh.write.assert_called_with(
'testball: Fix for CVE-1234-5678\n\n\ncommit\nmessage\nwith\n'
'cves\n\n\ncommit\nmessage\nwith\ncves\n\nCVEs fixed in this '
'build:\nCVE-1234-5678\ncve1\ncve2\n\nKey imported:\nkeyinfo '
'content\n')
def test_scan_for_changes(self):
"""
Tests scan_for_changes using temporary directories
"""
with tempfile.TemporaryDirectory() as tmpd:
with open(os.path.join(tmpd, 'changelog.txt'), 'w') as newsfile:
newsfile.write('new changelog file')
with tempfile.TemporaryDirectory() as tmpd1:
commitmessage.scan_for_changes(tmpd1, tmpd)
self.assertTrue(os.path.isfile(tmpd1 + '/ChangeLog'))
GOOD_NEWS = """
GOOD NEWS -- History of user-visible changes.
* Version 0.0.1
change1.1
change1.2
change2.1
text explaining change2.1
change2.2
text explaining change2.2
* Version 0.0.0
This better not show up
"""
if __name__ == '__main__':
unittest.main(buffer=True)