Set Reference field name on Entity construction

proteanhq · Sep 3, 2021 · 6416c33 · 6416c33
1 parent 2ff3444
commit 6416c33
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 5 deletions.
diff --git a/src/protean/core/entity.py b/src/protean/core/entity.py
@@ -133,6 +133,7 @@ def __init_subclass__(subclass) -> None:
         super().__init_subclass__()
 
         subclass.__set_id_field()
+        subclass.__set_up_reference_fields()
 
     @classmethod
     def __set_id_field(new_class):
@@ -180,6 +181,14 @@ def __create_id_field(new_class):
         field_objects["id"] = id_field
         setattr(new_class, _FIELDS, field_objects)
 
+    @classmethod
+    def __set_up_reference_fields(subclass):
+        """Walk through relation fields and setup shadow attributes"""
+        for _, field in fields(subclass).items():
+            if isinstance(field, Reference):
+                shadow_field_name, shadow_field = field.get_shadow_field()
+                shadow_field.__set_name__(subclass, shadow_field_name)
+
     def __init__(self, *template, **kwargs):  # noqa: C901
         """
         Initialise the entity object.

diff --git a/src/protean/server.py b/src/protean/server.py
@@ -25,7 +25,7 @@
 from protean.utils.importlib import import_from_full_path
 
 logging.basicConfig(
-    level=logging.DEBUG,  # FIXME Pick up log level from config
+    level=logging.INFO,  # FIXME Pick up log level from config
     format="%(threadName)10s %(name)18s: %(message)s",
     stream=sys.stderr,
 )

diff --git a/tests/adapters/broker/celery_broker/config.py b/tests/adapters/broker/celery_broker/config.py
@@ -53,10 +53,10 @@
     },
     "handlers": {
         "console": {
-            "level": "DEBUG",
+            "level": "INFO",
             "class": "logging.StreamHandler",
             "formatter": "console",
         },
     },
-    "loggers": {"protean": {"handlers": ["console"], "level": "DEBUG"}},
+    "loggers": {"protean": {"handlers": ["console"], "level": "INFO"}},
 }
diff --git a/tests/adapters/broker/redis_broker/config.py b/tests/adapters/broker/redis_broker/config.py
@@ -48,10 +48,10 @@
     },
     "handlers": {
         "console": {
-            "level": "DEBUG",
+            "level": "INFO",
             "class": "logging.StreamHandler",
             "formatter": "console",
         },
     },
-    "loggers": {"protean": {"handlers": ["console"], "level": "DEBUG"}},
+    "loggers": {"protean": {"handlers": ["console"], "level": "INFO"}},
 }
diff --git a/tests/aggregate/test_aggregate_association.py b/tests/aggregate/test_aggregate_association.py
@@ -1,6 +1,8 @@
 import mock
 import pytest
 
+from protean.utils.container import attributes
+
 from .elements import (
     Account,
     AccountVia,
@@ -240,3 +242,8 @@ def test_that_entities_up_to_configured_limit_value_are_retrieved(
 
         updated_post = test_domain.get_dao(Post).get(persisted_post.id)
         assert len(updated_post.comments) == 12
+
+
+class TestReference:
+    def test_that_reference_field_attribute_name_is_set_properly(self):
+        assert attributes(Author)["account_email"].attribute_name is not None