opensource/dify
3
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-10-16 15:35:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(api): build the same-named schema model with explicit fields

Browse Source 
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Stephen Zhou
2026-10-16 06:11:54 +08:00
co-authored by Claude Opus 5.5
parent 03fa068bf9
commit 68002efd6a
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/tests/unit_tests/controllers/common/test_schema.py
+4 -8
View File
@@ -127,14 +127,10 @@ def test_register_schema_model_promotes_nested_pydantic_definitions():
assert called_schemas["ChildModel"] == parent_schema["$defs"]["ChildModel"]
def _make_same_named_model(fields: dict[str, type]) -> type[BaseModel]:
def _make_same_named_model() -> type[BaseModel]:
"""Build a model whose qualified name differs from UserModel but whose bare name matches."""
return create_model(
"UserModel",
__module__="other_module",
**{name: (annotation, ...) for name, annotation in fields.items()},
)
return create_model("UserModel", __module__="other_module", email=(str, ...))
def test_register_schema_model_allows_reregistering_the_same_model():
@@ -155,7 +151,7 @@ def test_register_schema_model_rejects_different_model_with_same_name():
register_schema_model(namespace, UserModel)
with pytest.raises(SchemaNameCollisionError, match="'UserModel'.*produced by both"):
register_schema_model(namespace, _make_same_named_model({"email": str}))
register_schema_model(namespace, _make_same_named_model())
def test_register_schema_model_rejects_nested_definition_with_same_name():
@@ -208,7 +204,7 @@ def test_register_schema_model_keeps_namespaces_independent():
from controllers.common.schema import register_schema_model
register_schema_model(Namespace("first"), UserModel)
register_schema_model(Namespace("second"), _make_same_named_model({"email": str}))
register_schema_model(Namespace("second"), _make_same_named_model())
def test_register_schema_model_applies_schema_name_overrides(monkeypatch: pytest.MonkeyPatch):