Add watchlist command
This commit is contained in:
32
tests/helpers/mocks.py
Normal file
32
tests/helpers/mocks.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import requests
|
||||
import json
|
||||
|
||||
|
||||
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)
|
||||
elif "comments" in args[0]:
|
||||
pr = args[0].split("/")[-2]
|
||||
with open(f"tests/fixtures/comments_{pr}.json") as f:
|
||||
data = f.read()
|
||||
return MockedResponse(data, 200)
|
Reference in New Issue
Block a user