openclaw/src/plugin-sdk/dedupe-runtime.test.ts

import { afterEach, describe, expect, it, vi } from "vitest";
import { createPersistentDedupeCache } from "./dedupe-runtime.js";

type Record = { at: number };

function createMemoryStore() {
  const entries = new Map<string, Record>();
  return {
    entries,
    store: {
      register: vi.fn(async (key: string, value: Record) => {
        entries.set(key, value);
      }),
      lookup: vi.fn(async (key: string) => entries.get(key)),
    },
  };
}

function createCache(params?: {
  openStore?: () => ReturnType<typeof createMemoryStore>["store"] | undefined;
  logError?: (error: unknown) => void;
  readTimestamp?: (record: Record) => number | undefined;
}) {
  const backing = createMemoryStore();
  const cache = createPersistentDedupeCache<Record>({
    // Plain Symbol() is unique per cache, so parallel tests never share memory layers.
    globalKey: Symbol("test.persistent-dedupe"),
    ttlMs: 60_000,
    maxSize: 100,
    persistent: {
      namespace: "test.persistent-dedupe",
      maxEntries: 100,
      openStore: params?.openStore ?? (() => backing.store),
      logError: params?.logError,
      readTimestamp: params?.readTimestamp,
    },
  });
  return { cache, backing };
}

describe("createPersistentDedupeCache", () => {
  afterEach(() => {
    vi.useRealTimers();
  });

  it("records presence in both layers and answers from memory first", async () => {
    const { cache, backing } = createCache();
    await cache.register("k1", { at: 1 });
    expect(cache.peek("k1")).toBe(true);
    expect(await cache.lookup("k1")).toBe(true);
    expect(backing.store.register).toHaveBeenCalledWith("k1", { at: 1 });
    expect(backing.store.lookup).not.toHaveBeenCalled();
  });

  it("falls back to persistence and re-primes memory on a hit", async () => {
    const { cache, backing } = createCache();
    backing.entries.set("k2", { at: 42 });
    expect(cache.peek("k2")).toBe(false);
    expect(await cache.lookup("k2")).toBe(true);
    expect(cache.peek("k2")).toBe(true);
  });

  it("re-primes memory with the persisted timestamp when provided", async () => {
    vi.useFakeTimers();
    vi.setSystemTime(1_000_000);
    const { cache, backing } = createCache({ readTimestamp: (record) => record.at });
    backing.entries.set("k3", { at: 1_000_000 - 59_000 });
    expect(await cache.lookup("k3")).toBe(true);
    // Re-primed at the original timestamp: expires 59s later instead of a fresh 60s TTL.
    vi.setSystemTime(1_000_000 + 2_000);
    expect(cache.peek("k3")).toBe(false);
  });

  it("disables persistence after an open failure and never rejects", async () => {
    const logError = vi.fn();
    const openStore = vi.fn(() => {
      throw new Error("sqlite unavailable");
    });
    const { cache } = createCache({ openStore, logError });
    await expect(cache.register("k4", { at: 1 })).resolves.toBeUndefined();
    expect(cache.peek("k4")).toBe(true);
    expect(await cache.lookup("k5")).toBe(false);
    expect(openStore).toHaveBeenCalledTimes(1);
    expect(logError).toHaveBeenCalledTimes(1);
  });

  it("disables persistence after a lookup failure", async () => {
    const logError = vi.fn();
    const store = {
      register: vi.fn(async () => {}),
      lookup: vi.fn(async () => {
        throw new Error("read failed");
      }),
    };
    const { cache } = createCache({ openStore: () => store, logError });
    expect(await cache.lookup("k6")).toBe(false);
    expect(logError).toHaveBeenCalledTimes(1);
    await cache.register("k7", { at: 1 });
    expect(store.register).not.toHaveBeenCalled();
  });

  it("clearForTest resets memory and re-enables persistence", async () => {
    const openStore = vi
      .fn()
      .mockImplementationOnce(() => {
        throw new Error("boom");
      })
      .mockImplementation(() => createMemoryStore().store);
    const { cache } = createCache({ openStore });
    await cache.register("k8", { at: 1 });
    cache.clearForTest();
    expect(cache.peek("k8")).toBe(false);
    await cache.register("k9", { at: 1 });
    expect(openStore).toHaveBeenCalledTimes(2);
  });
});