56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
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)
|