Merge branch 'main' into asyncio

tests/test_pydantic_integrations.py

@@ -0,0 +1,49 @@
+import abc
+import datetime
+from collections import namedtuple
+
+import pytest
+from pydantic import EmailStr, ValidationError
+
+from aredis_om import Field, HashModel, Migrator
+
+
+today = datetime.date.today()
+
+
+@pytest.fixture
+async def m(key_prefix, redis):
+    class BaseHashModel(HashModel, abc.ABC):
+        class Meta:
+            global_key_prefix = key_prefix
+
+    class Member(BaseHashModel):
+        first_name: str
+        last_name: str
+        email: EmailStr = Field(index=True)
+        join_date: datetime.date
+        age: int
+
+    await Migrator(redis).run()
+
+    return namedtuple("Models", ["Member"])(Member)
+
+
+def test_email_str(m):
+    with pytest.raises(ValidationError):
+        m.Member(
+            first_name="Andrew",
+            last_name="Brookins",
+            email="not an email!",
+            age=38,
+            join_date=today,
+        )
+
+    with pytest.raises(ValidationError):
+        m.Member(
+            first_name="Andrew",
+            last_name="Brookins",
+            email="andrew@bad-domain",
+            age=38,
+            join_date=today,
+        )