Fixed minor bug on unspecified property through constructor

trivago · Aug 3, 2018 · fb26439 · fb26439
1 parent ef98ff3
commit fb26439
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/rebase/core/object.py b/rebase/core/object.py
@@ -157,6 +157,7 @@ def _init_attributes(self):
                 if isinstance(attribute, str) and attribute:
                     data = self._get_attr_recurse(
                         attribute, self._raw_attributes)
+                    data = data if data != attribute else None
                 elif callable(attribute):
                     data = attribute()
 

diff --git a/tests/core/test_object.py b/tests/core/test_object.py
@@ -47,6 +47,7 @@ def test_object_basic(self):
                 'gender': ('gender', lambda x: int(x == 'Male')),
                 'city': 'location.city',
                 'country': ('location.country', lambda x: x.upper()),
+                'occupation': ('occupation', lambda x: x)
             }
         )
     )
@@ -58,3 +59,4 @@ def test_object_init(self):
         self.assertEqual(self.obj.gender, 1)
         self.assertEqual(self.obj.city, 'Paris')
         self.assertEqual(self.obj.country, 'FRANCE')
+        self.assertEqual(self.obj.occupation, None)