def test_update_branches_new(self, list_jobs_mock):
list_jobs_mock.return_value = ['job0']
repo = Mock()
repo.branches = []
repo.remotes = MagicMock()
repo.remotes['origin'].refs = []
update_branches(repo)
repo.create_head.assert_called_once_with('job0', 'master')
def test_delete_local_branch(self):
branch = Mock()
branch.name = 'jobs0'
repo = Mock()
repo.working_dir = '/tmp/aos-cd-jobs'
repo.branches = {'jobs0': branch}
update_branches(repo)
self.assertTrue(branch.delete.called)
def test_update_branches_remote(self, create_mock):
repo = Mock()
repo.working_dir = '/tmp/aos-cd-jobs'
repo.branches = []
repo.remotes = MagicMock()
repo.remotes['origin'].refs = {'job0': Mock()}
update_branches(repo)
repo.create_head.assert_has_calls((call('job0', 'origin/job0'),))
create_mock.assert_has_calls((call(repo, 'job0'),))
def test_update_branches(self, create_mock):
repo = Mock()
repo.working_dir = '/tmp/aos-cd-jobs'
repo.branches = {'job0': Mock(), 'job1': Mock()}
update_branches(repo)
create_mock.assert_has_calls((call(repo, 'job0'), call(repo, 'job1')))