import { createKyselyMock } from "@tailor-platform/sdk/vitest";
import { describe, expect, it } from "vitest";
import type { Namespace } from "../generated/kysely-tailordb";
import { baseTaxIdIdentification } from "../testing/fixtures";
import { run } from "./getPartnerIdentification";

describe("getPartnerIdentification", () => {
  it("returns identification when found", async () => {
    const mock = createKyselyMock<Namespace["main-db"]>();
    mock.enqueueResults([baseTaxIdIdentification]);

    const result = await run(mock.db, { id: "ident-1" });

    expect(result.ok).toBe(true);
    if (result.ok) {
      expect(result.value.partnerIdentification).toEqual(baseTaxIdIdentification);
    }
  });

  it("returns null when not found", async () => {
    const mock = createKyselyMock<Namespace["main-db"]>();
    mock.enqueueResults([]);

    const result = await run(mock.db, { id: "nonexistent" });

    expect(result.ok).toBe(true);
    if (result.ok) {
      expect(result.value.partnerIdentification).toBeNull();
    }
  });
});
