Add more tests

Fatal1ty · Nov 23, 2024 · fcfe2d3 · fcfe2d3
1 parent 24403f7
commit fcfe2d3
Showing 1 changed file with 29 additions and 1 deletion.
diff --git a/tests/test_jsonschema/test_jsonschema_plugins.py b/tests/test_jsonschema/test_jsonschema_plugins.py
@@ -4,7 +4,11 @@
 import pytest
 
 from mashumaro.jsonschema.builder import JSONSchemaBuilder, build_json_schema
-from mashumaro.jsonschema.models import Context, JSONSchema
+from mashumaro.jsonschema.models import (
+    Context,
+    JSONSchema,
+    JSONSchemaInstanceType,
+)
 from mashumaro.jsonschema.plugins import BasePlugin, DocstringDescriptionPlugin
 from mashumaro.jsonschema.schema import Instance
 
@@ -25,6 +29,30 @@ class DataClassWithoutDocstring:
     x: int
 
 
+def test_basic_plugin():
+    assert build_json_schema(int, plugins=[BasePlugin()]) == JSONSchema(
+        type=JSONSchemaInstanceType.INTEGER
+    )
+
+
+def test_plugin_with_not_implemented_error():
+    class NotImplementedErrorPlugin(BasePlugin):
+        def get_schema(
+            self,
+            instance: Instance,
+            ctx: Context,
+            schema: Optional[JSONSchema] = None,
+        ) -> Optional[JSONSchema]:
+            raise NotImplementedError
+
+    assert build_json_schema(
+        int, plugins=[NotImplementedErrorPlugin()]
+    ) == JSONSchema(type=JSONSchemaInstanceType.INTEGER)
+    assert JSONSchemaBuilder(plugins=[NotImplementedErrorPlugin()]).build(
+        int
+    ) == JSONSchema(type=JSONSchemaInstanceType.INTEGER)
+
+
 @pytest.mark.parametrize(
     ("obj", "docstring"),
     (