mock_responses.py

"""Simple mock responses definitions."""
from unittest import mock


class MockResponse:
    """Class for mock request response."""

    def __init__(
        self,
        json_data,
        status_code,
        headers={},
        raw_data=None,
        raise_error=None,
    ):
        """Initialize mock get response."""
        self.json_data = json_data
        self.status = status_code
        self.raw_data = raw_data
        self.reason = "foobar"
        self.headers = headers
        self.read = mock.AsyncMock(return_value=self.raw_data)
        self.raise_error = raise_error
        self.text = mock.AsyncMock(return_vlaue="some text")

    async def json(self):
        """Return json data from get_request."""
        if self.raise_error:
            raise self.raise_error("I'm broken", "")
        return self.json_data

    def get(self, name):
        """Return field for json."""
        return self.json_data[name]