162 lines
7.2 KiB
TypeScript
162 lines
7.2 KiB
TypeScript
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");
|
||
});
|
||
});
|