mock_classes.py

import typing
from urllib.parse import urljoin

import httpx
import pytest
from fastapi import status

from alws.app import app
from alws.config import settings
from alws.utils import jwt_utils
from tests.constants import ADMIN_USER_ID


@pytest.mark.anyio
class BaseAsyncTestCase:
    user_id: int = ADMIN_USER_ID
    token: str = ""
    headers: dict = {}
    status_codes = status

    async def make_request(
        self,
        method: str,
        endpoint: str,
        headers: typing.Optional[dict] = None,
        json: typing.Optional[dict] = None,
        files: typing.Optional[dict] = None,
        data: typing.Optional[dict] = None,
        base_url: str = "http://localhost:8080",
    ) -> httpx.Response:
        if not headers:
            headers = {}
        headers.update(self.headers)
        request = httpx.Request(
            method=method,
            url=urljoin(base_url, endpoint),
            headers=headers,
            json=json,
            data=data,
            files=files,
        )
        async with httpx.AsyncClient(
            transport=httpx.ASGITransport(app=app)
        ) as client:
            return await client.send(request)

    @classmethod
    def generate_jwt_token(
        cls,
        user_id: str,
    ) -> str:
        return jwt_utils.generate_JWT_token(
            user_id,
            settings.jwt_secret,
            "HS256",
        )

    @classmethod
    def setup_class(cls):
        cls.token = cls.generate_jwt_token(str(cls.user_id))
        cls.headers.update({
            "Authorization": f"Bearer {cls.token}",
        })

    def get_assertion_message(
        self,
        response_text: str,
        message: str = "",
    ) -> str:
        return f"{message}\n{response_text}"