import { vi, type Mock } from "vitest";
import type { PurchaseItemRepository } from "../repository/purchaseItemRepository";
import type { PurchaseOrderRepository } from "../repository/purchaseOrderRepository";

export interface PurchaseItemRepositoryMock extends PurchaseItemRepository {
  findByItemId: Mock<PurchaseItemRepository["findByItemId"]>;
  findByItemIds: Mock<PurchaseItemRepository["findByItemIds"]>;
  save: Mock<PurchaseItemRepository["save"]>;
  deleteByItemId: Mock<PurchaseItemRepository["deleteByItemId"]>;
}

export function mockPurchaseItemRepository(): PurchaseItemRepositoryMock {
  return {
    findByItemId: vi.fn<PurchaseItemRepository["findByItemId"]>().mockResolvedValue(null),
    findByItemIds: vi.fn<PurchaseItemRepository["findByItemIds"]>().mockResolvedValue(new Map()),
    save: vi.fn<PurchaseItemRepository["save"]>(),
    deleteByItemId: vi.fn<PurchaseItemRepository["deleteByItemId"]>(),
  };
}

export interface PurchaseOrderRepositoryMock extends PurchaseOrderRepository {
  findById: Mock<PurchaseOrderRepository["findById"]>;
  findByLineIds: Mock<PurchaseOrderRepository["findByLineIds"]>;
  save: Mock<PurchaseOrderRepository["save"]>;
}

// Injected through the commands' repositoryFactory parameter. vi.mock is not
// an option: vitest runs with isolate: false, which shares the module cache
// across test files.
export function mockPurchaseOrderRepository(): PurchaseOrderRepositoryMock {
  return {
    findById: vi.fn<PurchaseOrderRepository["findById"]>().mockResolvedValue(null),
    findByLineIds: vi.fn<PurchaseOrderRepository["findByLineIds"]>().mockResolvedValue([]),
    save: vi.fn<PurchaseOrderRepository["save"]>(),
  };
}