feat: pipeline improvements

packages/collection-utils/test/tree-group.spec.js

@@ -0,0 +1,76 @@
+/*
+ * Copyright (C) 2022 StApps
+ *  This program is free software: you can redistribute it and/or modify it
+ *  under the terms of the GNU General Public License as published by the Free
+ *  Software Foundation, version 3.
+ *
+ *  This program is distributed in the hope that it will be useful, but WITHOUT
+ *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ *  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ *  more details.
+ *
+ *  You should have received a copy of the GNU General Public License along with
+ *  this program. If not, see <https://www.gnu.org/licenses/>.
+ */
+import {treeGroupBy} from '../src/index.js';
+import {expect} from 'chai';
+
+/**
+ * @typedef {{id: number; path?: string[]}} TestItem
+ * @typedef {import('../src/types.js').Tree<TestItem>} TestItemTree
+ */
+describe('tree-group', function () {
+  it('should create a tree', function () {
+    /** @type {TestItem[]} */
+    const items = [
+      {
+        id: 1,
+        path: ['a', 'b', 'c'],
+      },
+      {
+        id: 2,
+        path: ['a', 'b', 'd'],
+      },
+    ];
+
+    const tree = treeGroupBy(items, item => item.path ?? []);
+
+    /** @type {TestItemTree} */
+    const expectedTree = {
+      a: /** @type {TestItemTree} */ ({
+        b: {
+          c: {_: [items[0]]},
+          d: {_: [items[1]]},
+        },
+      }),
+    };
+
+    expect(tree).to.deep.equal(expectedTree);
+  });
+
+  it('should also sort empty paths', () => {
+    /** @type {TestItem[]} */
+    const items = [
+      {
+        id: 1,
+        path: ['a', 'b', 'c'],
+      },
+      {
+        id: 2,
+      },
+    ];
+
+    const tree = treeGroupBy(items, item => item.path ?? []);
+
+    const expectedTree = /** @type {TestItemTree} */ ({
+      a: {
+        b: {
+          c: {_: [items[0]]},
+        },
+      },
+      _: [items[1]],
+    });
+
+    expect(tree).to.deep.equal(expectedTree);
+  });
+});