import dotenv from "dotenv";
import { createRuntime } from "../../../test/createRuntime";
import { GetTellMeAboutYourselfConversation1 } from "../../../test/data";
import { getOrCreateRelationship } from "../../../test/getOrCreateRelationship";
import { populateMemories } from "../../../test/populateMemories";
import { runAiTest } from "../../../test/runAiTest";
import { type User } from "../../../test/types";
import { zeroUuid } from "../../constants";
import { type BgentRuntime } from "../../runtime";
import { type Message, type UUID } from "../../types";
import action from "../wait"; // Import the wait action

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

describe("Wait Action Behavior", () => {
  let user: User;
  let runtime: BgentRuntime;
  let room_id: UUID;

  afterAll(async () => {
    await cleanup();
  });

  beforeEach(async () => {
    await cleanup();
  });

  beforeAll(async () => {
    const setup = await createRuntime({
      env: process.env as Record<string, string>,
      actions: [action],
    });
    user = setup.session.user;
    runtime = setup.runtime;

    const data = await getOrCreateRelationship({
      runtime,
      userA: user?.id as UUID,
      userB: zeroUuid,
    });

    if (!data) {
      throw new Error("Relationship not found");
    }

    room_id = data?.room_id;

    await cleanup();
  });

  async function cleanup() {
    await runtime.factManager.removeAllMemories(room_id);
    await runtime.messageManager.removeAllMemories(room_id);
  }

  test("Test wait action behavior", async () => {
    await runAiTest("Test wait action behavior", async () => {
      const message: Message = {
        user_id: zeroUuid as UUID,
        content: {
          content: "Please wait a moment, I need to think about this...",
          action: "WAIT",
        },
        room_id: room_id as UUID,
      };

      const handler = action.handler!;

      await populateMemories(runtime, user, room_id, [
        GetTellMeAboutYourselfConversation1,
      ]);

      const result = (await handler(runtime, message)) as boolean;
      // Expectation depends on the implementation of the wait action.
      // For instance, it might be that there's no immediate output,
      // or the output indicates waiting, so adjust the expectation accordingly.
      return result === true;
    });
  }, 60000);
});