if repository.name == 'RepoTwo' else 'master')
in_branch = scm.git.query_local_repository_branch(git_dir)
self.assertEqual(want_branch, in_branch)
def test_foreach_repo(self):
test_root = os.path.join(self.base_temp_dir, 'foreach_test')
pos_args = [1, 2, 3]
kwargs = {'a': 'A', 'b': 'B'}
scm = SpinnakerSourceCodeManager(self.options, test_root)
all_repos = [
scm.make_repository_spec(repo_name, origin=origin)
for repo_name, origin in self.ORIGIN_URLS.items()
]
expect = {
repository.name: (repository, pos_args, kwargs)
for repository in all_repos
}
def _foreach_func(repository, *pos_args, **kwargs):
return (repository, list(pos_args), dict(kwargs))
got = scm.foreach_source_repository(all_repos, _foreach_func,
*pos_args, **kwargs)
self.assertEqual(expect, got)
if __name__ == '__main__':
init_runtime()
unittest.main(verbosity=2)
registry = {}
bomtool_main.add_standard_parser_args(self.parser, defaults)
buildtool.spinnaker_commands.register_commands(
registry, self.subparsers, defaults)
factory = registry['new_release_branch']
factory.init_argparser(self.parser, defaults)
options = self.parser.parse_args(['new_release_branch'])
mock_push_tag = self.patch_method(GitRunner, 'push_tag_to_origin')
mock_push_branch = self.patch_method(GitRunner, 'push_branch_to_origin')
command = factory.make_command(options)
command()
base_git_dir = os.path.join(options.input_dir, 'new_release_branch')
self.assertEqual(os.listdir(base_git_dir), [EXTRA_REPO])
git_dir = os.path.join(base_git_dir, EXTRA_REPO)
self.assertEqual(
GitRunner(options).query_local_repository_commit_id(git_dir),
self.repo_commit_map[EXTRA_REPO][EXTRA_REPO + '-branch'])
mock_push_branch.assert_called_once_with(git_dir, 'NewSpinnakerVersion')
self.assertEqual(0, mock_push_tag.call_count)
if __name__ == '__main__':
init_runtime()
unittest.main(verbosity=2)
