Merge pull request #21 from tarsil/fix/uniques

Add unique index validation

mongoz/core/db/documents/metaclasses.py

@@ -340,10 +340,12 @@ def __search_for_fields(base: Type, attrs: Any) -> None:
             if not new_class.is_proxy_document:
                 # For the indexes
                 _index: Union[Index, None] = None
-                if hasattr(field, "index") and field.index and field.unique is True:
+                if hasattr(field, "index") and field.index and field.unique:
                     _index = Index(name, unique=True, sparse=field.sparse)
                 elif hasattr(field, "index") and field.index:
                     _index = Index(name, sparse=field.sparse)
+                elif hasattr(field, "unique") and field.unique:
+                    _index = Index(name, unique=True)
 
                 if _index is not None:
                     index_names = [index.name for index in meta.indexes or []]

tests/indexes/test_unique.py

@@ -0,0 +1,41 @@
+from typing import AsyncGenerator
+
+import pydantic
+import pytest
+from pymongo.errors import DuplicateKeyError
+from tests.conftest import client
+
+import mongoz
+from mongoz import Document
+
+pytestmark = pytest.mark.anyio
+pydantic_version = pydantic.__version__[:3]
+
+
+class Movie(Document):
+    name: str = mongoz.String(unique=True)
+    is_published: bool = mongoz.Boolean(default=False)
+
+    class Meta:
+        registry = client
+        database = "test_db"
+
+
+@pytest.fixture(scope="function", autouse=True)
+async def prepare_database() -> AsyncGenerator:
+    await Movie.drop_indexes(force=True)
+    await Movie.objects.delete()
+    await Movie.create_indexes()
+    yield
+    await Movie.drop_indexes(force=True)
+    await Movie.objects.delete()
+    await Movie.create_indexes()
+
+
+async def test_unique():
+    movie = await Movie.objects.create(name="Barbie")
+
+    assert movie.is_published is False
+
+    with pytest.raises(DuplicateKeyError):
+        await Movie.objects.create(name="Barbie")