import { describe, it, expect } from '@jest/globals';
import {
GuardrailBlockError,
GuardrailInitError,
GuardrailExecutionError,
} from '../../src/guardrails/errors.js';
describe('GuardrailBlockError', () => {
it('should create error with plugin name and reason', () => {
const error = new GuardrailBlockError('rate_limiter', 'Too many requests');
expect(error.pluginName).toBe('rate_limiter');
expect(error.reason).toBe('Too many requests');
});
it('should format message correctly', () => {
const error = new GuardrailBlockError('pii_redactor', 'Sensitive data detected');
expect(error.message).toBe(
"Request blocked by guardrail 'pii_redactor': Sensitive data detected"
);
});
it('should have correct error name', () => {
const error = new GuardrailBlockError('test', 'reason');
expect(error.name).toBe('GuardrailBlockError');
});
it('should be instanceof Error', () => {
const error = new GuardrailBlockError('test', 'reason');
expect(error).toBeInstanceOf(Error);
});
});
describe('GuardrailInitError', () => {
it('should create error with plugin name and message', () => {
const error = new GuardrailInitError('token_limiter', 'Invalid configuration');
expect(error.pluginName).toBe('token_limiter');
expect(error.cause).toBeUndefined();
});
it('should format message correctly', () => {
const error = new GuardrailInitError('pattern_blocker', 'Missing required field');
expect(error.message).toBe(
"Failed to initialize guardrail plugin 'pattern_blocker': Missing required field"
);
});
it('should store cause error when provided', () => {
const cause = new Error('Original error');
const error = new GuardrailInitError('test_plugin', 'Init failed', cause);
expect(error.cause).toBe(cause);
});
it('should have correct error name', () => {
const error = new GuardrailInitError('test', 'message');
expect(error.name).toBe('GuardrailInitError');
});
});
describe('GuardrailExecutionError', () => {
it('should create error with plugin name, phase, and message', () => {
const error = new GuardrailExecutionError(
'rate_limiter',
'pre_request',
'Execution failed'
);
expect(error.pluginName).toBe('rate_limiter');
expect(error.phase).toBe('pre_request');
expect(error.cause).toBeUndefined();
});
it('should format message correctly', () => {
const error = new GuardrailExecutionError(
'pii_redactor',
'post_response',
'Pattern matching failed'
);
expect(error.message).toBe(
"Guardrail plugin 'pii_redactor' failed during 'post_response': Pattern matching failed"
);
});
it('should store cause error when provided', () => {
const cause = new TypeError('Cannot read property');
const error = new GuardrailExecutionError(
'test_plugin',
'pre_tool_input',
'Unexpected error',
cause
);
expect(error.cause).toBe(cause);
});
it('should have correct error name', () => {
const error = new GuardrailExecutionError('test', 'phase', 'message');
expect(error.name).toBe('GuardrailExecutionError');
});
});
