fix(json): pick correct root element in serialize

Pick the correct root element for references

Author: patricksmms

packages/json-serializer/lib/serializeJSON.js

@@ -1,17 +1,19 @@
 function collectElementByIDs(element) {
   const dataStructures = {};
-
   const { parents } = element;
-  if (parents) {
-    const rootElement = parents.get(0);
 
-    if (rootElement) {
-      rootElement.recursiveChildren.forEach((element) => {
-        if (element.id) {
-          dataStructures[element.id.toValue()] = element;
-        }
-      });
-    }
+  if (!parents || parents.isEmpty) {
+    return dataStructures;
+  }
+
+  const rootElement = parents.get(parents.length - 1);
+
+  if (rootElement) {
+    rootElement.recursiveChildren.forEach((element) => {
+      if (element.id) {
+        dataStructures[element.id.toValue()] = element;
+      }
+    });
   }
 
   return dataStructures;

packages/json-serializer/test/serializeJSON-test.js

@@ -26,7 +26,9 @@ describe('#serializeJSON', () => {
       new namespace.elements.Category([
         new namespace.elements.String('hello world', { id: 'message' }),
       ], { classes: ['dataStructures'] }),
-      element,
+      new namespace.elements.Category([
+        element,
+      ]),
     ]).freeze();
 
     expect(serializeJSON(element)).to.equal('"hello world"');