import { PrismaReadService } from "@/modules/prisma/prisma-read.service";
import { PrismaWriteService } from "@/modules/prisma/prisma-write.service";
import { TestingModule } from "@nestjs/testing";

import type { Prisma } from "@calcom/prisma/client";

export class OOORepositoryFixture {
  private prismaReadClient: PrismaReadService["prisma"];
  private prismaWriteClient: PrismaWriteService["prisma"];

  constructor(module: TestingModule) {
    this.prismaReadClient = module.get(PrismaReadService).prisma;
    this.prismaWriteClient = module.get(PrismaWriteService).prisma;
  }

  async create(data: Prisma.OutOfOfficeEntryCreateInput) {
    return this.prismaWriteClient.outOfOfficeEntry.create({
      data: {
        ...data,
      },
    });
  }

  async delete(id: number) {
    return this.prismaWriteClient.outOfOfficeEntry.delete({
      where: {
        id,
      },
    });
  }
}