tests indexed fields (#795)

BeanieODM · Dec 3, 2023 · 5f03c6f · 5f03c6f
1 parent 09b3607
commit 5f03c6f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -44,7 +44,8 @@ test = [
     "httpx>=0.23.0",
     "fastapi>=0.100",
     "pydantic-settings>=2",
-    "pydantic-extra-types>=2"
+    "pydantic-extra-types>=2",
+    "pydantic[email]",
 ]
 doc = [
     "Pygments>=2.8.0",

diff --git a/tests/odm/conftest.py b/tests/odm/conftest.py
@@ -43,6 +43,7 @@
     DocumentWithDeprecatedHiddenField,
     DocumentWithExtras,
     DocumentWithHttpUrlField,
+    DocumentWithIndexedObjectId,
     DocumentWithIndexMerging1,
     DocumentWithIndexMerging2,
     DocumentWithKeepNullsFalse,
@@ -279,6 +280,7 @@ async def init(db):
         DocumentWithOptionalBackLink,
         DocumentWithOptionalListBackLink,
         DocumentWithComplexDictKey,
+        DocumentWithIndexedObjectId,
     ]
     await init_beanie(
         database=db,

diff --git a/tests/odm/models.py b/tests/odm/models.py
@@ -27,6 +27,7 @@
     UUID4,
     BaseModel,
     ConfigDict,
+    EmailStr,
     Field,
     HttpUrl,
     PrivateAttr,
@@ -1053,3 +1054,9 @@ class DocumentWithHttpUrlField(Document):
 
 class DocumentWithComplexDictKey(Document):
     dict_field: Dict[UUID, datetime.datetime]
+
+
+class DocumentWithIndexedObjectId(Document):
+    pyid: Indexed(PydanticObjectId)
+    uuid: Annotated[UUID4, Indexed(unique=True)]
+    email: Annotated[EmailStr, Indexed(unique=True)]