import { describe, expect, it } from "vitest";

import { formulaEngine } from "@/lib/formula";
import { formulaRegressionCorpus } from "@/tests/fixtures/formula-corpus";

describe("sanitized five-workbook regression corpus", () => {
  it("contains exactly the expected identities and operator distribution", () => {
    const analyses = formulaRegressionCorpus.map((fixture) =>
      formulaEngine.analyze(fixture.formula),
    );
    const countUsing = (key: keyof (typeof analyses)[number]) =>
      analyses.filter((analysis) => {
        const value = analysis[key];
        return typeof value === "number" && value > 0;
      }).length;

    expect(formulaRegressionCorpus).toHaveLength(379);
    expect(
      new Set(formulaRegressionCorpus.map((fixture) => fixture.workbookId)),
    ).toHaveLength(5);
    expect(countUsing("additions")).toBe(314);
    expect(countUsing("divisions")).toBe(335);
    expect(countUsing("decimalLiterals")).toBe(43);
    expect(countUsing("subtractions")).toBe(0);
    expect(countUsing("multiplications")).toBe(0);
    expect(countUsing("percentages")).toBe(0);
    expect(countUsing("unarySigns")).toBe(0);
    expect(
      formulaRegressionCorpus.some((fixture) =>
        /\b[A-Z]+\d+\b/u.test(fixture.formula),
      ),
    ).toBe(true);
    expect(
      formulaRegressionCorpus.some((fixture) => fixture.formula.includes("$")),
    ).toBe(false);
    expect(
      formulaRegressionCorpus.some((fixture) => fixture.formula.includes("!")),
    ).toBe(false);
    expect(
      formulaRegressionCorpus.some((fixture) => /[A-Z]+\(/u.test(fixture.formula)),
    ).toBe(false);
  });

  it("evaluates all 379 cells to their deterministic expected values", () => {
    for (const fixture of formulaRegressionCorpus) {
      const result = formulaEngine.evaluate(
        [
          ...Object.entries(fixture.inputs).map(([key, value]) => ({
            key,
            value,
          })),
          {
            key: "result",
            formula: fixture.formula,
            displayPrecision: 2,
          },
        ],
        { referenceAliases: fixture.a1ToField },
      );

      expect(result.fields.result.error, fixture.id).toBeUndefined();
      expect(result.fields.result.canonical, fixture.id).toBe(fixture.expected);
    }
  });
});