File size: 4,322 Bytes
5710dd0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | import { describe, expect, it } from 'vitest';
import { buildArgs, parseParamFields, smartParse } from './methodArgs';
describe('parseParamFields', () => {
it('parses a single named parameter', () => {
expect(parseParamFields('sessionId')).toEqual([{ kind: 'value', name: 'sessionId' }]);
});
it('parses several named parameters', () => {
expect(parseParamFields('id, decision')).toEqual([
{ kind: 'value', name: 'id' },
{ kind: 'value', name: 'decision' },
]);
});
it('keeps declared defaults', () => {
expect(parseParamFields("count = 1, mode = 'auto'")).toEqual([
{ kind: 'value', name: 'count', defaultValue: '1' },
{ kind: 'value', name: 'mode', defaultValue: "'auto'" },
]);
});
it('does not split commas inside defaults', () => {
expect(parseParamFields("ids = ['a', 'b'], flag")).toEqual([
{ kind: 'value', name: 'ids', defaultValue: "['a', 'b']" },
{ kind: 'value', name: 'flag' },
]);
});
it('expands a destructured object parameter into keys', () => {
expect(parseParamFields('{ workspaceId, limit }')).toEqual([
{ kind: 'object', name: '{ workspaceId, limit }', keys: ['workspaceId', 'limit'] },
]);
});
it('strips key defaults and renames in object patterns', () => {
expect(parseParamFields('{ limit = 10, workspaceId: ws }')).toEqual([
{
kind: 'object',
name: '{ limit = 10, workspaceId: ws }',
keys: ['limit', 'workspaceId'],
},
]);
});
it('keeps the object-pattern default', () => {
expect(parseParamFields('{ a, b } = {}')).toEqual([
{ kind: 'object', name: '{ a, b } = {}', keys: ['a', 'b'], defaultValue: '{}' },
]);
});
it('drops rest elements inside object patterns', () => {
expect(parseParamFields('{ a, ...rest }')).toEqual([
{ kind: 'object', name: '{ a, ...rest }', keys: ['a'] },
]);
});
it('falls back to raw for rest and array patterns', () => {
expect(parseParamFields('...args')).toEqual([{ kind: 'raw', label: '...args' }]);
expect(parseParamFields('[a, b]')).toEqual([{ kind: 'raw', label: '[a, b]' }]);
});
it('returns no fields for empty params', () => {
expect(parseParamFields('')).toEqual([]);
});
});
describe('smartParse', () => {
it('parses JSON values', () => {
expect(smartParse('5')).toBe(5);
expect(smartParse('true')).toBe(true);
expect(smartParse('{"a":1}')).toEqual({ a: 1 });
expect(smartParse('"x"')).toBe('x');
});
it('passes non-JSON through as a plain string', () => {
expect(smartParse('main')).toBe('main');
expect(smartParse('k2-thinking')).toBe('k2-thinking');
});
});
describe('buildArgs', () => {
it('builds a single argument from a value field', () => {
const fields = parseParamFields('sessionId');
expect(buildArgs(fields, { '0': 'abc' })).toEqual(['abc']);
});
it('assembles a destructured object from its key fields', () => {
const fields = parseParamFields('{ workspaceId, limit }');
expect(buildArgs(fields, { '0.workspaceId': 'ws1', '0.limit': '5' })).toEqual([
{ workspaceId: 'ws1', limit: 5 },
]);
});
it('drops empty keys and omits a fully-empty object', () => {
const fields = parseParamFields('{ workspaceId, limit }');
expect(buildArgs(fields, { '0.limit': '5' })).toEqual([{ limit: 5 }]);
expect(buildArgs(fields, {})).toEqual([]);
});
it('uses declared defaults for empty value fields', () => {
const fields = parseParamFields("count = 1, mode = 'auto'");
expect(buildArgs(fields, {})).toEqual([1, 'auto']);
expect(buildArgs(fields, { '1': 'yolo' })).toEqual([1, 'yolo']);
});
it('truncates trailing empty params but keeps interior holes', () => {
const fields = parseParamFields('a, b, c');
expect(buildArgs(fields, { '0': 'x' })).toEqual(['x']);
// Interior hole: `b` empty, `c` filled — the hole serializes as null.
expect(JSON.parse(JSON.stringify(buildArgs(fields, { '0': 'x', '2': 'z' })))).toEqual([
'x',
null,
'z',
]);
});
it('parses raw fallback fields as strict JSON', () => {
const fields = parseParamFields('...args');
expect(buildArgs(fields, { '0': '[1, 2]' })).toEqual([[1, 2]]);
expect(() => buildArgs(fields, { '0': 'not json' })).toThrow();
});
});
|