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 response_model (#14062)

pull/9867/merge
山崎ヒカル 2 days ago
committed by GitHub
parent
c8a29944f4
commit
11d424c3dc
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
4 changed files with 32 additions and 180 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. 23
      tests/test_tutorial/test_response_model/test_tutorial003.py
  2. 17
      tests/test_tutorial/test_response_model/test_tutorial003_04.py
  3. 12
      tests/test_tutorial/test_response_model/test_tutorial003_04_py310.py
  4. 160
      tests/test_tutorial/test_response_model/test_tutorial003_py310.py

23
tests/test_tutorial/test_response_model/test_tutorial003.py
View File

@ -1,12 +1,27 @@
import importlib
import pytest
from dirty_equals import IsDict, IsOneOf from dirty_equals import IsDict, IsOneOf
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from docs_src.response_model.tutorial003 import app from ...utils import needs_py310
@pytest.fixture(
name="client",
params=[
"tutorial003",
pytest.param("tutorial003_py310", marks=needs_py310),
],
)
def get_client(request: pytest.FixtureRequest):
mod = importlib.import_module(f"docs_src.response_model.{request.param}")
client = TestClient(app) client = TestClient(mod.app)
return client
def test_post_user(): def test_post_user(client: TestClient):
response = client.post( response = client.post(
"/user/", "/user/",
json={ json={
@ -24,7 +39,7 @@ def test_post_user():
} }
def test_openapi_schema(): 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
assert response.json() == { assert response.json() == {

17
tests/test_tutorial/test_response_model/test_tutorial003_04.py
View File

@ -1,9 +1,18 @@
import importlib
import pytest import pytest
from fastapi.exceptions import FastAPIError from fastapi.exceptions import FastAPIError
from ...utils import needs_py310
def test_invalid_response_model():
with pytest.raises(FastAPIError):
from docs_src.response_model.tutorial003_04 import app
assert app # pragma: no cover @pytest.mark.parametrize(
"module_name",
[
"tutorial003_04",
pytest.param("tutorial003_04_py310", marks=needs_py310),
],
)
def test_invalid_response_model(module_name: str) -> None:
with pytest.raises(FastAPIError):
importlib.import_module(f"docs_src.response_model.{module_name}")

12
tests/test_tutorial/test_response_model/test_tutorial003_04_py310.py
View File

@ -1,12 +0,0 @@
import pytest
from fastapi.exceptions import FastAPIError
from ...utils import needs_py310
@needs_py310
def test_invalid_response_model():
with pytest.raises(FastAPIError):
from docs_src.response_model.tutorial003_04_py310 import app
assert app # pragma: no cover

160
tests/test_tutorial/test_response_model/test_tutorial003_py310.py
View File

@ -1,160 +0,0 @@
import pytest
from dirty_equals import IsDict, IsOneOf
from fastapi.testclient import TestClient
from ...utils import needs_py310
@pytest.fixture(name="client")
def get_client():
from docs_src.response_model.tutorial003_py310 import app
client = TestClient(app)
return client
@needs_py310
def test_post_user(client: TestClient):
response = client.post(
"/user/",
json={
"username": "foo",
"password": "fighter",
"email": "foo@example.com",
"full_name": "Grave Dohl",
},
)
assert response.status_code == 200, response.text
assert response.json() == {
"username": "foo",
"email": "foo@example.com",
"full_name": "Grave Dohl",
}
@needs_py310
def test_openapi_schema(client: TestClient):
response = client.get("/openapi.json")
assert response.status_code == 200, response.text
assert response.json() == {
"openapi": "3.1.0",
"info": {"title": "FastAPI", "version": "0.1.0"},
"paths": {
"/user/": {
"post": {
"responses": {
"200": {
"description": "Successful Response",
"content": {
"application/json": {
"schema": {"$ref": "#/components/schemas/UserOut"}
}
},
},
"422": {
"description": "Validation Error",
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/HTTPValidationError"
}
}
},
},
},
"summary": "Create User",
"operationId": "create_user_user__post",
"requestBody": {
"content": {
"application/json": {
"schema": {"$ref": "#/components/schemas/UserIn"}
}
},
"required": True,
},
}
}
},
"components": {
"schemas": {
"UserOut": {
"title": "UserOut",
"required": IsOneOf(
["username", "email", "full_name"],
# TODO: remove when deprecating Pydantic v1
["username", "email"],
),
"type": "object",
"properties": {
"username": {"title": "Username", "type": "string"},
"email": {
"title": "Email",
"type": "string",
"format": "email",
},
"full_name": IsDict(
{
"title": "Full Name",
"anyOf": [{"type": "string"}, {"type": "null"}],
}
)
| IsDict(
# TODO: remove when deprecating Pydantic v1
{"title": "Full Name", "type": "string"}
),
},
},
"UserIn": {
"title": "UserIn",
"required": ["username", "password", "email"],
"type": "object",
"properties": {
"username": {"title": "Username", "type": "string"},
"password": {"title": "Password", "type": "string"},
"email": {
"title": "Email",
"type": "string",
"format": "email",
},
"full_name": IsDict(
{
"title": "Full Name",
"anyOf": [{"type": "string"}, {"type": "null"}],
}
)
| IsDict(
# TODO: remove when deprecating Pydantic v1
{"title": "Full Name", "type": "string"}
),
},
},
"ValidationError": {
"title": "ValidationError",
"required": ["loc", "msg", "type"],
"type": "object",
"properties": {
"loc": {
"title": "Location",
"type": "array",
"items": {
"anyOf": [{"type": "string"}, {"type": "integer"}]
},
},
"msg": {"title": "Message", "type": "string"},
"type": {"title": "Error Type", "type": "string"},
},
},
"HTTPValidationError": {
"title": "HTTPValidationError",
"type": "object",
"properties": {
"detail": {
"title": "Detail",
"type": "array",
"items": {"$ref": "#/components/schemas/ValidationError"},
}
},
},
}
},
}
Write Preview
Loading…
Cancel
Save