Python-algorithm/web_programming/test_fetch_github_info.py at master · rasata/Python-algorithm · GitHub

27 lines (20 loc) · 853 Bytes

import json

import httpx

from .fetch_github_info import AUTHENTICATED_USER_ENDPOINT, fetch_github_info

def test_fetch_github_info(monkeypatch):

class FakeResponse:

def __init__(self, content) -> None:

assert isinstance(content, (bytes, str))

self.content = content

def json(self):

return json.loads(self.content)

def mock_response(*args, **kwargs):

assert args[0] == AUTHENTICATED_USER_ENDPOINT

assert "Authorization" in kwargs["headers"]

assert kwargs["headers"]["Authorization"].startswith("token ")

assert "Accept" in kwargs["headers"]

return FakeResponse(b'{"login":"test","id":1}')

monkeypatch.setattr(httpx, "get", mock_response)

result = fetch_github_info("token")

assert result["login"] == "test"

assert result["id"] == 1

Expand file tree