def test_mergeBranch_fails_on_source_fetch(self):
with mock.patch('subprocess.check_output',
side_effect=['origin /dev/null/target/Trilinos',
'df324ae']) as m_check_out, \
mock.patch('subprocess.check_call',
side_effect=[None,
CalledProcessError(-1, 'cmd'),
CalledProcessError(-2, 'cmd'),
CalledProcessError(-3, 'cmd')]) as m_check_call:
if sys.version_info.major is not 3:
with self.assertRaisesRegexp(SystemExit, '12'):
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland',
'fake_develop', 'df324ae')
else:
with self.assertRaisesRegex(SystemExit, '12'):
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland',
'fake_develop', 'df324ae')
m_check_out.assert_has_calls([mock.call(['git', 'remote', '-v'])])
m_check_call.assert_has_calls([
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
])
def test_echoJenkinsVars(self):
with self.m_environ:
env_string_io = StringIO()
for key in os.environ:
print(key + ' = ' + os.environ[key], file=env_string_io)
expected_string = '''
================================================================================
Jenkins Environment Variables:
- WORKSPACE : /dev/null/TEST_WORKSPACE
================================================================================
Environment:
pwd = {cwd}
{environ}
================================================================================
'''.format(cwd=os.getcwd(), environ=env_string_io.getvalue())
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout, \
self.m_environ:
PullRequestLinuxDriverMerge.echoJenkinsVars(
os.path.join(os.sep, 'dev', 'null', 'TEST_WORKSPACE'))
self.assertEqual(expected_string, m_stdout.getvalue())
def test_mergeBranch_fails_on_SHA_mismatch(self):
with mock.patch('subprocess.check_output',
side_effect=['origin /dev/null/target/Trilinos',
'df324ae']) as m_check_out, \
mock.patch('subprocess.check_call') as m_check_call, \
mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
if sys.version_info.major is not 3:
with self.assertRaisesRegexp(SystemExit, '-1'):
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland',
'fake_develop', 'foobar')
else:
with self.assertRaisesRegex(SystemExit, '-1'):
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland',
'fake_develop', 'foobar')
m_check_out.assert_has_calls([
mock.call(['git', 'remote', '-v']),
mock.call(['git', 'rev-parse', 'source_remote/neverland'])
])
m_check_call.assert_has_calls([
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
])
self.assertEqual(
'''The SHA (df324ae) for the last commit on branch neverland
in repo /dev/null/source/Trilinos.git is different than the expected SHA,
which is: foobar.\n''', m_stdout.getvalue())
def test_mergeBranch_without_source_remote(self):
with mock.patch('subprocess.check_output',
side_effect=['origin /dev/null/target/Trilinos',
'df324ae']) as m_check_out, \
mock.patch('subprocess.check_call') as m_check_call:
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland', 'fake_develop',
'df324ae')
m_check_out.assert_has_calls([
mock.call(['git', 'remote', '-v']),
mock.call(['git', 'rev-parse', 'source_remote/neverland'])
])
m_check_call.assert_has_calls([
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
mock.call(['git', 'fetch', 'origin', 'fake_develop']),
mock.call(['git', 'reset', '--hard', 'HEAD']),
mock.call([
'git', 'checkout', '-B', 'fake_develop', 'origin/fake_develop'
]),
mock.call(['git', 'merge', '--no-edit',
'source_remote/neverland']),
])
def test_mergeBranch_with_source_remote(self):
with mock.patch('subprocess.check_output',
side_effect=['''origin /dev/null/target/Trilinos
source_remote /dev/null/source12/Trilinos.git''', 'df324ae']) as m_check_out, \
mock.patch('subprocess.check_call') as m_check_call, \
mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
PullRequestLinuxDriverMerge.merge_branch(
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'), 'neverland', 'fake_develop',
'df324ae')
m_check_out.assert_has_calls([
mock.call(['git', 'remote', '-v']),
mock.call(['git', 'rev-parse', 'source_remote/neverland'])
])
m_check_call.assert_has_calls([
mock.call(['git', 'remote', 'rm', 'source_remote']),
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
mock.call(['git', 'fetch', 'origin', 'fake_develop']),
mock.call(['git', 'reset', '--hard', 'HEAD']),
mock.call([
'git', 'checkout', '-B', 'fake_develop', 'origin/fake_develop'
]),
mock.call(['git', 'merge', '--no-edit',
'source_remote/neverland']),
])
self.assertEqual("git remote exists, removing it\n",
m_stdout.getvalue())
def test_writeHeader(self):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
PullRequestLinuxDriverMerge.write_header()
self.assertEqual(
'''--------------------------------------------------------------------------------
-
- Begin: PullRequestLinuxDriver-Merge.py
-
--------------------------------------------------------------------------------
''', m_stdout.getvalue())
def test_writeHeader(self):
print(
"BEGIN: Test PullRequestLinuxDriverMerge.py :: `write_header()`:")
#m_stdout = StringIO()
#with contextlib.redirect_stdout(m_stdout):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
PRMerge.write_header()
self.assertIn("Begin: PullRequestLinuxDriverMerge.py",
m_stdout.getvalue())
print(
"FINISH: Test PullRequestLinuxDriverMerge.py :: `write_header()`:")
return
def test_run_fails_on_bad_fetch(self, m_parseArgs, m_chdir, m_join):
"""
TODO: Add a docstring explaining this test and its purpose.
"""
side_effect_check_output = [
'origin /dev/null/target/Trilinos', 'df324ae'
]
side_effect_check_call = []
side_effect_check_call.append(None)
side_effect_check_call.append(CalledProcessError(-1, 'cmd'))
side_effect_check_call.append(CalledProcessError(-2, 'cmd'))
side_effect_check_call.append(CalledProcessError(-3, 'cmd'))
with mock.patch('subprocess.check_output',
side_effect=side_effect_check_output):
with mock.patch('subprocess.check_call',
side_effect=side_effect_check_call):
# I'm not sure if we really needed to mock out stdio, we don't check it
# so would it be more useful to provide that to the output if the test
# should fail? Otherwise, are we flying blin?
with mock.patch('sys.stdout', new_callable=StringIO):
self.assertFalse(PRMerge.run())
return
def test_mergeBranch_with_source_remote(self, m_check_call):
"""
"""
tmp_path = os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git')
side_effect_list = [
"origin /dev/null/target/Trilinos\nsource_remote /dev/null/source12/Trilinos.git",
'df324ae'
]
with mock.patch('subprocess.check_output',
side_effect=side_effect_list) as m_check_out:
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
PRMerge.merge_branch(tmp_path, 'neverland', 'fake_develop',
'df324ae')
expected_calls = []
expected_calls.append(mock.call(['git', 'remote', '-v']))
expected_calls.append(
mock.call(['git', 'rev-parse', 'source_remote/neverland']))
m_check_out.assert_has_calls(expected_calls)
m_check_call.assert_has_calls([
mock.call(['git', 'remote', 'rm', 'source_remote']),
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]),
mock.call(['git', 'fetch', 'source_remote', 'neverland']),
mock.call(['git', 'fetch', 'origin', 'fake_develop']),
mock.call(['git', 'reset', '--hard', 'HEAD']),
mock.call([
'git', 'checkout', '-B', 'fake_develop', 'origin/fake_develop'
]),
mock.call(['git', 'merge', '--no-edit',
'source_remote/neverland']),
])
self.assertIn("git remote exists, removing it", m_stdout.getvalue())
return
def test_mergeBranch_fails_on_source_fetch(self, m_check_out):
"""
Tests that ``merge_branch`` should fail on source fetch if a failure
occurrs in the *fetch* operation
"""
tmp_path = os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git')
side_effect_list = [
None,
CalledProcessError(-1, 'cmd'),
CalledProcessError(-2, 'cmd'),
CalledProcessError(-3, 'cmd')
]
with self.assertRaises(SystemExit):
with mock.patch('subprocess.check_call',
side_effect=side_effect_list) as m_check_call:
PRMerge.merge_branch(tmp_path, 'neverland', 'fake_develop',
'df324ae')
expected_calls = []
expected_calls.append(mock.call(['git', 'remote', '-v']))
m_check_out.assert_has_calls(expected_calls)
expected_calls = []
expected_calls.append(
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]))
expected_calls.append(
mock.call(['git', 'fetch', 'source_remote', 'neverland']))
expected_calls.append(
mock.call(['git', 'fetch', 'source_remote', 'neverland']))
expected_calls.append(
mock.call(['git', 'fetch', 'source_remote', 'neverland']))
m_check_call.assert_has_calls(expected_calls)
return
def test_mergeBranch_fails_on_SHA_mismatch(self, m_check_call,
m_check_out):
"""
Test that ``merge_branch`` will fail if a SHA mismatch detected.
"""
with self.assertRaises(SystemExit):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
tmp_path = os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git')
PRMerge.merge_branch(tmp_path, 'neverland', 'fake_develop',
'foobar')
expected_calls = []
expected_calls.append(mock.call(['git', 'remote', '-v']))
expected_calls.append(
mock.call(['git', 'rev-parse', 'source_remote/neverland']))
m_check_out.assert_has_calls(expected_calls)
expected_calls = []
expected_calls.append(
mock.call([
'git', 'remote', 'add', 'source_remote',
'/dev/null/source/Trilinos.git'
]))
expected_calls.append(
mock.call(['git', 'fetch', 'source_remote', 'neverland']))
m_check_call.assert_has_calls(expected_calls)
stdout_actual = m_stdout.getvalue()
stdout_expected_1 = "The SHA (df324ae) for the last commit on branch neverland"
stdout_expected_2 = "is different from the expected SHA"
stdout_expected_3 = "which is: foobar"
self.assertIn(stdout_expected_1, stdout_actual)
self.assertIn(stdout_expected_2, stdout_actual)
self.assertIn(stdout_expected_3, stdout_actual)
return
def test_run_fails_on_bad_fetch(self):
with mock.patch('subprocess.check_output',
side_effect=['origin /dev/null/target/Trilinos',
'df324ae']), \
mock.patch('subprocess.check_call',
side_effect=[None,
CalledProcessError(-1, 'cmd'),
CalledProcessError(-2, 'cmd'),
CalledProcessError(-3, 'cmd')]), \
mock.patch('PullRequestLinuxDriverMerge.parseArgs'), \
mock.patch('sys.stdout', new_callable=StringIO), \
mock.patch('os.path.join'), \
mock.patch('os.chdir'):
self.assertFalse(PullRequestLinuxDriverMerge.run())
def test_parseInsufficientArgs_fails(self):
test_namespace = Namespace()
setattr(test_namespace, 'sourceRepo', '/dev/null/source/Trilinos.git')
expected_output = '''usage: programName [-h]
sourceRepo sourceBranch targetRepo targetBranch sourceSHA
workspaceDir
programName: error: the following arguments are required: sourceRepo, \
sourceBranch, targetRepo, targetBranch, sourceSHA, workspaceDir
'''
if sys.version_info.major is not 3:
expected_output = '''usage: programName [-h]
sourceRepo sourceBranch targetRepo targetBranch sourceSHA
workspaceDir\nprogramName: error: too few arguments
'''
with mock.patch.object(sys, 'argv', ['programName']), \
mock.patch('sys.stderr', new_callable=StringIO) as m_stderr:
if sys.version_info.major is not 3:
with self.assertRaisesRegexp(SystemExit, '2'):
PullRequestLinuxDriverMerge.parseArgs()
else:
with self.assertRaisesRegex(SystemExit, '2'):
PullRequestLinuxDriverMerge.parseArgs()
self.assertEqual(expected_output, m_stderr.getvalue())
def test_echoJenkinsVars(self):
print(
"BEGIN: Test PullRequestLinuxDriverMerge.py :: `echoJenkinsVars()`:"
)
with self.m_environ:
env_string_io = StringIO()
for key in os.environ:
print(key + ' = ' + os.environ[key], file=env_string_io)
tmp_path = os.path.join(os.sep, 'dev', 'null', 'TEST_WORKSPACE')
with mock.patch('sys.stdout',
new_callable=StringIO) as m_stdout, self.m_environ:
PRMerge.echoJenkinsVars(tmp_path)
stdout_actual = m_stdout.getvalue()
self.assertIn("JOB_BASE_NAME = TEST_JOB_BASE_NAME", stdout_actual)
self.assertIn("JOB_NAME = TEST_JOB_NAME", stdout_actual)
self.assertIn("WORKSPACE = /dev/null/TEST_WORKSPACE", stdout_actual)
self.assertIn("NODE_NAME = TEST_NODE_NAME", stdout_actual)
print(
"FINISH: Test PullRequestLinuxDriverMerge.py :: `echoJenkinsVars()`:"
)
return
def test_run(self, m_join, m_mergeBranch, m_echoJenkins, m_writeHeader,
m_chdir, m_parser):
"""
TODO: Add docstring that explains this test's purpose
"""
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
self.assertTrue(PRMerge.run())
m_parser.assert_called_once_with()
m_chdir.assert_called_once_with(
m_join(m_parser().workspaceDir, 'Trilinos'))
m_writeHeader.assert_called_once_with()
m_echoJenkins.assert_called_once_with(m_parser().workspaceDir)
m_mergeBranch.assert_called_once_with(m_parser().sourceRepo,
m_parser().sourceBranch,
m_parser().targetBranch,
m_parser().sourceSHA)
self.assertIn('Set CWD = ' + str(m_join()) + '\n', m_stdout.getvalue())
return
def test_run(self):
with mock.patch('PullRequestLinuxDriverMerge.parseArgs') as m_parser, \
mock.patch('os.chdir') as m_chdir, \
mock.patch('PullRequestLinuxDriverMerge.write_header') as m_writeHeader, \
mock.patch('PullRequestLinuxDriverMerge.echoJenkinsVars') as m_echoJenkins, \
mock.patch('PullRequestLinuxDriverMerge.merge_branch') as m_mergeBranch, \
mock.patch('os.path.join') as m_join, \
mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
self.assertTrue(PullRequestLinuxDriverMerge.run())
m_parser.assert_called_once_with()
m_chdir.assert_called_once_with(
m_join(m_parser().workspaceDir, 'Trilinos'))
m_writeHeader.assert_called_once_with()
m_echoJenkins.assert_called_once_with(m_parser().workspaceDir)
m_mergeBranch.assert_called_once_with(m_parser().sourceRepo,
m_parser().sourceBranch,
m_parser().targetBranch,
m_parser().sourceSHA)
self.assertEqual('Set CWD = ' + str(m_join()) + '\n',
m_stdout.getvalue())
def test_run_fails_on_bad_remote_add(self):
expected_string = '''Recieved subprocess.CalledProcessError - returned -1
from command test_cmd
output None
stdout None
stderr None\n'''
if sys.version_info.major is not 3:
expected_string = '''Recieved subprocess.CalledProcessError - returned -1
from command test_cmd
output None\n'''
with mock.patch('subprocess.check_output',
side_effect=['origin /dev/null/target/Trilinos',
'df324ae']), \
mock.patch('subprocess.check_call',
side_effect=CalledProcessError(-1, 'test_cmd')), \
mock.patch('PullRequestLinuxDriverMerge.parseArgs'), \
mock.patch('sys.stdout', new_callable=StringIO) as m_stdout, \
mock.patch('os.path.join'), \
mock.patch('os.chdir'):
self.assertFalse(PullRequestLinuxDriverMerge.run())
self.assertTrue(m_stdout.getvalue().endswith(expected_string))
def test_run_fails_on_bad_remote_add(self, m_stdout, m_parseArgs, m_chdir,
m_path_join):
expected_string = '''Recieved subprocess.CalledProcessError - returned -1
from command test_cmd
output None
stdout None
stderr None\n'''
if sys.version_info.major != 3:
expected_string = '''Recieved subprocess.CalledProcessError - returned -1
from command test_cmd
output None\n'''
side_effect_check_output = [
'origin /dev/null/target/Trilinos', 'df324ae'
]
side_effect_check_call = CalledProcessError(-1, 'test_cmd')
with mock.patch('subprocess.check_output', side_effect=side_effect_check_output), \
mock.patch('subprocess.check_call', side_effect=side_effect_check_call):
self.assertFalse(PRMerge.run())
expected_string_list = []
expected_string_list.append(
"Received subprocess.CalledProcessError - returned -1")
expected_string_list.append("from command test_cmd")
expected_string_list.append("output None")
if sys.version_info.major == 3:
expected_string_list.append("output None")
expected_string_list.append("stdout None")
expected_string_list.append("stderr None")
for expected_string_i in expected_string_list:
self.assertIn(expected_string_i, m_stdout.getvalue())
return
def test_parseArgs(self):
test_namespace = Namespace()
setattr(test_namespace, 'sourceRepo', '/dev/null/source/Trilinos.git')
setattr(test_namespace, 'sourceBranch',
'fake_test_branch_fixing_issue_neverland')
setattr(test_namespace, 'targetRepo', '/dev/null/target/Trilinos.git')
setattr(test_namespace, 'targetBranch', 'fake_develop')
setattr(test_namespace, 'sourceSHA', '0123456789abcdef')
setattr(test_namespace, 'workspaceDir', '/dev/null/workspace')
with mock.patch.object(sys, 'argv', [
'programName',
os.path.join(os.path.sep, 'dev', 'null', 'source',
'Trilinos.git'),
'fake_test_branch_fixing_issue_neverland',
os.path.join(os.path.sep, 'dev', 'null', 'target',
'Trilinos.git'), 'fake_develop',
'0123456789abcdef',
os.path.join(os.path.sep, 'dev', 'null', 'workspace')
]):
args = PullRequestLinuxDriverMerge.parseArgs()
self.assertEqual(test_namespace, args)
def test_run_fails_on_bad_parse(self):
with mock.patch('PullRequestLinuxDriverMerge.parseArgs',
side_effect=SystemExit(2)):
self.assertFalse(PullRequestLinuxDriverMerge.run())