import { vi, type Mock } from "vitest";
import type { SalesItemRepository } from "../repository/salesItemRepository";
import type { SalesOrderRepository } from "../repository/salesOrderRepository";

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

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

export interface SalesOrderRepositoryMock extends SalesOrderRepository {
  findById: Mock<SalesOrderRepository["findById"]>;
  findByLineIds: Mock<SalesOrderRepository["findByLineIds"]>;
  save: Mock<SalesOrderRepository["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 mockSalesOrderRepository(): SalesOrderRepositoryMock {
  return {
    findById: vi.fn<SalesOrderRepository["findById"]>().mockResolvedValue(null),
    findByLineIds: vi.fn<SalesOrderRepository["findByLineIds"]>().mockResolvedValue([]),
    save: vi.fn<SalesOrderRepository["save"]>(),
  };
}