Jyotisha/frontend/tests/consultation-thinking-plan.test.ts

import assert from "node:assert/strict";
import test from "node:test";

import {
  applyThinkingSectionProgress,
  consultationContinueMessages,
  consultationContinuePrompt,
  consultationReportHeadings,
  consultationSpokenHeadingRule,
  dailyAnswerShapeInstruction,
  dailyConsultationThinkingPlan,
  natalAnswerShapeInstruction,
  DAILY_HEADING,
  natalConsultationThinkingPlan,
  REPORT_HEADING,
  splitAnswerByHeadings,
  thinkPlanSteps,
  visibleThinkingSteps,
  windowConsultationThinkingPlan,
} from "../src/lib/consultation-thinking-plan.ts";

test("multi-domain thinking plan uses Chinese product labels and hides tool ids", () => {
  const sections = natalConsultationThinkingPlan({
    domains: ["career", "wealth"],
    requiredBlocks: [
      "raw_structure",
      "raman_six_step",
      "yoga_table",
      "timing",
      "synthesis",
      "technique_audit_table",
      "modern_wrap",
    ],
    mustUseLayers: ["D1", "run-jyotish-consultation", "D10", "skill_read"],
  });
  const encoded = JSON.stringify(sections);
  assert.deepEqual(sections.map((section) => section.heading), [
    REPORT_HEADING.question,
    REPORT_HEADING.support,
    REPORT_HEADING.timing,
    REPORT_HEADING.action,
  ]);
  assert.ok(sections.some((section) => section.id === "support"));
  assert.ok(JSON.stringify(sections).includes("事业"));
  assert.equal(encoded.includes("run-jyotish"), false);
  assert.equal(encoded.includes("skill_read"), false);
  assert.ok(sections.some((section) => section.steps.some((step) => step.label.includes("D1"))));
});

test("window thinking plan stays on stable-layer copy", () => {
  const [section] = windowConsultationThinkingPlan();
  assert.equal(section?.heading, REPORT_HEADING.question);
  assert.equal(JSON.stringify(section).includes("run-jyotish"), false);
});

test("visible steps collapse extras into a remaining count", () => {
  const steps = Array.from({ length: 6 }, (_, index) => ({
    id: `s${index}`,
    label: `步骤 ${index + 1}`,
    status: "pending" as const,
  }));
  const visible = visibleThinkingSteps(steps);
  assert.equal(visible.visible.length, 4);
  assert.equal(visible.hiddenCount, 2);
});

test("answer split follows the required heading order", () => {
  const headings = consultationReportHeadings(["career", "wealth"]);
  const text = [
    `## ${REPORT_HEADING.question}`,
    "外松内紧。",
    `## ${REPORT_HEADING.support}`,
    "事业宫被土星压着。",
    `## ${REPORT_HEADING.timing}`,
    "这轮大运还没过完。",
    `## ${REPORT_HEADING.action}`,
    "这周只把合同条款写清楚。",
  ].join("\n");
  const { slices } = splitAnswerByHeadings(text, headings);
  assert.match(slices[REPORT_HEADING.question] ?? "", /外松内紧/);
  assert.match(slices[REPORT_HEADING.support] ?? "", /土星/);
  assert.match(slices[REPORT_HEADING.timing] ?? "", /大运/);
  assert.match(slices[REPORT_HEADING.action] ?? "", /合同条款/);
});

test("settled answers mark leftover thinking sections done", () => {
  const sections = natalConsultationThinkingPlan({ domains: ["career"] });
  const progressed = applyThinkingSectionProgress(
    sections,
    "今日行运已经写完，但没有领域二级标题。",
    { settled: true },
  );
  assert.equal(progressed.every((section) => section.steps.every((step) => step.status === "done")), true);
});

test("progress marks the current heading active and completed ones done", () => {
  const sections = natalConsultationThinkingPlan({ domains: ["career"] });
  const progressed = applyThinkingSectionProgress(sections, `## ${REPORT_HEADING.question}\n外松内紧。\n`);
  assert.equal(progressed[0]?.steps[0]?.status, "done");
  assert.equal(progressed[1]?.steps[0]?.status, "active");
});

