import dotenv from "dotenv";
import { createRuntime } from "../../test/createRuntime"; // Adjust the import path as needed
import { getOrCreateRelationship } from "../../test/getOrCreateRelationship";
import { type User } from "../../test/types";
import { zeroUuid } from "../constants";
import { createRelationship, getRelationships } from "../relationships"; // Adjust the import path as needed
import { BgentRuntime } from "../runtime";
import { type UUID } from "../types";

dotenv.config({ path: ".dev.vars" });

describe("Relationships Module", () => {
  let runtime: BgentRuntime;
  let user: User;

  beforeAll(async () => {
    const setup = await createRuntime({
      env: process.env as Record<string, string>,
    });
    runtime = setup.runtime;
    user = setup.session.user;
    if (!user.id) {
      throw new Error("User ID is undefined");
    }
  });

  test("createRelationship creates a new relationship", async () => {
    const userA = user.id as UUID;
    const userB = zeroUuid;
    if (userA === undefined) throw new Error("userA is undefined");
    const relationship = await createRelationship({
      runtime,
      userA,
      userB,
    });
    expect(relationship).toBe(true);
  });

  test("getRelationship retrieves an existing relationship", async () => {
    const userA = user?.id as UUID;
    const userB = zeroUuid;

    await createRelationship({ runtime, userA, userB });

    const relationship = await getOrCreateRelationship({
      runtime,
      userA,
      userB,
    });
    expect(relationship).toBeDefined();
    expect(relationship?.user_a).toBe(userA);
    expect(relationship?.user_b).toBe(userB);
  });

  test("getRelationships retrieves all relationships for a user", async () => {
    const userA = user?.id as UUID;
    const userB = zeroUuid;

    await createRelationship({ runtime, userA, userB });

    const relationships = await getRelationships({
      runtime,
      user_id: userA,
    });
    expect(relationships).toBeDefined();
    expect(relationships.length).toBeGreaterThan(0);
    expect(
      relationships.some((r) => r.user_a === userA || r.user_b === userA),
    ).toBeTruthy();
  });
});