Added ref login
This commit is contained in:
@@ -2,7 +2,7 @@
|
||||
from typing import AsyncGenerator, Generator
|
||||
|
||||
import pytest
|
||||
from app.core.auth import get_current_user, get_optional_user
|
||||
from app.core.auth import get_admin_user, get_optional_user
|
||||
from app.database import Base, get_db
|
||||
|
||||
# Import your app and models
|
||||
@@ -52,7 +52,7 @@ async def client(db: Session) -> AsyncGenerator[AsyncClient, None]:
|
||||
pass
|
||||
|
||||
# Strict Auth: Always requires a token (simulated by header presence)
|
||||
def override_get_current_user(request: Request):
|
||||
def override_get_admin_user(request: Request):
|
||||
if "Authorization" not in request.headers:
|
||||
# Let FastAPI raise the 401 naturally if header is missing
|
||||
raise pytest.skip("Auth header missing in strict auth test")
|
||||
@@ -65,7 +65,7 @@ async def client(db: Session) -> AsyncGenerator[AsyncClient, None]:
|
||||
return None
|
||||
|
||||
app.dependency_overrides[get_db] = override_get_db
|
||||
app.dependency_overrides[get_current_user] = override_get_current_user
|
||||
app.dependency_overrides[get_admin_user] = override_get_admin_user
|
||||
app.dependency_overrides[get_optional_user] = override_get_optional_user
|
||||
|
||||
async with AsyncClient(
|
||||
|
||||
Reference in New Issue
Block a user