writing-helper/src/app/lib/__tests__/iron-rules.test.ts

import { describe, it, expect } from 'vitest';
import { buildIronRulesPrompt } from '@/app/lib/chapter-prompts';

describe('buildIronRulesPrompt', () => {
  it('includes all project info in the prompt', () => {
    const prompt = buildIronRulesPrompt({
      novelType: 'xianxia',
      worldDescription: '修仙世界',
      protagonistDescription: '少年剑客',
      plotDirection: '复仇与成长',
    });
    expect(prompt).toContain('xianxia');
    expect(prompt).toContain('修仙世界');
    expect(prompt).toContain('少年剑客');
    expect(prompt).toContain('复仇与成长');
  });

  it('handles empty fields gracefully', () => {
    const prompt = buildIronRulesPrompt({
      novelType: 'custom',
      worldDescription: '',
      protagonistDescription: '',
      plotDirection: '',
    });
    expect(prompt).toContain('未提供');
  });
});