from django.contrib.auth.models import AnonymousUser, User
from django.http import HttpRequest
from ninja import NinjaAPI
from ninja.security import SessionAuth
from ninja.testing import TestClient


class AsyncSessionAuth(SessionAuth):
    async def __call__(self, request):
        return await self.authenticate(request, None)

    async def authenticate(self, request: HttpRequest, key: str | None) -> User | AnonymousUser | None:
        user = await request.auser()
        return user if user.is_authenticated else None


def test_foobar(api: NinjaAPI):
    @api.get("/foobar", auth=AsyncSessionAuth())
    def foobar(request) -> str:
        return "foobar"

    client = TestClient(api)
    response = client.get("/foobar")
    assert response.status_code == 200
    assert response.json() == "foobar"