import { describe, it, expect } from 'vitest';
import { z } from 'zod';
import { convertParameterToZodSchema, convertPathToToolName, sanitizePropertyName } from './schema.js';
import { MinimalParameter } from './minimal-types.js';
describe('schema', () => {
describe('convertParameterToZodSchema', () => {
it('should convert string parameter to ZodString', () => {
const parameter: MinimalParameter = {
name: 'name',
in: 'query',
type: 'string',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema).toBeInstanceOf(z.ZodString);
});
it('should convert integer parameter to ZodNumber with int constraint', () => {
const parameter: MinimalParameter = {
name: 'id',
in: 'path',
type: 'integer',
required: true
};
const schema = convertParameterToZodSchema(parameter);
// ZodNumber.int() creates a ZodEffects schema internally, but vitest might see the outer type
expect((schema._def as unknown as { typeName: string }).typeName).toMatch(/^Zod(Effects|Number)$/);
});
it('should convert number parameter to ZodNumber', () => {
const parameter: MinimalParameter = {
name: 'amount',
in: 'query',
type: 'number',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema).toBeInstanceOf(z.ZodNumber);
});
it('should convert boolean parameter to ZodBoolean', () => {
const parameter: MinimalParameter = {
name: 'active',
in: 'query',
type: 'boolean',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema).toBeInstanceOf(z.ZodBoolean);
});
it('should convert unknown type to ZodAny', () => {
const parameter: MinimalParameter = {
name: 'data',
in: 'query',
type: 'array',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema).toBeInstanceOf(z.ZodAny);
});
it('should make schema optional when not required', () => {
const parameter: MinimalParameter = {
name: 'optional_param',
in: 'query',
type: 'string',
required: false
};
const schema = convertParameterToZodSchema(parameter);
expect((schema._def as unknown as { typeName: string }).typeName).toBe('ZodOptional');
});
it('should add description when provided', () => {
const parameter: MinimalParameter = {
name: 'name',
in: 'query',
type: 'string',
description: 'The name of the resource',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema._def.description).toBe('The name of the resource');
});
it('should handle type string directly', () => {
const parameter: MinimalParameter = {
name: 'legacy_param',
in: 'query',
type: 'string',
required: true
};
const schema = convertParameterToZodSchema(parameter);
expect(schema).toBeInstanceOf(z.ZodString);
});
});
describe('convertPathToToolName', () => {
it('should convert simple path to tool name', () => {
const path = '/api/1/users';
const result = convertPathToToolName(path);
expect(result).toBe('users');
});
it('should convert path with ID parameter', () => {
const path = '/api/1/users/{id}';
const result = convertPathToToolName(path);
expect(result).toBe('users_by_id');
});
it('should convert nested path', () => {
const path = '/api/1/companies/{company_id}/deals';
const result = convertPathToToolName(path);
expect(result).toBe('companies_by_id_deals');
});
it('should convert path with multiple parameters', () => {
const path = '/api/1/companies/{company_id}/deals/{deal_id}';
const result = convertPathToToolName(path);
expect(result).toBe('companies_by_id_deals_by_id');
});
it('should handle paths without API version prefix', () => {
const path = '/users/{id}/profile';
const result = convertPathToToolName(path);
expect(result).toBe('_users_by_id_profile');
});
it('should truncate very long tool names', () => {
const longPath = '/api/1/very/long/path/with/many/segments/that/exceeds/the/limit';
const result = convertPathToToolName(longPath);
expect(result.length).toBeLessThanOrEqual(50);
expect(result).toBe('very_long_path_with_many_segments_that_exceeds_the');
});
it('should handle empty path', () => {
const path = '';
const result = convertPathToToolName(path);
expect(result).toBe('');
});
it('should handle root path', () => {
const path = '/';
const result = convertPathToToolName(path);
expect(result).toBe('_');
});
it('should handle path with trailing slash', () => {
const path = '/api/1/users/';
const result = convertPathToToolName(path);
expect(result).toBe('users_');
});
});
describe('sanitizePropertyName', () => {
it('should keep valid property names unchanged', () => {
const validNames = ['company_id', 'user-name', 'item.count', 'param123'];
validNames.forEach(name => {
expect(sanitizePropertyName(name)).toBe(name);
});
});
it('should replace square brackets with underscores', () => {
expect(sanitizePropertyName('visible_tags[]')).toBe('visible_tags__');
expect(sanitizePropertyName('visible_ids[]')).toBe('visible_ids__');
expect(sanitizePropertyName('array[0]')).toBe('array_0_');
});
it('should replace invalid characters with underscores', () => {
expect(sanitizePropertyName('user@name')).toBe('user_name');
expect(sanitizePropertyName('param#1')).toBe('param_1');
expect(sanitizePropertyName('key:value')).toBe('key_value');
expect(sanitizePropertyName('data/path')).toBe('data_path');
});
it('should handle spaces by replacing with underscores', () => {
expect(sanitizePropertyName('company id')).toBe('company_id');
expect(sanitizePropertyName('user name')).toBe('user_name');
});
it('should truncate names longer than 64 characters', () => {
const longName = 'a'.repeat(100);
const result = sanitizePropertyName(longName);
expect(result.length).toBe(64);
expect(result).toBe('a'.repeat(64));
});
it('should handle empty string by returning underscore', () => {
expect(sanitizePropertyName('')).toBe('_');
});
it('should handle string with only invalid characters', () => {
expect(sanitizePropertyName('###')).toBe('___');
expect(sanitizePropertyName('@@@')).toBe('___');
});
it('should ensure result matches MCP pattern', () => {
const mcpPattern = /^[a-zA-Z0-9_.-]{1,64}$/;
const testCases = [
'visible_tags[]',
'user@name',
'param#1',
'company id',
'a'.repeat(100),
'',
'###'
];
testCases.forEach(testCase => {
const result = sanitizePropertyName(testCase);
expect(result).toMatch(mcpPattern);
});
});
});
});
