Motov Yurii
9 months ago
committed by
GitHub
GitHub
13 changed files with 268 additions and 71 deletions
|
After Width: | Height: | Size: 6.3 KiB |
|
After Width: | Height: | Size: 10 KiB |
@ -0,0 +1,203 @@ |
|||
# Test with parts from, and to verify the report in: |
|||
# https://github.com/fastapi/fastapi/discussions/14177 |
|||
# Made an issue in: |
|||
# https://github.com/fastapi/fastapi/issues/14247 |
|||
from enum import Enum |
|||
from typing import List |
|||
|
|||
from fastapi import FastAPI |
|||
from fastapi.testclient import TestClient |
|||
from inline_snapshot import snapshot |
|||
from pydantic import BaseModel, Field |
|||
|
|||
from tests.utils import pydantic_snapshot |
|||
|
|||
|
|||
class MessageEventType(str, Enum): |
|||
alpha = "alpha" |
|||
beta = "beta" |
|||
|
|||
|
|||
class MessageEvent(BaseModel): |
|||
event_type: MessageEventType = Field(default=MessageEventType.alpha) |
|||
output: str |
|||
|
|||
|
|||
class MessageOutput(BaseModel): |
|||
body: str = "" |
|||
events: List[MessageEvent] = [] |
|||
|
|||
|
|||
class Message(BaseModel): |
|||
input: str |
|||
output: MessageOutput |
|||
|
|||
|
|||
app = FastAPI(title="Minimal FastAPI App", version="1.0.0") |
|||
|
|||
|
|||
@app.post("/messages", response_model=Message) |
|||
async def create_message(input_message: str) -> Message: |
|||
return Message( |
|||
input=input_message, |
|||
output=MessageOutput(body=f"Processed: {input_message}"), |
|||
) |
|||
|
|||
|
|||
client = TestClient(app) |
|||
|
|||
|
|||
def test_create_message(): |
|||
response = client.post("/messages", params={"input_message": "Hello"}) |
|||
assert response.status_code == 200, response.text |
|||
assert response.json() == { |
|||
"input": "Hello", |
|||
"output": {"body": "Processed: Hello", "events": []}, |
|||
} |
|||
|
|||
|
|||
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": "Minimal FastAPI App", "version": "1.0.0"}, |
|||
"paths": { |
|||
"/messages": { |
|||
"post": { |
|||
"summary": "Create Message", |
|||
"operationId": "create_message_messages_post", |
|||
"parameters": [ |
|||
{ |
|||
"name": "input_message", |
|||
"in": "query", |
|||
"required": True, |
|||
"schema": {"type": "string", "title": "Input Message"}, |
|||
} |
|||
], |
|||
"responses": { |
|||
"200": { |
|||
"description": "Successful Response", |
|||
"content": { |
|||
"application/json": { |
|||
"schema": { |
|||
"$ref": "#/components/schemas/Message" |
|||
} |
|||
} |
|||
}, |
|||
}, |
|||
"422": { |
|||
"description": "Validation Error", |
|||
"content": { |
|||
"application/json": { |
|||
"schema": { |
|||
"$ref": "#/components/schemas/HTTPValidationError" |
|||
} |
|||
} |
|||
}, |
|||
}, |
|||
}, |
|||
} |
|||
} |
|||
}, |
|||
"components": { |
|||
"schemas": { |
|||
"HTTPValidationError": { |
|||
"properties": { |
|||
"detail": { |
|||
"items": { |
|||
"$ref": "#/components/schemas/ValidationError" |
|||
}, |
|||
"type": "array", |
|||
"title": "Detail", |
|||
} |
|||
}, |
|||
"type": "object", |
|||
"title": "HTTPValidationError", |
|||
}, |
|||
"Message": { |
|||
"properties": { |
|||
"input": {"type": "string", "title": "Input"}, |
|||
"output": {"$ref": "#/components/schemas/MessageOutput"}, |
|||
}, |
|||
"type": "object", |
|||
"required": ["input", "output"], |
|||
"title": "Message", |
|||
}, |
|||
"MessageEvent": { |
|||
"properties": { |
|||
"event_type": pydantic_snapshot( |
|||
v2=snapshot( |
|||
{ |
|||
"$ref": "#/components/schemas/MessageEventType", |
|||
"default": "alpha", |
|||
} |
|||
), |
|||
v1=snapshot( |
|||
{ |
|||
"allOf": [ |
|||
{ |
|||
"$ref": "#/components/schemas/MessageEventType" |
|||
} |
|||
], |
|||
"default": "alpha", |
|||
} |
|||
), |
|||
), |
|||
"output": {"type": "string", "title": "Output"}, |
|||
}, |
|||
"type": "object", |
|||
"required": ["output"], |
|||
"title": "MessageEvent", |
|||
}, |
|||
"MessageEventType": pydantic_snapshot( |
|||
v2=snapshot( |
|||
{ |
|||
"type": "string", |
|||
"enum": ["alpha", "beta"], |
|||
"title": "MessageEventType", |
|||
} |
|||
), |
|||
v1=snapshot( |
|||
{ |
|||
"type": "string", |
|||
"enum": ["alpha", "beta"], |
|||
"title": "MessageEventType", |
|||
"description": "An enumeration.", |
|||
} |
|||
), |
|||
), |
|||
"MessageOutput": { |
|||
"properties": { |
|||
"body": {"type": "string", "title": "Body", "default": ""}, |
|||
"events": { |
|||
"items": {"$ref": "#/components/schemas/MessageEvent"}, |
|||
"type": "array", |
|||
"title": "Events", |
|||
"default": [], |
|||
}, |
|||
}, |
|||
"type": "object", |
|||
"title": "MessageOutput", |
|||
}, |
|||
"ValidationError": { |
|||
"properties": { |
|||
"loc": { |
|||
"items": { |
|||
"anyOf": [{"type": "string"}, {"type": "integer"}] |
|||
}, |
|||
"type": "array", |
|||
"title": "Location", |
|||
}, |
|||
"msg": {"type": "string", "title": "Message"}, |
|||
"type": {"type": "string", "title": "Error Type"}, |
|||
}, |
|||
"type": "object", |
|||
"required": ["loc", "msg", "type"], |
|||
"title": "ValidationError", |
|||
}, |
|||
} |
|||
}, |
|||
} |
|||
) |
|||
Loading…
Reference in new issue