Make code testable

tests/test_pr.py

@@ -0,0 +1,55 @@
+import unittest
+import unittest.mock
+import requests
+import json
+
+from nixprstatus.pr import commit_in_branch, pr_merge_status
+
+
+def mocked_requests_get(*args, **kwargs):
+    class MockedResponse:
+        def __init__(self, json_data, status_code):
+            self.json_data = json_data
+            self.status_code = status_code
+
+        def json(self):
+            return json.loads(self.json_data)
+
+        def raise_for_status(self):
+            if self.status_code not in [200, 201]:
+                raise requests.exceptions.HTTPError()
+
+    if "pulls" in args[0]:
+        pr = args[0].split("/")[-1]
+        with open(f"tests/fixtures/pulls_{pr}.json") as f:
+            data = f.read()
+        return MockedResponse(data, 200)
+    elif "compare" in args[0]:
+        branch, commit_sha = args[0].split("/")[-1].split("...")
+        with open(f"tests/fixtures/compare_{branch}_{commit_sha}.json") as f:
+            data = f.read()
+        return MockedResponse(data, 200)
+
+
+class TestPRMergeStatus(unittest.TestCase):
+    @unittest.mock.patch("requests.get", side_effect=mocked_requests_get)
+    def test_pr_merge_status_345583(self, mock_get):
+        pr = 345583
+        branches = ["master", "nixos-unstable", "nixos-24.05"]
+
+        res = pr_merge_status(pr, branches)
+        self.assertTrue(res.merged)
+        self.assertTrue(res.branches["master"])
+        self.assertTrue(res.branches["nixos-unstable"])
+        self.assertFalse(res.branches["nixos-24.05"])
+
+
+class TestCommitInBranch(unittest.TestCase):
+    @unittest.mock.patch("requests.get", side_effect=mocked_requests_get)
+    def test_commit_in_branch_2c5fac3e(self, mock_get):
+        commit_sha = "2c5fac3edf2d00d948253e392ec1604b29b38f14"
+
+        res = commit_in_branch(commit_sha, "master")
+        self.assertTrue(res)
+        res = commit_in_branch(commit_sha, "nixos-24.05")
+        self.assertFalse(res)