committed by
GitHub
2 changed files with 142 additions and 2 deletions
@ -0,0 +1,131 @@ |
|||
# Ref: https://github.com/fastapi/fastapi/issues/14454 |
|||
|
|||
from typing import Optional |
|||
|
|||
from fastapi import APIRouter, FastAPI, Security |
|||
from fastapi.security import OAuth2AuthorizationCodeBearer |
|||
from fastapi.testclient import TestClient |
|||
from inline_snapshot import snapshot |
|||
|
|||
oauth2_scheme = OAuth2AuthorizationCodeBearer( |
|||
authorizationUrl="authorize", |
|||
tokenUrl="token", |
|||
auto_error=True, |
|||
scopes={"read": "Read access", "write": "Write access"}, |
|||
) |
|||
|
|||
app = FastAPI(dependencies=[Security(oauth2_scheme)]) |
|||
|
|||
|
|||
@app.get("/") |
|||
async def root(): |
|||
return {"message": "Hello World"} |
|||
|
|||
|
|||
router = APIRouter(dependencies=[Security(oauth2_scheme, scopes=["read"])]) |
|||
|
|||
|
|||
@router.get("/items/") |
|||
async def read_items(token: Optional[str] = Security(oauth2_scheme)): |
|||
return {"token": token} |
|||
|
|||
|
|||
@router.post("/items/") |
|||
async def create_item( |
|||
token: Optional[str] = Security(oauth2_scheme, scopes=["read", "write"]), |
|||
): |
|||
return {"token": token} |
|||
|
|||
|
|||
app.include_router(router) |
|||
|
|||
client = TestClient(app) |
|||
|
|||
|
|||
def test_root(): |
|||
response = client.get("/", headers={"Authorization": "Bearer testtoken"}) |
|||
assert response.status_code == 200, response.text |
|||
assert response.json() == {"message": "Hello World"} |
|||
|
|||
|
|||
def test_read_token(): |
|||
response = client.get("/items/", headers={"Authorization": "Bearer testtoken"}) |
|||
assert response.status_code == 200, response.text |
|||
assert response.json() == {"token": "testtoken"} |
|||
|
|||
|
|||
def test_create_token(): |
|||
response = client.post("/items/", headers={"Authorization": "Bearer testtoken"}) |
|||
assert response.status_code == 200, response.text |
|||
assert response.json() == {"token": "testtoken"} |
|||
|
|||
|
|||
def test_openapi_schema(): |
|||
response = client.get("/openapi.json") |
|||
assert response.status_code == 200, response.text |
|||
assert response.json() == snapshot( |
|||
{ |
|||
"openapi": "3.1.0", |
|||
"info": {"title": "FastAPI", "version": "0.1.0"}, |
|||
"paths": { |
|||
"/": { |
|||
"get": { |
|||
"summary": "Root", |
|||
"operationId": "root__get", |
|||
"responses": { |
|||
"200": { |
|||
"description": "Successful Response", |
|||
"content": {"application/json": {"schema": {}}}, |
|||
} |
|||
}, |
|||
"security": [{"OAuth2AuthorizationCodeBearer": []}], |
|||
} |
|||
}, |
|||
"/items/": { |
|||
"get": { |
|||
"summary": "Read Items", |
|||
"operationId": "read_items_items__get", |
|||
"responses": { |
|||
"200": { |
|||
"description": "Successful Response", |
|||
"content": {"application/json": {"schema": {}}}, |
|||
} |
|||
}, |
|||
"security": [ |
|||
{"OAuth2AuthorizationCodeBearer": ["read"]}, |
|||
], |
|||
}, |
|||
"post": { |
|||
"summary": "Create Item", |
|||
"operationId": "create_item_items__post", |
|||
"responses": { |
|||
"200": { |
|||
"description": "Successful Response", |
|||
"content": {"application/json": {"schema": {}}}, |
|||
} |
|||
}, |
|||
"security": [ |
|||
{"OAuth2AuthorizationCodeBearer": ["read", "write"]}, |
|||
], |
|||
}, |
|||
}, |
|||
}, |
|||
"components": { |
|||
"securitySchemes": { |
|||
"OAuth2AuthorizationCodeBearer": { |
|||
"type": "oauth2", |
|||
"flows": { |
|||
"authorizationCode": { |
|||
"scopes": { |
|||
"read": "Read access", |
|||
"write": "Write access", |
|||
}, |
|||
"authorizationUrl": "authorize", |
|||
"tokenUrl": "token", |
|||
} |
|||
}, |
|||
} |
|||
} |
|||
}, |
|||
} |
|||
) |
|||
Loading…
Reference in new issue