def request_pull(self, github_token, repo_name, branch_name):
print("Entering method requestpull(**********, " + repo_name + "," +
branch_name + ")")
gitator = Gitator()
gitator.create_github_connection(github_token)
gitator.check_repo_existence(repo_name)
if branch_name == 'merge_back':
branch_list = gitator.list_branches(repo_name)
gitator.delete_prs(repo_name)
branch_name_origin = 'master'
for branch in branch_list:
branch_name_destination = branch.name
i = gitator.create_issue(repo_name, branch_name_origin,
branch_name_destination)
gitator.create_pr(repo_name, branch_name_origin,
branch_name_destination, i)
else:
branch_name_origin = branch_name
branch_name_destination = 'master'
if not gitator.check_branch_existence(repo_name,
branch_name_origin):
raise Exception('Branch not found.')
i = gitator.create_issue(repo_name, branch_name_origin,
branch_name_destination)
gitator.create_pr(repo_name, branch_name_origin,
branch_name_destination, i)
def test_create_github_connection_bad():
thing = Gitator()
mock_class = create_autospec(AuthenticatedUser, instance=True)
type(mock_class).id = PropertyMock(
side_effect=Exception("Couldn't connect to github."))
Github.get_user = MagicMock(return_value=mock_class)
with pytest.raises(Exception) as e:
thing.create_github_connection("token")
assert e.value.__str__() == "Couldn't connect to github."
def test_create_github_connection_good():
thing = Gitator()
mock_class = create_autospec(AuthenticatedUser, instance=True)
mock_class.id = 1234
Github.get_user = MagicMock(return_value=mock_class)
assert thing.create_github_connection("token") == 1234
def test_check_repo_existence_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
thing.g.get_repo = MagicMock(return_value=False)
with pytest.raises(Exception) as ex:
assert thing.check_repo_existence('repo_name') == False
assert ex.value.__str__() == "'bool' object has no attribute 'id'"
def test_check_branch_existence_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
repo.get_branch = MagicMock()
thing.g.get_repo = MagicMock(return_value=repo)
assert thing.check_branch_existence('repo_name', 'branch_name') == True
repo.get_branch.assert_called_once_with('branch_name')
def test_delete_prs_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
assert thing.delete_prs("repo_name") == True
repo.get_pulls.assert_called_once_with(state='open')
def test_tag_master_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
assert thing.tag_master("repo_name", 'commit_sha') == True
repo.create_git_ref.assert_called_once_with('refs/tags/latest',
'commit_sha')
def test_merge_pr_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
assert thing.merge_pr("repo_name", pr) == True
pr.merge.assert_called_once_with()
def test_check_pr_existence_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
repo.get_pull = MagicMock()
pr = create_autospec(PullRequest.PullRequest, instance=True)
thing.g.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
assert thing.check_pr_existence('repo_name', 1) == pr
repo.get_pull.assert_called_once_with(1)
def test_create_branch_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
# thing.run_checks = MagicMock(return_value=True)
assert thing.create_branch("repo_name", "branch_name",
"commit_sha") == True
repo.create_git_ref.assert_called_once_with("refs/heads/branch_name",
"commit_sha")
def test_get_current_commit_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_commits = MagicMock(side_effect=Exception("Can't get commits"))
# thing.run_checks = MagicMock(return_value=True)
with pytest.raises(Exception) as ex:
assert thing.get_current_commit("repo_name") == False
repo.get_commits.assert_called_once_with()
def test_check_pr_mergeabilityy_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
pr.mergeable = True
assert thing.check_pr_mergeability(pr) == True
def test_delete_prs_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pulls = MagicMock(side_effect=Exception("Can't get pulls"))
# thing.run_checks = MagicMock(return_value=True)
with pytest.raises(Exception) as ex:
assert thing.delete_prs("repo_name") == False
repo.get_pulls.assert_called_once_with(state='open')
assert ex.value.__str__() == "Can't get pulls"
def test_list_forks_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
repo.forks = 1
forks = create_autospec(PaginatedList.PaginatedList, instance=True)
thing.g.get_repo = MagicMock(return_value=repo)
repo.get_forks = MagicMock(return_value=forks)
assert thing.list_forks(
'repo_name'
) == 'O repositório repo_name possui 1 forks. Segue lista abaixo:'
repo.get_forks.assert_called_once_with()
def test_list_branches_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_branches = MagicMock(
side_effect=Exception("Problem creating pull request."))
with pytest.raises(Exception) as ex:
assert thing.list_branches("repo_name")
repo.get_branches.assert_called_once_with()
def test_check_pr_mergeability_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
pr.mergeable = False
with pytest.raises(Exception) as ex:
assert thing.check_pr_mergeability(pr) == False
assert str(ex.value) == "PR is not mergeable!"
def test_check_pr_existence_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
ex = Exception()
ex.status = 404
ex.message = "PR not found"
repo.get_pull = MagicMock(side_effect=ex)
thing.g.get_repo = MagicMock(return_value=repo)
with pytest.raises(Exception) as ex:
assert thing.check_pr_existence('repo_name', 1) == False
assert str(ex.value.message) == "PR not found"
repo.get_pull.assert_called_once_with(1)
def test_tag_master_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.create_git_ref = MagicMock(
side_effect=Exception("Problem creating ref."))
with pytest.raises(Exception) as ex:
assert thing.tag_master("repo_name", 'commit_sha') == False
assert str(ex.value) == "Problem creating ref."
repo.create_git_ref.assert_called_once_with('refs/tags/latest',
'commit_sha')
def test_list_branches_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
branches = []
repo.get_branches = MagicMock(return_value=branches)
# thing.run_checks = MagicMock(return_value=True)
assert thing.list_branches("repo_name") == branches
repo.get_branches.assert_called_once_with()
def test_create_pr_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
issue = create_autospec(Issue.Issue, instance=True)
# thing.run_checks = MagicMock(return_value=True)
assert thing.create_pr("repo_name", "branch_name_origin",
"branch_name_destination", issue) == True
repo.create_pull.assert_called_once_with(issue=issue,
base="branch_name_destination",
head="branch_name_origin")
def test_check_branch_existence_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
ex = Exception()
ex.status = 500
ex.message = "Problem talking to github"
repo.get_branch = MagicMock(side_effect=ex)
thing.g.get_repo = MagicMock(return_value=repo)
with pytest.raises(Exception) as ex:
assert thing.check_branch_existence('repo_name',
'branch_name') == False
assert str(ex.value.message) == "Problem talking to github"
repo.get_branch.assert_called_once_with('branch_name')
def test_get_current_commit_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
commits = create_autospec(PaginatedList.PaginatedList)
repo.get_commits = MagicMock(return_value=commits)
commits[0].sha = 'commit_sha'
# thing.run_checks = MagicMock(return_value=True)
assert thing.get_current_commit("repo_name") == 'commit_sha'
repo.get_commits.assert_called_once_with()
def test_delete_branch_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
branch = create_autospec(GitRef.GitRef, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
repo.get_git_ref = MagicMock(return_value=branch)
pr.head.ref = 'hotfix-branch'
assert thing.delete_branch("repo_name", pr) == True
branch.delete.assert_called_once_with()
def test_create_branch_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
repo.create_git_ref = MagicMock(
side_effect=Exception("Problem branching."))
with pytest.raises(Exception) as ex:
assert thing.create_branch("repo_name", "branch_name",
"commit_sha") == False
assert str(ex.value) == "Problem branching."
repo.create_git_ref.assert_called_once_with("refs/heads/branch_name",
"commit_sha")
def test_create_issue_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
repo.create_issue = MagicMock(
side_effect=Exception("Problem creating issue."))
# thing.run_checks = MagicMock(return_value=True)
with pytest.raises(Exception) as ex:
assert thing.create_issue("repo_name", "branch_name_origin",
"branch_name_destination") == False
repo.create_issue.assert_called_once_with(
"Pull Request from branch branch_name_origin to branch branch_name_destination"
)
def test_create_issue_good():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
issue = create_autospec(Issue.Issue, instance=True)
repo.create_issue = MagicMock(return_value=issue)
# thing.run_checks = MagicMock(return_value=True)
assert thing.create_issue("repo_name", "branch_name_origin",
"branch_name_destination") == issue
repo.create_issue.assert_called_once_with(
"Pull Request from branch branch_name_origin to branch branch_name_destination"
)
def test_merge_pr_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
e = Exception("Problem merging pull request.")
pr.merge = MagicMock(side_effect=e)
with pytest.raises(Exception) as ex:
assert thing.merge_pr("repo_name", pr) == False
assert str(ex.value) == "Problem merging pull request."
pr.merge.assert_called_once_with()
def test_list_forks_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
thing.g = MagicMock()
repo = create_autospec(Repository, instance=True)
repo.forks = 1
ex = Exception()
ex.status = 404
ex.message = "Problem listing forks"
repo.get_forks = MagicMock(side_effect=ex)
thing.g.get_repo = MagicMock(return_value=repo)
with pytest.raises(Exception) as ex:
assert thing.list_forks('repo_name') == False
assert str(ex.value.message) == "Problem listing forks"
repo.get_forks.assert_called_once_with()
def test_create_pr_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
thing.g = github_connection
repo = create_autospec(Repository.Repository, instance=True)
github_connection.get_repo = MagicMock(return_value=repo)
issue = create_autospec(Issue.Issue, instance=True)
repo.create_pull = MagicMock(
side_effect=Exception("Problem creating pull request."))
# thing.run_checks = MagicMock(return_value=True)
with pytest.raises(Exception) as ex:
assert thing.create_pr("repo_name", "branch_name_origin",
"branch_name_destination", issue) == False
repo.create_pull.assert_called_once_with(
issue=issue,
base="branch_name_destination",
head="branch_name_origin")
def test_delete_branch_bad():
thing = Gitator()
thing.create_github_connection = MagicMock()
github_connection = create_autospec(Github)
repo = create_autospec(Repository.Repository, instance=True)
pr = create_autospec(PullRequest.PullRequest, instance=True)
branch = create_autospec(GitRef.GitRef, instance=True)
thing.g = github_connection
github_connection.get_repo = MagicMock(return_value=repo)
repo.get_pull = MagicMock(return_value=pr)
repo.get_git_ref = MagicMock(return_value=branch)
pr.head.ref = 'hotfix-branch'
branch.delete = MagicMock(
side_effect=Exception("Problem deleting branch."))
with pytest.raises(Exception) as ex:
assert thing.delete_branch("repo_name", pr) == False
assert str(ex.value) == "Problem deleting branch."
branch.delete.assert_called_once_with()