import { expect, it } from "@jest/globals";

import { whereAndSelect } from "@lib/prisma";

it("can decorate using whereAndSelect", async () => {
  whereAndSelect(
    (queryObj) => {
      expect(queryObj).toStrictEqual({ where: { id: 1 }, select: { example: true } });
    },
    { id: 1 },
    ["example"]
  );
});

it("can do nested selects using . seperator", async () => {
  whereAndSelect(
    (queryObj) => {
      expect(queryObj).toStrictEqual({
        where: {
          uid: 1,
        },
        select: {
          description: true,
          attendees: {
            select: {
              email: true,
              name: true,
            },
          },
        },
      });
    },
    { uid: 1 },
    ["description", "attendees.email", "attendees.name"]
  );
});

it("can handle nesting deeply", async () => {
  whereAndSelect(
    (queryObj) => {
      expect(queryObj).toStrictEqual({
        where: {
          uid: 1,
        },
        select: {
          description: true,
          attendees: {
            select: {
              email: {
                select: {
                  nested: true,
                },
              },
              name: true,
            },
          },
        },
      });
    },
    { uid: 1 },
    ["description", "attendees.email.nested", "attendees.name"]
  );
});

it("can handle nesting multiple", async () => {
  whereAndSelect(
    (queryObj) => {
      expect(queryObj).toStrictEqual({
        where: {
          uid: 1,
        },
        select: {
          description: true,
          attendees: {
            select: {
              email: true,
              name: true,
            },
          },
          bookings: {
            select: {
              id: true,
              name: true,
            },
          },
        },
      });
    },
    { uid: 1 },
    ["description", "attendees.email", "attendees.name", "bookings.id", "bookings.name"]
  );
});