91 lines
4.0 KiB
Python
91 lines
4.0 KiB
Python
import json
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
import pr_overlap
|
|
|
|
|
|
class OpenPullRequestsTests(unittest.TestCase):
|
|
def test_collects_pull_requests_beyond_the_previous_limit(self):
|
|
pull_requests = [
|
|
{"number": number, "user": {"login": f"author{number}"},
|
|
"head": {"sha": f"sha{number}"}, "body": None}
|
|
for number in range(1, 202)
|
|
]
|
|
pull_requests[-1]["body"] = "Depends on: #12\n- #34\n"
|
|
|
|
def response(*args):
|
|
if args[0] == "api":
|
|
return json.dumps(pull_requests)
|
|
limit = int(args[args.index("-L") + 1])
|
|
return json.dumps([
|
|
{"number": pr["number"], "author": pr["user"], "headRefOid": pr["head"]["sha"],
|
|
"body": pr["body"]} for pr in pull_requests[:limit]
|
|
])
|
|
|
|
with patch.object(pr_overlap, "REPO", "owner/project", create=True), \
|
|
patch.object(pr_overlap, "gh", side_effect=response):
|
|
found = pr_overlap.open_prs("main")
|
|
self.assertEqual(set(found), set(range(1, 202)))
|
|
self.assertEqual(found[201], {
|
|
"ref": "refs/pr/201", "sha": "sha201", "author": "author201", "depends": {12, 34},
|
|
})
|
|
self.assertEqual(found[1]["depends"], set())
|
|
|
|
def test_base_branch_is_encoded_in_the_paginated_request(self):
|
|
with patch.object(pr_overlap, "REPO", "owner/project", create=True), \
|
|
patch.object(pr_overlap, "gh", return_value="[]") as gh:
|
|
self.assertEqual(pr_overlap.open_prs("topic/one&two"), {})
|
|
gh.assert_called_once_with(
|
|
"api", "--paginate",
|
|
"repos/owner/project/pulls?state=open&base=topic%2Fone%26two&per_page=100",
|
|
)
|
|
|
|
def test_single_page_preserves_author_head_and_dependencies(self):
|
|
pull_requests = [{"number": 7, "user": {"login": "contributor"}, "head": {"sha": "head-sha"},
|
|
"body": "Depends on: none\nUnrelated issue #42"}]
|
|
with patch.object(pr_overlap, "REPO", "owner/project", create=True), \
|
|
patch.object(pr_overlap, "gh", return_value=json.dumps(pull_requests)):
|
|
found = pr_overlap.open_prs("main")
|
|
self.assertEqual(found, {
|
|
7: {"ref": "refs/pr/7", "sha": "head-sha", "author": "contributor", "depends": set()},
|
|
})
|
|
|
|
|
|
class FetchTests(unittest.TestCase):
|
|
def test_fetches_each_head_commit_the_api_reported(self):
|
|
prs = {
|
|
7: {"ref": "refs/pr/7", "sha": "a" * 40, "author": "one", "depends": set()},
|
|
9: {"ref": "refs/pr/9", "sha": "b" * 40, "author": "two", "depends": set()},
|
|
}
|
|
with patch.object(pr_overlap, "REPO", "owner/project", create=True), \
|
|
patch.object(pr_overlap, "git", return_value="") as git:
|
|
pr_overlap.fetch("main", prs)
|
|
git.assert_called_once_with(
|
|
"fetch", "--no-tags", "-q", "https://github.com/owner/project.git",
|
|
"+refs/heads/main:refs/pr/base", f"+{'a' * 40}:refs/pr/7", f"+{'b' * 40}:refs/pr/9",
|
|
)
|
|
|
|
def test_batches_large_sets_of_pull_request_heads(self):
|
|
prs = {
|
|
number: {
|
|
"ref": f"refs/pr/{number}", "sha": f"{number:040x}",
|
|
"author": "contributor", "depends": set(),
|
|
}
|
|
for number in range(250)
|
|
}
|
|
with patch.object(pr_overlap, "REPO", "owner/project", create=True), \
|
|
patch.object(pr_overlap, "git", return_value="") as git:
|
|
pr_overlap.fetch("main", prs)
|
|
self.assertEqual(git.call_count, 3)
|
|
self.assertEqual([len(call.args) - 4 for call in git.call_args_list], [100, 100, 51])
|
|
self.assertEqual(git.call_args_list[0].args[4], "+refs/heads/main:refs/pr/base")
|
|
self.assertEqual(git.call_args_list[-1].args[-1], "+00000000000000000000000000000000000000f9:refs/pr/249")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|