mem0/mem0-ts/src/client/tests/integration/search.test.ts

/**
 * Integration tests: Search and history operations.
 *
 * Tests search, filtered search, and memory history against the real API.
 *
 * Run: MEM0_API_KEY=your-key npx jest search.test.ts --forceExit
 */
import { MemoryClient } from "../../mem0";
import { randomUUID } from "crypto";
import {
  describeIntegration,
  createTestClient,
  suppressTelemetryNoise,
  seedTestMemories,
  cleanupTestUser,
  waitForSearchResults,
} from "./helpers";

jest.setTimeout(120_000);

const TEST_USER_ID = `integration-search-${randomUUID()}`;

describeIntegration("MemoryClient Integration — Search & History", () => {
  let client: MemoryClient;
  let cleanup: () => void;
  let memoryIds: string[] = [];

  beforeAll(async () => {
    cleanup = suppressTelemetryNoise();
    client = createTestClient();
    memoryIds = await seedTestMemories(client, TEST_USER_ID);
  });

  afterAll(async () => {
    await cleanupTestUser(client, TEST_USER_ID);
    cleanup();
  });

  // ─── Search ─────────────────────────────────────────────
  describe("search", () => {
    test("searches memories by user_id and returns results with scores", async () => {
      // Search index may lag behind listing index — poll until ready
      const results = await waitForSearchResults(
        client,
        "What is my favorite color?",
        { filters: { userId: TEST_USER_ID } },
      );

      expect(Array.isArray(results)).toBe(true);
      expect(results.length).toBeGreaterThan(0);

      const first = results[0];
      expect(typeof first.id).toBe("string");
      expect(typeof first.memory).toBe("string");
      expect(typeof first.score).toBe("number");
      expect(first.score).toBeGreaterThan(0);
    });
  });

  // ─── Search with filters ─────────────────────────────────
  describe("search with filters", () => {
    test("searches with OR filters and returns results", async () => {
      const results = await waitForSearchResults(
        client,
        "What do you know about me?",
        {
          filters: { OR: [{ userId: TEST_USER_ID }] },
        },
      );

      expect(Array.isArray(results)).toBe(true);
      expect(results.length).toBeGreaterThan(0);

      const first = results[0];
      expect(typeof first.id).toBe("string");
      expect(typeof first.memory).toBe("string");
      expect(typeof first.score).toBe("number");
    });
  });

  // ─── History ──────────────────────────────────────────────
  describe("memory history", () => {
    test("returns history with at least an ADD event", async () => {
      const memoryId = memoryIds[0];
      const history = await client.history(memoryId);

      expect(Array.isArray(history)).toBe(true);
      expect(history.length).toBeGreaterThanOrEqual(1);

      const entry = history[0];
      expect(typeof entry.id).toBe("string");
      expect(typeof entry.memoryId).toBe("string");
      expect(["ADD", "UPDATE", "DELETE", "NOOP"]).toContain(entry.event);
      expect(new Date(entry.createdAt).toString()).not.toBe("Invalid Date");
      expect(new Date(entry.updatedAt).toString()).not.toBe("Invalid Date");
      expect(
        entry.newMemory === null || typeof entry.newMemory === "string",
      ).toBe(true);
      expect(
        entry.oldMemory === null || typeof entry.oldMemory === "string",
      ).toBe(true);

      const events = history.map((h) => h.event);
      expect(events).toContain("ADD");
    });
  });

  // ─── Edge cases ─────────────────────────────────────────
  describe("edge cases", () => {
    test("search for non-existent user returns empty results", async () => {
      const results = await client.search("anything", {
        filters: { userId: `nonexistent-user-${randomUUID()}` },
      });

      expect(Array.isArray(results)).toBe(true);
      expect(results.length).toBe(0);
    });

    test("search with top_k param does not throw", async () => {
      const results = await client.search(
        "Tell me about integration test user",
        {
          filters: { userId: TEST_USER_ID },
          topK: 1,
        },
      );

      expect(Array.isArray(results)).toBe(true);
    });
  });
});