import { describe, expect, it, vi } from "vitest";
import { planAssistantMcpDiscovery } from "../src/services/assistantMcpDiscoveryPlanner";
import { executeAssistantMcpDiscoveryPilot } from "../src/services/assistantMcpDiscoveryPilotExecutor";

function buildDeps(rows: Array<Record<string, unknown>>, error: string | null = null) {
  return {
    executeAddressMcpQuery: vi.fn(async () => ({
      fetched_rows: rows.length,
      matched_rows: error ? 0 : rows.length,
      raw_rows: rows,
      rows: error ? [] : rows,
      error
    }))
  };
}

describe("assistant MCP discovery pilot executor", () => {
  it("executes only the lifecycle query_documents primitive through injected MCP deps", async () => {
    const planner = planAssistantMcpDiscovery({
      turnMeaning: {
        asked_domain_family: "counterparty_lifecycle",
        asked_action_family: "activity_duration",
        explicit_entity_candidates: ["SVK"]
      }
    });
    const deps = buildDeps([
      { Период: "2020-01-15T00:00:00", Регистратор: "CP_CUSTOMER_ACTIVITY_FIRST", Контрагент: "SVK" },
      { Период: "2023-12-20T00:00:00", Регистратор: "CP_CUSTOMER_ACTIVITY", Контрагент: "SVK" }
    ]);

    const result = await executeAssistantMcpDiscoveryPilot(planner, deps);

    expect(result.pilot_status).toBe("executed");
    expect(result.mcp_execution_performed).toBe(true);
    expect(result.executed_primitives).toEqual(["query_documents"]);
    expect(result.skipped_primitives).toEqual(["resolve_entity_reference", "probe_coverage", "explain_evidence_basis"]);
    expect(result.evidence.evidence_status).toBe("confirmed");
    expect(result.evidence.confirmed_facts[0]).toContain("SVK");
    expect(result.evidence.inferred_facts[0]).toContain("may be inferred");
    expect(result.evidence.unknown_facts).toContain("Legal registration date is not proven by this MCP discovery pilot");
    expect(result.source_rows_summary).toBe("2 MCP document rows fetched, 2 matched lifecycle scope");
    expect(result.derived_activity_period).toEqual({
      first_activity_date: "2020-01-15",
      latest_activity_date: "2023-12-20",
      matched_rows: 2,
      duration_total_months: 47,
      duration_years: 3,
      duration_months_remainder: 11,
      duration_human_ru: "3 года 11 месяцев",
      inference_basis: "first_and_latest_confirmed_1c_activity_rows"
    });
    expect(result.reason_codes).toContain("pilot_query_documents_mcp_executed");
    expect(result.reason_codes).toContain("pilot_derived_activity_period_from_confirmed_rows");

    expect(deps.executeAddressMcpQuery).toHaveBeenCalledTimes(1);
    const call = deps.executeAddressMcpQuery.mock.calls[0]?.[0];
    expect(String(call?.query ?? "")).toContain("Документ.ПоступлениеНаРасчетныйСчет");
    expect(call?.limit).toBeGreaterThan(0);
  });

  it("does not execute MCP when dry-run still needs clarification", async () => {
    const planner = planAssistantMcpDiscovery({
      turnMeaning: {
        asked_domain_family: "counterparty_value",
        asked_action_family: "turnover",
        explicit_entity_candidates: ["SVK"]
      }
    });
    const deps = buildDeps([]);

    const result = await executeAssistantMcpDiscoveryPilot(planner, deps);

    expect(result.pilot_status).toBe("skipped_needs_clarification");
    expect(result.mcp_execution_performed).toBe(false);
    expect(result.evidence.evidence_status).toBe("insufficient");
    expect(deps.executeAddressMcpQuery).not.toHaveBeenCalled();
  });

  it("executes value-flow query_movements and derives a guarded turnover sum", async () => {
    const planner = planAssistantMcpDiscovery({
      turnMeaning: {
        asked_domain_family: "counterparty_value",
        asked_action_family: "turnover",
        explicit_entity_candidates: ["SVK"],
        explicit_date_scope: "2020"
      }
    });
    const deps = buildDeps([
      { Period: "2020-01-15T00:00:00", Amount: 1250, Counterparty: "SVK" },
      { Period: "2020-02-20T00:00:00", Amount: "2 500,50", Counterparty: "SVK" }
    ]);

    const result = await executeAssistantMcpDiscoveryPilot(planner, deps);

    expect(result.pilot_status).toBe("executed");
    expect(result.pilot_scope).toBe("counterparty_value_flow_query_movements_v1");
    expect(result.mcp_execution_performed).toBe(true);
    expect(result.executed_primitives).toEqual(["query_movements"]);
    expect(result.skipped_primitives).toEqual(["resolve_entity_reference", "aggregate_by_axis", "probe_coverage"]);
    expect(result.evidence.evidence_status).toBe("confirmed");
    expect(result.evidence.confirmed_facts[0]).toContain("value-flow rows");
    expect(result.source_rows_summary).toBe("2 MCP value-flow rows fetched, 2 matched value-flow scope");
    expect(result.derived_value_flow).toMatchObject({
      counterparty: "SVK",
      period_scope: "2020",
      rows_matched: 2,
      rows_with_amount: 2,
      total_amount: 3750.5,
      first_movement_date: "2020-01-15",
      latest_movement_date: "2020-02-20",
      inference_basis: "sum_of_confirmed_1c_value_flow_rows"
    });
    expect(result.reason_codes).toContain("pilot_query_movements_mcp_executed");
    expect(result.reason_codes).toContain("pilot_derived_value_flow_from_confirmed_rows");

    expect(deps.executeAddressMcpQuery).toHaveBeenCalledTimes(1);
    const call = deps.executeAddressMcpQuery.mock.calls[0]?.[0];
    expect(String(call?.query ?? "")).toContain("ПоступлениеНаРасчетныйСчет");
    expect(call?.limit).toBeGreaterThan(0);
  });

  it("keeps non-lifecycle ready plans unsupported until a dedicated pilot exists", async () => {
    const planner = planAssistantMcpDiscovery({
      turnMeaning: {
        asked_domain_family: "counterparty_documents",
        asked_action_family: "list_documents",
        explicit_entity_candidates: ["SVK"]
      }
    });
    const deps = buildDeps([]);

    const result = await executeAssistantMcpDiscoveryPilot(planner, deps);

    expect(result.pilot_status).toBe("unsupported");
    expect(result.mcp_execution_performed).toBe(false);
    expect(result.skipped_primitives).toEqual(["resolve_entity_reference", "query_documents", "probe_coverage"]);
    expect(result.reason_codes).toContain("pilot_scope_unsupported_for_live_execution");
    expect(deps.executeAddressMcpQuery).not.toHaveBeenCalled();
  });

  it("records MCP errors as limitations without converting them into facts", async () => {
    const planner = planAssistantMcpDiscovery({
      turnMeaning: {
        asked_domain_family: "counterparty_lifecycle",
        asked_action_family: "activity_duration",
        explicit_entity_candidates: ["SVK"]
      }
    });
    const deps = buildDeps([], "MCP fetch failed: timeout");

    const result = await executeAssistantMcpDiscoveryPilot(planner, deps);

    expect(result.pilot_status).toBe("executed");
    expect(result.mcp_execution_performed).toBe(true);
    expect(result.evidence.evidence_status).toBe("insufficient");
    expect(result.evidence.confirmed_facts).toEqual([]);
    expect(result.derived_activity_period).toBeNull();
    expect(result.query_limitations).toContain("MCP fetch failed: timeout");
    expect(result.reason_codes).toContain("pilot_query_documents_mcp_error");
  });
});