4 changed files with 38 additions and 9 deletions
@ -0,0 +1,29 @@ |
|||
from fastapi import Depends, FastAPI |
|||
from fastapi.security import HTTPBasic |
|||
from fastapi.testclient import TestClient |
|||
|
|||
def test_default_realm_is_included(): |
|||
app = FastAPI() |
|||
security = HTTPBasic() |
|||
|
|||
@app.get("/protected") |
|||
def protected(_: str = Depends(security)): |
|||
return {"ok": True} |
|||
|
|||
client = TestClient(app) |
|||
response = client.get("/protected") |
|||
assert response.status_code == 401 |
|||
assert response.headers["www-authenticate"] == 'Basic realm="fastapi"' |
|||
|
|||
def test_custom_realm_is_respected(): |
|||
app = FastAPI() |
|||
security = HTTPBasic(realm="custom") |
|||
|
|||
@app.get("/protected") |
|||
def protected(_: str = Depends(security)): |
|||
return {"ok": True} |
|||
|
|||
client = TestClient(app) |
|||
response = client.get("/protected") |
|||
assert response.status_code == 401 |
|||
assert response.headers["www-authenticate"] == 'Basic realm="custom"' |
|||
Loading…
Reference in new issue