import { describe, expect, it, vi } from "vitest";
import { runAssistantAddressLaneAttemptRuntime } from "../src/services/assistantAddressLaneAttemptRuntimeAdapter";

describe("assistant address lane attempt runtime adapter", () => {
  it("uses scoped followup context when available", async () => {
    const runAddressQueryTryHandle = vi.fn(async () => ({
      response_type: "READY"
    }));
    const result = await runAssistantAddressLaneAttemptRuntime({
      messageUsed: "msg",
      carryMeta: {
        followupContext: {
          previous_intent: "docs_by_counterparty"
        }
      },
      analysisDateHint: "2020-07-31",
      activeOrganization: "ООО Тест",
      knownOrganizations: ["ООО Тест", "ООО Лютик"],
      mergeFollowupContextWithOrganizationScope: () => ({
        previous_intent: "docs_by_counterparty",
        active_organization: "ООО Тест"
      }),
      runAddressQueryTryHandle
    });

    expect(runAddressQueryTryHandle).toHaveBeenCalledWith("msg", {
      followupContext: {
        previous_intent: "docs_by_counterparty",
        active_organization: "ООО Тест"
      },
      analysisDateHint: "2020-07-31",
      activeOrganization: "ООО Тест",
      knownOrganizations: ["ООО Тест", "ООО Лютик"]
    });
    expect(result).toEqual({
      response_type: "READY"
    });
  });

  it("falls back to plain attempt when scoped followup context is empty", async () => {
    const runAddressQueryTryHandle = vi.fn(async () => null);
    await runAssistantAddressLaneAttemptRuntime({
      messageUsed: "msg",
      carryMeta: null,
      analysisDateHint: null,
      activeOrganization: null,
      knownOrganizations: [],
      mergeFollowupContextWithOrganizationScope: () => null,
      runAddressQueryTryHandle
    });

    expect(runAddressQueryTryHandle).toHaveBeenCalledWith("msg", {
      analysisDateHint: null
    });
  });

  it("forwards llm semantic hints into query options", async () => {
    const runAddressQueryTryHandle = vi.fn(async () => ({
      response_type: "READY"
    }));

    await runAssistantAddressLaneAttemptRuntime({
      messageUsed: "что на складе конторы альтернатива",
      carryMeta: null,
      analysisDateHint: null,
      llmSemanticHints: {
        scope_target_kind: "organization",
        scope_target_text: "Альтернатива",
        date_scope_kind: "implicit_current",
        self_scope_detected: false,
        selected_object_scope_detected: false
      },
      activeOrganization: null,
      knownOrganizations: ["ООО Альтернатива Плюс"],
      mergeFollowupContextWithOrganizationScope: () => null,
      runAddressQueryTryHandle
    });

    expect(runAddressQueryTryHandle).toHaveBeenCalledWith("что на складе конторы альтернатива", {
      analysisDateHint: null,
      knownOrganizations: ["ООО Альтернатива Плюс"],
      llmSemanticHints: expect.objectContaining({
        scope_target_kind: "organization",
        scope_target_text: "Альтернатива"
      })
    });
  });
});