mirror
/
tiangolo.fastapi
mirror of https://github.com/tiangolo/fastapi
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

Simplify tests for schema_extra_example #13150

pull/13506/head
Valentyn Druzhynin 3 weeks ago
parent
4e40e1e85d
commit
8938c4d517
2 changed files with 5 additions and 140 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 10
      tests/test_tutorial/test_schema_extra_example/test_tutorial001.py
  2. 135
      tests/test_tutorial/test_schema_extra_example/test_tutorial001_pv1.py

10
tests/test_tutorial/test_schema_extra_example/test_tutorial001.py
View File

@ -3,14 +3,16 @@ import importlib
import pytest import pytest
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from ...utils import needs_py310, needs_pydanticv2 from ...utils import needs_py310, needs_pydanticv1, needs_pydanticv2
@pytest.fixture( @pytest.fixture(
name="client", name="client",
params=[ params=[
"tutorial001", pytest.param("tutorial001", marks=[needs_pydanticv2]),
pytest.param("tutorial001_py310", marks=needs_py310), pytest.param("tutorial001_py310", marks=[needs_py310, needs_pydanticv2]),
pytest.param("tutorial001_pv1", marks=[needs_pydanticv1]),
pytest.param("tutorial001_pv1_py310", marks=[needs_py310, needs_pydanticv1]),
], ],
) )
def get_client(request: pytest.FixtureRequest): def get_client(request: pytest.FixtureRequest):
@ -20,7 +22,6 @@ def get_client(request: pytest.FixtureRequest):
return client return client
@needs_pydanticv2
def test_post_body_example(client: TestClient): def test_post_body_example(client: TestClient):
response = client.put( response = client.put(
"/items/5", "/items/5",
@ -34,7 +35,6 @@ def test_post_body_example(client: TestClient):
assert response.status_code == 200 assert response.status_code == 200
@needs_pydanticv2
def test_openapi_schema(client: TestClient): def test_openapi_schema(client: TestClient):
response = client.get("/openapi.json") response = client.get("/openapi.json")
assert response.status_code == 200, response.text assert response.status_code == 200, response.text

135
tests/test_tutorial/test_schema_extra_example/test_tutorial001_pv1.py
View File

@ -1,135 +0,0 @@
import importlib
import pytest
from fastapi.testclient import TestClient
from ...utils import needs_py310, needs_pydanticv1
@pytest.fixture(
name="client",
params=[
"tutorial001_pv1",
pytest.param("tutorial001_pv1_py310", marks=needs_py310),
],
)
def get_client(request: pytest.FixtureRequest):
mod = importlib.import_module(f"docs_src.schema_extra_example.{request.param}")
client = TestClient(mod.app)
return client
@needs_pydanticv1
def test_post_body_example(client: TestClient):
response = client.put(
"/items/5",
json={
"name": "Foo",
"description": "A very nice Item",
"price": 35.4,
"tax": 3.2,
},
)
assert response.status_code == 200
@needs_pydanticv1
def test_openapi_schema(client: TestClient):
response = client.get("/openapi.json")
assert response.status_code == 200, response.text
# insert_assert(response.json())
assert response.json() == {
"openapi": "3.1.0",
"info": {"title": "FastAPI", "version": "0.1.0"},
"paths": {
"/items/{item_id}": {
"put": {
"summary": "Update Item",
"operationId": "update_item_items__item_id__put",
"parameters": [
{
"required": True,
"schema": {"type": "integer", "title": "Item Id"},
"name": "item_id",
"in": "path",
}
],
"requestBody": {
"content": {
"application/json": {
"schema": {"$ref": "#/components/schemas/Item"}
}
},
"required": True,
},
"responses": {
"200": {
"description": "Successful Response",
"content": {"application/json": {"schema": {}}},
},
"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",
},
"Item": {
"properties": {
"name": {"type": "string", "title": "Name"},
"description": {"type": "string", "title": "Description"},
"price": {"type": "number", "title": "Price"},
"tax": {"type": "number", "title": "Tax"},
},
"type": "object",
"required": ["name", "price"],
"title": "Item",
"examples": [
{
"name": "Foo",
"description": "A very nice Item",
"price": 35.4,
"tax": 3.2,
}
],
},
"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",
},
}
},
}
Write Preview
Loading…
Cancel
Save