test("continue prompt asks to resume after the last complete heading", () => {
  const prompt = consultationContinuePrompt("## 事业\n方向稳定。\n");
  assert.match(prompt, /最后一个完整标题是：## 事业/);
  assert.match(prompt, /不要重复已写出的段落/);
  assert.match(prompt, /不要写思考过程清单/);
});

test("natal spoken answers must use markdown lists for parallel points", () => {
  assert.match(consultationSpokenHeadingRule("natal"), /Markdown bullet lists/);
});

test("think plan flattens natal sections into v2 steps", () => {
  const steps = thinkPlanSteps(natalConsultationThinkingPlan({
    domains: ["career", "wealth"],
  }));
  assert.deepEqual(steps.map((step) => step.id), ["question", "support", "timing", "action"]);
});

test("section prompt asks for one heading and forbids another calculation", () => {
  // 原值: 锁 consultationSectionPrompt（只写一节、不要再调用排盘工具）与 consultationComposePrompt 禁写统一参数
  // 新值: 两个提示词都已删除；改锁搬进用户轮的本命写作要求：只用本轮计算结果、开场无标题、
  //       一次写完四个标题、禁写统一参数与技法审计表
  // 原因: BUG-1053 删除单独的 compose 流（分段写作早在 BUG-944 已取消，consultationSectionPrompt 是死代码）；
  //       写作形态要求改由主循环看到的用户轮携带。测试名保留，避免名单无说明地消失
  const natal = natalAnswerShapeInstruction();
  assert.match(natal, /只用结果里的盘面事实/);
  assert.match(natal, /开场不要标题/);
  for (const heading of Object.values(REPORT_HEADING)) assert.ok(natal.includes(`## ${heading}`));
  assert.match(natal, /不要写「统一参数与原始结构」/);
  assert.match(natal, /不要写技法审计表/);
  assert.doesNotMatch(natal, /不要再调用排盘工具/, "the loop still has to call the tool before writing");
});

test("daily thinking plan is three sections without a close heading", () => {
  const plan = dailyConsultationThinkingPlan();
  assert.deepEqual(plan.map((section) => section.heading), [
    DAILY_HEADING.trend,
    DAILY_HEADING.actAvoid,
    DAILY_HEADING.action,
  ]);
  assert.equal(plan.at(-1)?.title, "一个行动 + 边界句");
  assert.equal(plan.some((section) => section.id === "close"), false);
  assert.deepEqual(
    thinkPlanSteps(plan).map((step) => step.id),
    ["daily-trend", "daily-act-avoid", "daily-action"],
  );
  const written = [
    `## ${DAILY_HEADING.trend}\n今日宜把已谈妥的细节落成文字。`,
    `## ${DAILY_HEADING.actAvoid}\n适合推进合同核对，避开新开战场。`,
    `## ${DAILY_HEADING.action}\n今晚只做一件：把待确认条款列成清单。探索性日提示，不是确定预测。`,
  ].join("\n\n");
  const next = applyThinkingSectionProgress(plan, written);
  assert.equal(next.every((section) => section.steps.every((step) => step.status === "done")), true);
  assert.equal(written.startsWith(`## ${DAILY_HEADING.trend}`), true);
  assert.match(consultationSpokenHeadingRule("daily"), /今日趋势/);
});

test("empty-retry section prompt asks to write the heading directly", () => {
  // 原值: 锁 consultationSectionPrompt 的 empty-retry 分支（「上一段没有输出正文，请直接写这一节」）
  // 新值: 该提示词已删除；改锁今日入口的写作要求（三节标题、审计表与边界句进最后一节）和
  //       续写消息：带上本轮计算结果，再接已写正文与续写要求
  // 原因: BUG-1053 删除 compose 提示词链；空回答走 answer-retry，续写不得看不到证据。测试名保留
  const daily = dailyAnswerShapeInstruction();
  assert.match(daily, new RegExp(`${DAILY_HEADING.trend}、${DAILY_HEADING.actAvoid.replace(/[/]/g, "\\/")}、${DAILY_HEADING.action}`));
  assert.match(daily, /探索性日提示，不是确定预测/);
  const base = [{ role: "user" as const, content: "问题" }];
  const withEvidence = consultationContinueMessages(base, `## ${REPORT_HEADING.question}\n岁差`, { moon: 3.2259 });
  assert.equal(withEvidence.length, 4);
  assert.match(String(withEvidence[1]?.content), /本轮服务器计算结果[\s\S]*3\.2259/);
  assert.equal(withEvidence[2]?.role, "assistant");
  assert.equal(withEvidence[3]?.content, consultationContinuePrompt(`## ${REPORT_HEADING.question}\n岁差`));
  assert.equal(consultationContinueMessages(base, "x").length, 3, "no evidence, no evidence message");
});