ktx/packages/cli/test/context/ingest/memory-flow/visuals.test.ts
Andrey Avtomonov 56985b7e09
test: split cli tests from source tree (#216)
* feat(cli): define full warehouse dialect contract

* test(cli): keep dialect edge tests focused

* fix(cli): stabilize dialect contract foundation

* refactor(connectors): own read-only query preparation

* refactor(connectors): resolve dialects through registry

* refactor(connectors): keep concrete dialect classes internal

* chore(workspace): enforce dialect import boundary

* refactor(cli): resolve relationship dialect at scan boundary

* refactor(cli): use dialect display parsing for entity details

* refactor(cli): use dialect display parsing for warehouse catalog

* refactor(cli): use dialect SQL in relationship workflows

* test(cli): verify solid dialect scan workflow closure

* test: split cli tests from source tree

* refactor(cli): standardize BigQuery scope listing

* feat(sqlite): implement connector scope listing

* test(connectors): cover required table listing

* feat(cli): add warehouse driver registry

* refactor(setup): route scope discovery through driver registry

* refactor(cli): route local query execution through driver registry

* refactor(historic-sql): route dialect support through driver registry

* refactor(cli): test warehouse connections through driver registry

* fix(cli): close driver registry type export gaps

* Improve setup daemon diagnostics

* refactor(setup): centralize rail-prefixed diagnostics + query-history fallback

Extract errorMessage, writePrefixedLines, and flushPrefixedBufferedCommandOutput
into clack.ts so the setup wizard, managed daemons, and embedding/agent steps
share one rail-formatted writer. setup-databases.ts also adds a
"disable query history and retry" option when the schema-context build fails
and query history is the likely culprit, surfaced via a new
failed-query-history-unavailable status.

* fix(cli): carry catalog through the picker so BigQuery/Snowflake/SQL Server scope filters match

The setup picker's KtxTableListEntry was a 2-level { schema, name }, so
qualifiedTableId always wrote db.name into enabled_tables. When BigQuery,
Snowflake, or SQL Server later ran fast ingest, their introspect step filtered
the scope set with scopedTableNames(scope, { catalog: projectId|database, db })
— catalog was non-null on the introspect side but null in the scope refs, so
every entry was rejected, the live-database adapter staged zero table files,
and detect() failed with 'Adapter "live-database" did not recognize fetched
source output'.

Align the picker boundary with the canonical 3-level KtxTableRef:

- Add catalog: string | null to KtxTableListEntry.
- BigQuery/Snowflake/SQL Server listTables populate catalog from the
  resolved projectId / database; Postgres/MySQL/ClickHouse/SQLite set null.
- qualifiedTableId emits catalog.schema.name when catalog is non-null
  (resolveEnabledTables already accepts the 3-part shape) and
  schemasFromEnabledTables now goes through parseDottedTableEntry so it
  recovers the schema correctly from both 2-part and 3-part entries.
- Export parseDottedTableEntry from enabled-tables.ts (@internal) for picker
  reuse.

Update listTables expectations in all seven connector tests and the setup /
picker test fixtures. Add a picker regression test that covers the
catalog-bearing round-trip (save + refine).

* fix(cli): allow debug telemetry under opt-out env
2026-05-26 08:49:05 +02:00

70 lines
2.9 KiB
TypeScript

import { describe, expect, it } from 'vitest';
import {
buildMemoryFlowVisualModel,
memoryFlowStatusBadge,
renderMemoryFlowConnectorLine,
} from '../../../../src/context/ingest/memory-flow/visuals.js';
import type { MemoryFlowViewModel } from '../../../../src/context/ingest/memory-flow/types.js';
function viewWithStatuses(statuses: Array<'waiting' | 'active' | 'complete' | 'warning' | 'failed'>): MemoryFlowViewModel {
const titles = ['SOURCE', 'CHUNKS', 'WORKUNITS', 'ACTIONS', 'GATES', 'SAVED'];
const ids = ['source', 'chunks', 'workUnits', 'actions', 'gates', 'saved'] as const;
return {
title: 'KTX memory flow warehouse/metricflow running',
subtitle: 'Run run-1 Sync sync-1',
status: 'running',
activeLine: 'active: WorkUnit orders',
selectedTitle: 'WORKUNITS',
selectedDetails: ['orders: 1 raw, 0 peers, 1 deps'],
completionLine: null,
trustIssues: [],
details: { actions: [], provenance: [], transcripts: [] },
columns: statuses.map((status, index) => ({
id: ids[index],
title: titles[index],
status,
headline: `${titles[index].toLowerCase()} headline`,
counters: [],
chips: [],
details: [],
})),
};
}
describe('memory-flow visual helpers', () => {
it('uses ASCII badges with text meaning for every status', () => {
expect(memoryFlowStatusBadge('waiting')).toEqual({ label: '..', text: 'waiting' });
expect(memoryFlowStatusBadge('active')).toEqual({ label: '>>', text: 'active' });
expect(memoryFlowStatusBadge('complete')).toEqual({ label: 'OK', text: 'complete' });
expect(memoryFlowStatusBadge('warning')).toEqual({ label: '!!', text: 'warning' });
expect(memoryFlowStatusBadge('failed')).toEqual({ label: 'XX', text: 'failed' });
});
it('renders a no-color connector line with status badges and six columns', () => {
const view = viewWithStatuses(['complete', 'complete', 'active', 'waiting', 'waiting', 'waiting']);
expect(renderMemoryFlowConnectorLine(view)).toBe(
'OK SOURCE -> OK CHUNKS -> >> WORKUNITS -> .. ACTIONS -> .. GATES -> .. SAVED',
);
});
it('moves the pulse to the active column, then warnings, failures, and the last completed column', () => {
expect(
buildMemoryFlowVisualModel(viewWithStatuses(['complete', 'complete', 'active', 'waiting', 'waiting', 'waiting']))
.pulseColumnId,
).toBe('workUnits');
expect(
buildMemoryFlowVisualModel(viewWithStatuses(['complete', 'warning', 'complete', 'waiting', 'waiting', 'waiting']))
.pulseColumnId,
).toBe('chunks');
expect(
buildMemoryFlowVisualModel(viewWithStatuses(['complete', 'complete', 'failed', 'waiting', 'waiting', 'waiting']))
.pulseColumnId,
).toBe('workUnits');
expect(
buildMemoryFlowVisualModel(viewWithStatuses(['complete', 'complete', 'complete', 'complete', 'waiting', 'waiting']))
.pulseColumnId,
).toBe('actions');
});
});