2 changed files with 225 additions and 0 deletions
@ -0,0 +1,225 @@ |
|||||
|
import importlib |
||||
|
|
||||
|
import pytest |
||||
|
from fastapi.testclient import TestClient |
||||
|
from inline_snapshot import snapshot |
||||
|
|
||||
|
from tests.utils import needs_py310 |
||||
|
|
||||
|
|
||||
|
@pytest.fixture( |
||||
|
name="client", |
||||
|
params=[pytest.param("tutorial001_py310", marks=needs_py310)], |
||||
|
) |
||||
|
def get_client(request: pytest.FixtureRequest): |
||||
|
mod = importlib.import_module(f"docs_src.json_base64_bytes.{request.param}") |
||||
|
|
||||
|
client = TestClient(mod.app) |
||||
|
return client |
||||
|
|
||||
|
|
||||
|
def test_post_data(client: TestClient): |
||||
|
response = client.post( |
||||
|
"/data", |
||||
|
json={ |
||||
|
"description": "A file", |
||||
|
"data": "SGVsbG8sIFdvcmxkIQ==", |
||||
|
}, |
||||
|
) |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == {"description": "A file", "content": "Hello, World!"} |
||||
|
|
||||
|
|
||||
|
def test_get_data(client: TestClient): |
||||
|
response = client.get("/data") |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == {"description": "A plumbus", "data": "aGVsbG8="} |
||||
|
|
||||
|
|
||||
|
def test_post_data_in_out(client: TestClient): |
||||
|
response = client.post( |
||||
|
"/data-in-out", |
||||
|
json={ |
||||
|
"description": "A plumbus", |
||||
|
"data": "SGVsbG8sIFdvcmxkIQ==", |
||||
|
}, |
||||
|
) |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == { |
||||
|
"description": "A plumbus", |
||||
|
"data": "SGVsbG8sIFdvcmxkIQ==", |
||||
|
} |
||||
|
|
||||
|
|
||||
|
def test_openapi_schema(client: TestClient): |
||||
|
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": { |
||||
|
"/data": { |
||||
|
"get": { |
||||
|
"summary": "Get Data", |
||||
|
"operationId": "get_data_data_get", |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"$ref": "#/components/schemas/DataOutput" |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
"post": { |
||||
|
"summary": "Post Data", |
||||
|
"operationId": "post_data_data_post", |
||||
|
"requestBody": { |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": {"$ref": "#/components/schemas/DataInput"} |
||||
|
} |
||||
|
}, |
||||
|
"required": True, |
||||
|
}, |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": {"application/json": {"schema": {}}}, |
||||
|
}, |
||||
|
"422": { |
||||
|
"description": "Validation Error", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"$ref": "#/components/schemas/HTTPValidationError" |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
"/data-in-out": { |
||||
|
"post": { |
||||
|
"summary": "Post Data In Out", |
||||
|
"operationId": "post_data_in_out_data_in_out_post", |
||||
|
"requestBody": { |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"$ref": "#/components/schemas/DataInputOutput" |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
"required": True, |
||||
|
}, |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"$ref": "#/components/schemas/DataInputOutput" |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
"422": { |
||||
|
"description": "Validation Error", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"$ref": "#/components/schemas/HTTPValidationError" |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
"components": { |
||||
|
"schemas": { |
||||
|
"DataInput": { |
||||
|
"properties": { |
||||
|
"description": {"type": "string", "title": "Description"}, |
||||
|
"data": { |
||||
|
"type": "string", |
||||
|
"contentEncoding": "base64", |
||||
|
"contentMediaType": "application/octet-stream", |
||||
|
"title": "Data", |
||||
|
}, |
||||
|
}, |
||||
|
"type": "object", |
||||
|
"required": ["description", "data"], |
||||
|
"title": "DataInput", |
||||
|
}, |
||||
|
"DataInputOutput": { |
||||
|
"properties": { |
||||
|
"description": {"type": "string", "title": "Description"}, |
||||
|
"data": { |
||||
|
"type": "string", |
||||
|
"contentEncoding": "base64", |
||||
|
"contentMediaType": "application/octet-stream", |
||||
|
"title": "Data", |
||||
|
}, |
||||
|
}, |
||||
|
"type": "object", |
||||
|
"required": ["description", "data"], |
||||
|
"title": "DataInputOutput", |
||||
|
}, |
||||
|
"DataOutput": { |
||||
|
"properties": { |
||||
|
"description": {"type": "string", "title": "Description"}, |
||||
|
"data": { |
||||
|
"type": "string", |
||||
|
"contentEncoding": "base64", |
||||
|
"contentMediaType": "application/octet-stream", |
||||
|
"title": "Data", |
||||
|
}, |
||||
|
}, |
||||
|
"type": "object", |
||||
|
"required": ["description", "data"], |
||||
|
"title": "DataOutput", |
||||
|
}, |
||||
|
"HTTPValidationError": { |
||||
|
"properties": { |
||||
|
"detail": { |
||||
|
"items": { |
||||
|
"$ref": "#/components/schemas/ValidationError" |
||||
|
}, |
||||
|
"type": "array", |
||||
|
"title": "Detail", |
||||
|
} |
||||
|
}, |
||||
|
"type": "object", |
||||
|
"title": "HTTPValidationError", |
||||
|
}, |
||||
|
"ValidationError": { |
||||
|
"properties": { |
||||
|
"ctx": {"title": "Context", "type": "object"}, |
||||
|
"input": {"title": "Input"}, |
||||
|
"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