Serialize Base objects in MappedObject (#389)

Author: zliang-akamai

linode_api4/objects/base.py

@@ -109,9 +109,15 @@ def dict(self):
                 result[k] = v.dict
             elif isinstance(v, list):
                 result[k] = [
-                    item.dict if isinstance(item, cls) else item for item in v
+                    (
+                        item.dict
+                        if isinstance(item, cls)
+                        else item._raw_json if isinstance(item, Base) else item
+                    )
+                    for item in v
                 ]
-
+            elif isinstance(v, Base):
+                result[k] = v._raw_json
         return result
 
 

test/fixtures/testmappedobj1.json

@@ -0,0 +1,3 @@
+{
+    "bar": "bar"
+}

test/unit/objects/mapped_object_test.py

@@ -1,9 +1,9 @@
-from unittest import TestCase
+from test.unit.base import ClientBaseCase
 
-from linode_api4 import MappedObject
+from linode_api4.objects import Base, MappedObject, Property
 
 
-class MappedObjectCase(TestCase):
+class MappedObjectCase(ClientBaseCase):
     def test_mapped_object_dict(self):
         test_dict = {
             "key1": 1,
@@ -19,3 +19,26 @@ def test_mapped_object_dict(self):
 
         mapped_obj = MappedObject(**test_dict)
         self.assertEqual(mapped_obj.dict, test_dict)
+
+    def test_mapped_object_dict(self):
+        test_property_name = "bar"
+        test_property_value = "bar"
+
+        class Foo(Base):
+            api_endpoint = "/testmappedobj1"
+            id_attribute = test_property_name
+            properties = {
+                test_property_name: Property(mutable=True),
+            }
+
+        foo = Foo(self.client, test_property_value)
+        foo._api_get()
+
+        expected_dict = {
+            "foo": {
+                test_property_name: test_property_value,
+            }
+        }
+
+        mapped_obj = MappedObject(foo=foo)
+        self.assertEqual(mapped_obj.dict, expected_dict)