import { vi, type Mock } from "vitest";
import type { BusinessPartnerRepository } from "../repository/businessPartnerRepository";
import type { CustomerAccountRepository } from "../repository/customerAccountRepository";
import type { SupplierAccountRepository } from "../repository/supplierAccountRepository";

export interface BusinessPartnerRepositoryMock extends BusinessPartnerRepository {
  findById: Mock<BusinessPartnerRepository["findById"]>;
  findByAddressId: Mock<BusinessPartnerRepository["findByAddressId"]>;
  findByBankAccountId: Mock<BusinessPartnerRepository["findByBankAccountId"]>;
  save: Mock<BusinessPartnerRepository["save"]>;
}

export function mockBusinessPartnerRepository(): BusinessPartnerRepositoryMock {
  return {
    findById: vi.fn<BusinessPartnerRepository["findById"]>().mockResolvedValue(null),
    findByAddressId: vi.fn<BusinessPartnerRepository["findByAddressId"]>().mockResolvedValue(null),
    findByBankAccountId: vi
      .fn<BusinessPartnerRepository["findByBankAccountId"]>()
      .mockResolvedValue(null),
    save: vi.fn<BusinessPartnerRepository["save"]>(),
  };
}

export interface CustomerAccountRepositoryMock extends CustomerAccountRepository {
  findById: Mock<CustomerAccountRepository["findById"]>;
  findByCompanyAndCode: Mock<CustomerAccountRepository["findByCompanyAndCode"]>;
  save: Mock<CustomerAccountRepository["save"]>;
}

export interface SupplierAccountRepositoryMock extends SupplierAccountRepository {
  findById: Mock<SupplierAccountRepository["findById"]>;
  findByCompanyAndCode: Mock<SupplierAccountRepository["findByCompanyAndCode"]>;
  save: Mock<SupplierAccountRepository["save"]>;
}

export function mockSupplierAccountRepository(): SupplierAccountRepositoryMock {
  return {
    findById: vi.fn<SupplierAccountRepository["findById"]>().mockResolvedValue(null),
    findByCompanyAndCode: vi
      .fn<SupplierAccountRepository["findByCompanyAndCode"]>()
      .mockResolvedValue(null),
    save: vi.fn<SupplierAccountRepository["save"]>(),
  };
}

// Injected through commands' repositoryFactory parameters. vi.mock is not an
// option because vitest runs with isolate: false and shares the module cache.
export function mockCustomerAccountRepository(): CustomerAccountRepositoryMock {
  return {
    findById: vi.fn<CustomerAccountRepository["findById"]>().mockResolvedValue(null),
    findByCompanyAndCode: vi
      .fn<CustomerAccountRepository["findByCompanyAndCode"]>()
      .mockResolvedValue(null),
    save: vi.fn<CustomerAccountRepository["save"]>(),
  };
}