import { describe, expect, it } from 'vitest';
import { buildQuestions, QUESTIONS, toEvidence } from '../src/jev.ts';

const base = {
  claims_sector: { choice: 'solicitation', probabilities: { financial: 0.05, healthcare: 0, education: 0, solicitation: 0.95 } },
  specific_detail: { noul: 0.03 }, official_channel: { noul: 0.04 }, pressure: { noul: 0.2 }, agent_or_credentials: { noul: 0.1 },
  sales_pitch: { noul: 0.98 }, self_assertion: { noul: 0.1 }, category: { probabilities: { legit_alert: 0, scam: 0.3, marketing_robocall: 0.7 } },
};

describe('known-script question', () => {
  it('is omitted when no junk scripts are known', () => expect(buildQuestions([])).toBe(QUESTIONS));
  it('lists each known script plus none', () => {
    const q = buildQuestions([{ id: 1, sample: 'Prime loan approval now' }]) as Record<string, { criteria?: Record<string, string> }>;
    expect(Object.keys(q.known_script.criteria!)).toEqual(['s1', 'none']);
  });
  it('parses a match and a none answer', () => {
    expect(toEvidence({ ...base, known_script: { choice: 's1', probabilities: { s1: 0.97, none: 0.03 } } }).knownScript).toEqual({ id: 1, p: 0.97 });
    expect(toEvidence({ ...base, known_script: { choice: 'none', probabilities: { s1: 0.1, none: 0.9 } } }).knownScript).toBeNull();
    expect(toEvidence(base).knownScript).toBeNull();
  });
});