Fix broken input data and filename for test

Author: mattmilburn

server/src/utils/__tests__/get-descendants.spec.js

@@ -0,0 +1,98 @@
+'use strict';
+
+const getDescendants = require('../get-descendants');
+
+describe('getDescendants', () => {
+  it('should return descendants of an entity', () => {
+    const items = [
+      {
+        id: 1,
+        attributes: {
+          order: 0,
+          title: 'Item one',
+          url: '/one',
+          target: null,
+          parent: null,
+        },
+      },
+      {
+        id: 2,
+        attributes: {
+          order: 1,
+          title: 'Item two',
+          url: '/two',
+          target: null,
+          parent: null,
+        },
+      },
+      {
+        id: 3,
+        attributes: {
+          order: 0,
+          title: 'Item three',
+          url: '/three',
+          target: null,
+          parent: {
+            data: {
+              id: 2,
+            },
+          },
+        },
+      },
+      {
+        id: 4,
+        attributes: {
+          order: 0,
+          title: 'Item four',
+          url: '/four',
+          target: null,
+          parent: {
+            data: {
+              id: 3,
+            },
+          },
+        },
+      },
+    ];
+    const expected = [
+      {
+        id: 3,
+        attributes: {
+          order: 0,
+          title: 'Item three',
+          url: '/three',
+          target: null,
+          parent: {
+            data: {
+              id: 2,
+            },
+          },
+          children: {
+            data: [
+              {
+                id: 4,
+                attributes: {
+                  order: 0,
+                  title: 'Item four',
+                  url: '/four',
+                  target: null,
+                  parent: {
+                    data: {
+                      id: 3,
+                    },
+                  },
+                  children: {
+                    data: [],
+                  },
+                },
+              },
+            ],
+          },
+        },
+      },
+    ];
+    const result = getDescendants(items, 2);
+
+    expect(result).toEqual(expected);
+  });
+});

server/src/utils/serialize-nested-menu.js

@@ -0,0 +1,72 @@
+'use strict';
+
+const get = require('lodash/get');
+const has = require('lodash/has');
+const omit = require('lodash/omit');
+
+const sortByOrder = require('./sort-by-order');
+const getDescendants = require('./get-descendants');
+
+const removeParentData = (items) =>
+  items.reduce((acc, item) => {
+    const sanitizedItem = omit(item, 'attributes.parent');
+    sanitizedItem.attributes.children.data = removeParentData(item.attributes.children.data);
+
+    return [...acc, sanitizedItem];
+  }, []);
+
+const serializeEntity = (data, keepParentData) => {
+  const items = get(data, 'attributes.items.data', []);
+
+  // Do nothing if there are no items to serialize.
+  if (!items.length) {
+    return data;
+  }
+
+  const rootItems = items.filter((item) => !has(item, 'attributes.parent.data.id'));
+
+  // Assign ordered and nested items to root items.
+  const nestedItems = rootItems.reduce((acc, item) => {
+    const rootItem = {
+      ...item,
+      attributes: {
+        ...item.attributes,
+        children: {
+          data: getDescendants(items, item.id),
+        },
+      },
+    };
+
+    return [...acc, rootItem];
+  }, []);
+
+  const sanitizedItems = !keepParentData ? removeParentData(nestedItems) : nestedItems;
+
+  return {
+    ...data,
+    attributes: {
+      ...data.attributes,
+      items: {
+        data: sortByOrder(sanitizedItems),
+      },
+    },
+  };
+};
+
+const serializeNestedMenu = (res, keepParentData) => {
+  const data = get(res, 'data');
+  let sanitizedData;
+
+  if (Array.isArray(data)) {
+    sanitizedData = data.map((_data) => serializeEntity(_data, keepParentData));
+  } else {
+    sanitizedData = serializeEntity(data, keepParentData);
+  }
+
+  return {
+    ...res,
+    data: sanitizedData,
+  };
+};
+
+module.exports = serializeNestedMenu;