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