Browse Source

Update tests for Pydantic v2 and v1

pull/14246/head
Sebastián Ramírez 9 months ago
parent
commit
5c6b8432df
  1. 36
      tests/test_no_schema_split.py

36
tests/test_no_schema_split.py

@ -6,6 +6,8 @@ from fastapi.testclient import TestClient
from inline_snapshot import snapshot from inline_snapshot import snapshot
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from tests.utils import pydantic_snapshot
class MessageEventType(str, Enum): class MessageEventType(str, Enum):
alpha = "alpha" alpha = "alpha"
@ -112,21 +114,47 @@ def test_openapi_schema():
}, },
"MessageEvent": { "MessageEvent": {
"properties": { "properties": {
"event_type": { "event_type": pydantic_snapshot(
v2=snapshot(
{
"$ref": "#/components/schemas/MessageEventType", "$ref": "#/components/schemas/MessageEventType",
"default": "alpha", "default": "alpha",
}, }
),
v1=snapshot(
{
"allOf": [
{
"$ref": "#/components/schemas/MessageEventType"
}
],
"default": "alpha",
}
),
),
"output": {"type": "string", "title": "Output"}, "output": {"type": "string", "title": "Output"},
}, },
"type": "object", "type": "object",
"required": ["output"], "required": ["output"],
"title": "MessageEvent", "title": "MessageEvent",
}, },
"MessageEventType": { "MessageEventType": pydantic_snapshot(
v2=snapshot(
{
"type": "string", "type": "string",
"enum": ["alpha", "beta"], "enum": ["alpha", "beta"],
"title": "MessageEventType", "title": "MessageEventType",
}, }
),
v1=snapshot(
{
"type": "string",
"enum": ["alpha", "beta"],
"title": "MessageEventType",
"description": "An enumeration.",
}
),
),
"MessageOutput": { "MessageOutput": {
"properties": { "properties": {
"body": {"type": "string", "title": "Body", "default": ""}, "body": {"type": "string", "title": "Body", "default": ""},

Loading…
Cancel
Save