Add tests for fixtures and encoders

setup.py

@@ -43,6 +43,7 @@ def list_requirements(file):
             "flake8-black",
             "mypy",
             "invoke",
+            "fastapi",
         ],
         "typing": ["fastapi"],
     },

tests/test_encoders.py

@@ -0,0 +1,26 @@
+from typing import Type
+
+import pytest
+from bson import CodecOptions
+from bson.codec_options import TypeEncoder, TypeRegistry
+from motor.motor_asyncio import AsyncIOMotorDatabase
+
+from motor_odm import Document, encoders
+
+pytestmark = pytest.mark.asyncio
+
+
+@pytest.mark.parametrize("encoder", [encoders.SetEncoder, encoders.FrozensetEncoder])
+async def test_encoder(encoder: Type[TypeEncoder], db: AsyncIOMotorDatabase):
+    class Doc(Document):
+        class Mongo:
+            collection = "test"
+            codec_options = CodecOptions(type_registry=TypeRegistry([encoder()]))
+
+        val: encoder.python_type
+
+    doc = Doc(val={"A", "B", "C"})
+    await doc.save()
+    one = await Doc.find_one()
+    assert one.val == {"A", "B", "C"}
+    assert type(one.val) is encoder.python_type

tests/test_fixtures.py

@@ -0,0 +1,18 @@
+import fastapi
+from pydantic import BaseModel
+
+from motor_odm import Document, fixtures  # noqa: F401,F403
+
+
+def test_fastapi_fix():
+    class SomeModel(BaseModel):
+        pass
+
+    class SomeDoc(Document):
+        class Mongo:
+            collection = "test"
+
+    model = SomeModel()
+    doc = SomeDoc()
+    assert fastapi.routing._prepare_response_content(doc, exclude_unset=False) is doc
+    assert fastapi.routing._prepare_response_content(model, exclude_unset=False) == {}