test_fetch_github_info.py

importjson

importrequests

from .fetch_github_infoimportAUTHENTICATED_USER_ENDPOINT, fetch_github_info


deftest_fetch_github_info(monkeypatch):
classFakeResponse:
def__init__(self, content) ->None:
assertisinstance(content, (bytes, str))
self.content=content

defjson(self):
returnjson.loads(self.content)

defmock_response(*args, **kwargs):
assertargs[0] ==AUTHENTICATED_USER_ENDPOINT
assert"Authorization"inkwargs["headers"]
assertkwargs["headers"]["Authorization"].startswith("token ")
assert"Accept"inkwargs["headers"]
returnFakeResponse(b'{"login":"test","id":1}')

monkeypatch.setattr(requests, "get", mock_response)
result=fetch_github_info("token")
assertresult["login"] =="test"
assertresult["id"] ==1