feat: support enums in Model by dumping to json (#168)

* feat: support enums in Model by dumping to json

This adds support for Enums to the Model class. They'll be dumpeds to
JSON to store in redis.

#153

* fix: pre-commit

Author: andrewthetechie

Makefile

@@ -11,3 +11,6 @@ build: setup ## build python packages
 	pip install twine build
 	python -m build --sdist --wheel --outdir dist/
 	twine check dist/*
+
+test: setup ## Run unit tests
+	pytest

pydantic_aioredis/abstract.py

@@ -2,6 +2,7 @@
 import json
 from datetime import date
 from datetime import datetime
+from enum import Enum
 from ipaddress import IPv4Address
 from ipaddress import IPv4Network
 from ipaddress import IPv6Address
@@ -35,6 +36,7 @@
     SHAPE_FROZENSET,
     SHAPE_DICT,
     SHAPE_DEFAULTDICT,
+    Enum,
 ]
 
 

test/test_pydantic_aioredis.py

@@ -1,5 +1,6 @@
 """Tests for the redis orm"""
 from datetime import date
+from enum import Enum
 from ipaddress import ip_network
 from ipaddress import IPv4Network
 from random import randint
@@ -408,3 +409,25 @@ class TestModel(Model):
     this_model = TestModel(name="test", object=MyClass())
     with pytest.raises(TypeError):
         await TestModel.insert(this_model)
+
+
+@pytest.mark.asyncio
+async def test_enum_support(redis_store):
+    """Test case for aioredis support"""
+
+    class TestEnum(str, Enum):
+        test = "test"
+        foo = "foo"
+        bar = "bar"
+        baz = "baz"
+
+    class EnumModel(Model):
+        _primary_key_field = "id"
+        id: int
+        enum: TestEnum
+
+    redis_store.register_model(EnumModel)
+    this_model = EnumModel(id=0, enum="foo")
+    await EnumModel.insert(this_model)
+    from_redis = await EnumModel.select()
+    assert from_redis[0] == this_model