apunkt/ktx
1
0
Fork 0
mirror of https://github.com/Kaelio/ktx.git synced 2026-07-28 12:11:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactor(mongo-query): extract shared runMongoQuery execution seam

Browse source
This commit is contained in:
Kevin Messiaen 2026-07-06 19:54:41 +07:00
parent f5d431a398
commit 9a2e215bdc
No known key found for this signature in database
GPG key ID: 19FF750A17315202
3 changed files with 60 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

23
packages/cli/test/mcp/mongo-query-tool.test.ts
View file

@ -2,6 +2,7 @@ import { describe, expect, it, vi } from 'vitest';
import { KtxExpectedError } from '../../src/errors.js';
import { KtxMongoDbScanConnector } from '../../src/connectors/mongodb/connector.js';
import { createLocalProjectMcpContextPorts } from '../../src/context/mcp/local-project-ports.js';
import { runMongoQuery } from '../../src/context/mcp/mongo-query-runner.js';
import type { KtxLocalProject } from '../../src/context/project/project.js';
import type { KtxScanConnector } from '../../src/context/scan/types.js';
@ -80,3 +81,25 @@ describe('mongoQuery MCP port', () => {
).rejects.toBeInstanceOf(KtxExpectedError);
});
});
describe('runMongoQuery shared helper', () => {
it('fetches rows from a mongodb connector', async () => {
const result = await runMongoQuery(
(id) => (id === 'mongo' ? mongoConnector() : warehouseConnector()),
{ connectionId: 'mongo', collection: 'business', pipeline: [], limit: 100 },
);
expect(result.rowCount).toBe(2);
expect(result.headers).toEqual(['_id', 'city']);
});
it('throws an expected error naming the wrong driver for a non-mongodb connector', async () => {
await expect(
runMongoQuery((id) => (id === 'mongo' ? mongoConnector() : warehouseConnector()), {
connectionId: 'warehouse',
collection: 'x',
pipeline: [],
limit: 10,
}),
).rejects.toThrow(/is not a MongoDB connection/);
});
});