mirror of
https://github.com/Kaelio/ktx.git
synced 2026-07-01 08:59:39 +02:00
c22248dabf
* feat(context): add warehouse dialect dispatch * feat(context): read warehouse scan catalog * feat(context): add entity details verification tool * feat(context): add ingest SQL verification tool * feat(context): add raw warehouse discovery tool * feat(context): expose warehouse verification tools to ingest * docs(context): add ingest identifier verification protocol * test(context): guard ingest identifier verification prompts * chore(context): verify warehouse verification tools * docs: add warehouse verification tools plan and spec * fix(context): expose target warehouses to Notion ingest * fix(context): update ingest prompts for warehouse verification tools * fix(context): scope raw schema discovery to allowed connections * fix(context): verify warehouse column display targets * docs: add notion warehouse verification gap closure plan * fix(context): include raw discovery connection names * fix(context): expose warehouse targets for LookML and MetricFlow * fix(context): pass connection config to ingest query executors * fix(cli): enable read-only SQL probes for local ingest * docs: add warehouse verification final v1 closure plan * fix(context): align warehouse sql probe prompt shape * docs: add warehouse verification prompt shape closure plan * test(context): catch connectionless sql execution prompt examples * fix(context): include connection name in sl capture sql example * docs: add warehouse verification sql example closure plan * fix(context): report structured entity detail misses * docs: add warehouse verification structured target miss closure plan * fix: report untracked squash merge conflicts * feat: require ingest verification ledger * fix: stabilize ingest wiki references
54 lines
2.2 KiB
TypeScript
54 lines
2.2 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest';
|
|
import type { SlConnectionCatalogPort } from '../../../sl/index.js';
|
|
import type { ToolContext } from '../../../tools/index.js';
|
|
import { SqlExecutionTool } from './sql-execution.tool.js';
|
|
|
|
describe('SqlExecutionTool', () => {
|
|
const connections = {
|
|
executeQuery: vi.fn(),
|
|
} as unknown as SlConnectionCatalogPort & { executeQuery: ReturnType<typeof vi.fn> };
|
|
const tool = new SqlExecutionTool(connections);
|
|
const context: ToolContext = {
|
|
sourceId: 'ingest',
|
|
messageId: 'm1',
|
|
userId: 'system',
|
|
session: { allowedConnectionNames: new Set(['warehouse']) } as any,
|
|
};
|
|
|
|
it('wraps read-only SQL with a capped row limit', async () => {
|
|
connections.executeQuery.mockResolvedValue({ headers: ['status'], rows: [['paid']], totalRows: 1 });
|
|
|
|
const result = await tool.call(
|
|
{ connectionName: 'warehouse', sql: 'select status from public.orders', rowLimit: 5 },
|
|
context,
|
|
);
|
|
|
|
expect(connections.executeQuery).toHaveBeenCalledWith(
|
|
'warehouse',
|
|
'select * from (select status from public.orders) as ktx_query_result limit 5',
|
|
);
|
|
expect(result.markdown).toContain('| status |');
|
|
expect(result.structured.wrappedSql).toContain('limit 5');
|
|
});
|
|
|
|
it.each(['insert into x values (1)', 'drop table x', 'vacuum'])('rejects mutating SQL: %s', async (sql) => {
|
|
connections.executeQuery.mockClear();
|
|
|
|
const result = await tool.call({ connectionName: 'warehouse', sql }, context);
|
|
|
|
expect(result.markdown).toContain('Only read-only SELECT/WITH queries can be executed locally.');
|
|
expect(connections.executeQuery).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('surfaces connector errors verbatim', async () => {
|
|
connections.executeQuery.mockRejectedValue(new Error('relation "orbit_analytics.customer" does not exist'));
|
|
|
|
const result = await tool.call(
|
|
{ connectionName: 'warehouse', sql: 'select 1 from orbit_analytics.customer', rowLimit: 1 },
|
|
context,
|
|
);
|
|
|
|
expect(result.markdown).toContain('relation "orbit_analytics.customer" does not exist');
|
|
expect(result.structured.error).toContain('relation "orbit_analytics.customer" does not exist');
|
|
});
|
|
});
|