def test_push_handler_localbranch(mock_finish, mock_git):
# Given
mock_git.get_current_branch.return_value = 'foo'
mock_git.get_upstream_branch.return_value = 'localbranch'
mock_git.push.return_value = {'code': 0}
# When
GitPushHandler().post()
# Then
mock_git.get_current_branch.assert_called_with('test_path')
mock_git.get_upstream_branch.assert_called_with('test_path', 'foo')
mock_git.push.assert_called_with('.', 'HEAD:localbranch', 'test_path')
mock_finish.assert_called_with('{"code": 0}')
def test_push_handler_noupstream(mock_finish, mock_git):
# Given
mock_git.get_current_branch.return_value = 'foo'
mock_git.get_upstream_branch.return_value = ''
mock_git.push.return_value = {'code': 0}
# When
GitPushHandler().post()
# Then
mock_git.get_current_branch.assert_called_with('test_path')
mock_git.get_upstream_branch.assert_called_with('test_path', 'foo')
mock_git.push.assert_not_called()
mock_finish.assert_called_with(
'{"code": 128, "message": "fatal: The current branch foo has no upstream branch."}'
)