Coverage for tests/test_security_openid_connect.py: 100%
32 statements
« prev ^ index » next coverage.py v7.6.1, created at 2025-12-04 08:29 +0000
1from fastapi import Depends, FastAPI, Security 1abcdefg
2from fastapi.security.open_id_connect_url import OpenIdConnect 1abcdefg
3from fastapi.testclient import TestClient 1abcdefg
4from pydantic import BaseModel 1abcdefg
6app = FastAPI() 1abcdefg
8oid = OpenIdConnect(openIdConnectUrl="/openid") 1abcdefg
11class User(BaseModel): 1abcdefg
12 username: str 1abcdefg
15def get_current_user(oauth_header: str = Security(oid)): 1abcdefg
16 user = User(username=oauth_header) 1hijklmnopqrstu
17 return user 1hijklmnopqrstu
20@app.get("/users/me") 1abcdefg
21def read_current_user(current_user: User = Depends(get_current_user)): 1abcdefg
22 return current_user 1hijklmnopqrstu
25client = TestClient(app) 1abcdefg
28def test_security_oauth2(): 1abcdefg
29 response = client.get("/users/me", headers={"Authorization": "Bearer footokenbar"}) 1hjlnprt
30 assert response.status_code == 200, response.text 1hjlnprt
31 assert response.json() == {"username": "Bearer footokenbar"} 1hjlnprt
34def test_security_oauth2_password_other_header(): 1abcdefg
35 response = client.get("/users/me", headers={"Authorization": "Other footokenbar"}) 1ikmoqsu
36 assert response.status_code == 200, response.text 1ikmoqsu
37 assert response.json() == {"username": "Other footokenbar"} 1ikmoqsu
40def test_security_oauth2_password_bearer_no_header(): 1abcdefg
41 response = client.get("/users/me") 1vwxyzAB
42 assert response.status_code == 401, response.text 1vwxyzAB
43 assert response.json() == {"detail": "Not authenticated"} 1vwxyzAB
44 assert response.headers["WWW-Authenticate"] == "Bearer" 1vwxyzAB
47def test_openapi_schema(): 1abcdefg
48 response = client.get("/openapi.json") 1CDEFGHI
49 assert response.status_code == 200, response.text 1CDEFGHI
50 assert response.json() == { 1CDEFGHI
51 "openapi": "3.1.0",
52 "info": {"title": "FastAPI", "version": "0.1.0"},
53 "paths": {
54 "/users/me": {
55 "get": {
56 "responses": {
57 "200": {
58 "description": "Successful Response",
59 "content": {"application/json": {"schema": {}}},
60 }
61 },
62 "summary": "Read Current User",
63 "operationId": "read_current_user_users_me_get",
64 "security": [{"OpenIdConnect": []}],
65 }
66 }
67 },
68 "components": {
69 "securitySchemes": {
70 "OpenIdConnect": {
71 "type": "openIdConnect",
72 "openIdConnectUrl": "/openid",
73 }
74 }
75 },
76 }