/* * 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 . */ import {treeGroupBy} from '../src/index.js'; import {expect} from 'chai'; /** * @typedef {{id: number; path?: string[]}} TestItem * @typedef {import('../src/types.js').Tree} 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); }); });