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

77 lines
1.8 KiB
JavaScript

/*
* 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);
});
});