promptfoo--promptfoo/test/redteam/plugins/mcp.test.ts

import { beforeEach, describe, expect, it } from 'vitest';
import { MCPPluginGrader } from '../../../src/redteam/plugins/mcp';

describe('MCPPluginGrader', () => {
  let grader: MCPPluginGrader;

  beforeEach(() => {
    grader = new MCPPluginGrader();
  });

  describe('renderRubric', () => {
    it('should render rubric with provided variables', () => {
      const vars = {
        purpose: 'Test purpose',
        prompt: 'Test attack prompt',
        output: 'Test output',
      };

      const result = grader.renderRubric(vars);

      expect(result).toContain('# Goal');
      expect(result).toContain('Test purpose');
      expect(result).toContain('Test attack prompt');
      expect(result).toContain('Test output');
    });

    it('should handle empty variables', () => {
      const vars = {
        purpose: '',
        prompt: '',
        output: '',
      };

      const result = grader.renderRubric(vars);

      expect(result).toContain('# Goal');
      expect(result).toContain('# Instructions');
      expect(result).toContain('# Output Format');
    });
  });

  it('should have correct plugin ID', () => {
    expect(grader.id).toBe('promptfoo:redteam:mcp');
  });

  it('should initialize with empty rubric', () => {
    expect(grader.rubric).toBe('');
  });
});