def test_get_pull_requests_by_head__no_pulls(self, mock_util, repo):
self.init_github()
mock_util.mock_pulls()
pull_requests = get_pull_requests_by_head(repo, "test_branch")
assert pull_requests == []
pull_requests = get_pull_requests_by_head(repo, "master")
assert pull_requests is None
def test_get_pull_requests_by_head(self, mock_util, repo):
self.init_github()
mock_util.mock_pulls(
pulls=self._get_expected_pull_requests(1),
head=repo.owner.login + ":" + "some-other-branch",
)
pull_requests = get_pull_requests_by_head(repo, "some-other-branch")
assert 1 == len(pull_requests)
# ConnectionError present when we reachout with
# a branch name (url parameter) that we aren't expecting
with pytest.raises(ConnectionError):
get_pull_requests_by_head(repo, "does-not-exist")
def test_get_pull_request_by_head__multiple_pulls(self, mock_util, repo):
self.init_github()
mock_util.mock_pulls(pulls=self._get_expected_pull_requests(2))
pull_requests = get_pull_requests_by_head(repo, "test_branch")
assert 2 == len(pull_requests)
