Catch the wrapt __reduce_ex__ NotImplemented error that gets throw when a component uses the db_model decorator.

Author: adamghill

django_unicorn/utils.py

@@ -61,5 +61,9 @@ def get_cacheable_component(component):
         raise UnicornCacheError(
             "Cannot cache component because it is not picklable."
         ) from e
+    except NotImplementedError as e:
+        raise UnicornCacheError(
+            "Cannot cache component because it is not picklable."
+        ) from e
 
     return component

tests/decorators/test_db_model.py

@@ -3,6 +3,8 @@
 from django_unicorn.components import UnicornView
 from django_unicorn.db import DbModel
 from django_unicorn.decorators import db_model
+from django_unicorn.errors import UnicornCacheError
+from django_unicorn.utils import get_cacheable_component
 from example.coffee.models import Flavor
 
 
@@ -78,3 +80,12 @@ class Meta:
 
     with pytest.raises(AssertionError):
         component.get_pk({"name": "flavor", "pk": -99})
+
+
+@pytest.mark.django_db
+def test_component_db_model_not_pickleable(component):
+    """
+    This is not ideal and should hopefully be fixable by providing a `__reduce_ex__` method.
+    """
+    with pytest.raises(UnicornCacheError):
+        get_cacheable_component(component)