126 lines
3.3 KiB
TypeScript
126 lines
3.3 KiB
TypeScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
|
|
import { buildHostGroupTree } from './hostGroupTree.ts';
|
|
import { flattenHostGroupTree, hostTreeFlatRowKey } from './hostGroupTreeFlat.ts';
|
|
import type { Host } from '../types';
|
|
|
|
const host = (id: string, label: string, group?: string): Host => ({
|
|
id,
|
|
label,
|
|
hostname: `${id}.example.com`,
|
|
username: 'root',
|
|
port: 22,
|
|
group,
|
|
tags: [],
|
|
os: 'linux',
|
|
});
|
|
|
|
test('flattenHostGroupTree emits group rows before visible children in saved order', () => {
|
|
const { groupTree, ungroupedHosts } = buildHostGroupTree(
|
|
[
|
|
host('1', 'web-1', 'prod/web'),
|
|
host('2', 'db-1', 'prod/db'),
|
|
host('3', 'local'),
|
|
],
|
|
['prod/web'],
|
|
);
|
|
|
|
const expanded = new Set(['prod', 'prod/web', 'prod/db']);
|
|
const rows = flattenHostGroupTree({
|
|
groupNodes: groupTree,
|
|
ungroupedHosts,
|
|
expandedPaths: expanded,
|
|
searchActive: false,
|
|
});
|
|
|
|
assert.deepEqual(
|
|
rows.map((row) => (row.kind === 'group' ? `g:${row.node.path}` : `h:${row.host.id}`)),
|
|
['g:prod', 'g:prod/web', 'h:1', 'g:prod/db', 'h:2', 'h:3'],
|
|
);
|
|
});
|
|
|
|
test('flattenHostGroupTree uses saved group order when provided', () => {
|
|
const { groupTree, ungroupedHosts } = buildHostGroupTree(
|
|
[
|
|
host('1', 'web-1', 'prod/web'),
|
|
host('2', 'db-1', 'prod/db'),
|
|
],
|
|
['prod/web', 'prod/db'],
|
|
[
|
|
{ path: 'prod/db', order: 1000 },
|
|
{ path: 'prod/web', order: 2000 },
|
|
],
|
|
);
|
|
|
|
const rows = flattenHostGroupTree({
|
|
groupNodes: groupTree,
|
|
ungroupedHosts,
|
|
expandedPaths: new Set(['prod', 'prod/web', 'prod/db']),
|
|
searchActive: false,
|
|
});
|
|
|
|
assert.deepEqual(
|
|
rows.map((row) => (row.kind === 'group' ? `g:${row.node.path}` : `h:${row.host.id}`)),
|
|
['g:prod', 'g:prod/db', 'h:2', 'g:prod/web', 'h:1'],
|
|
);
|
|
});
|
|
|
|
test('flattenHostGroupTree uses saved group order for host-only groups', () => {
|
|
const { groupTree, ungroupedHosts } = buildHostGroupTree(
|
|
[
|
|
host('1', 'web-1', 'prod/web'),
|
|
host('2', 'db-1', 'prod/db'),
|
|
],
|
|
[],
|
|
[
|
|
{ path: 'prod/db', order: 1000 },
|
|
{ path: 'prod/web', order: 2000 },
|
|
],
|
|
);
|
|
|
|
const rows = flattenHostGroupTree({
|
|
groupNodes: groupTree,
|
|
ungroupedHosts,
|
|
expandedPaths: new Set(['prod', 'prod/web', 'prod/db']),
|
|
searchActive: false,
|
|
});
|
|
|
|
assert.deepEqual(
|
|
rows.map((row) => (row.kind === 'group' ? `g:${row.node.path}` : `h:${row.host.id}`)),
|
|
['g:prod', 'g:prod/db', 'h:2', 'g:prod/web', 'h:1'],
|
|
);
|
|
});
|
|
|
|
test('flattenHostGroupTree hides collapsed subtrees', () => {
|
|
const { groupTree, ungroupedHosts } = buildHostGroupTree(
|
|
[host('1', 'web-1', 'prod/web')],
|
|
[],
|
|
);
|
|
|
|
const rows = flattenHostGroupTree({
|
|
groupNodes: groupTree,
|
|
ungroupedHosts,
|
|
expandedPaths: new Set(['prod']),
|
|
searchActive: false,
|
|
});
|
|
|
|
assert.deepEqual(rows.map(hostTreeFlatRowKey), ['g:prod', 'g:prod/web']);
|
|
});
|
|
|
|
test('flattenHostGroupTree expands all rows while searching', () => {
|
|
const { groupTree, ungroupedHosts } = buildHostGroupTree(
|
|
[host('1', 'web-1', 'prod/web')],
|
|
[],
|
|
);
|
|
|
|
const rows = flattenHostGroupTree({
|
|
groupNodes: groupTree,
|
|
ungroupedHosts,
|
|
expandedPaths: new Set(),
|
|
searchActive: true,
|
|
});
|
|
|
|
assert.deepEqual(rows.map(hostTreeFlatRowKey), ['g:prod', 'g:prod/web', 'h:1']);
|
|
});
|