File size: 2,439 Bytes
76289e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | import { StatementSync } from "node:sqlite";
import { afterEach, expect, it, vi } from "vitest";
import { openOpenClawStateDatabase } from "../state/openclaw-state-db.js";
import { withOpenClawTestState } from "../test-utils/openclaw-test-state.js";
import {
createPluginBlobStoreForTests,
resetPluginBlobStoreForTests,
} from "./plugin-blob-store.js";
afterEach(() => {
vi.restoreAllMocks();
resetPluginBlobStoreForTests();
});
it("fetches only the eviction prefix needed for a bounded blob write", async () => {
await withOpenClawTestState({ label: "blob-eviction-prefix" }, async (state) => {
const store = createPluginBlobStoreForTests(
"diffs",
{
namespace: "artifacts",
maxEntries: 64,
maxBytesPerEntry: 16,
maxBytesPerNamespace: 1024,
},
state.env,
);
const { db } = openOpenClawStateDatabase({ env: state.env });
const insert = db.prepare(`INSERT INTO plugin_blob_entries
(plugin_id, namespace, entry_key, metadata_json, blob, created_at, expires_at)
VALUES ('diffs', 'artifacts', ?, '{}', ?, ?, NULL)`);
for (let index = 0; index < 64; index += 1) {
insert.run(`old-${index}`, new Uint8Array([index]), index);
}
let fetchedRows = 0;
// eslint-disable-next-line @typescript-eslint/unbound-method -- Invoke with the original native receiver.
const iterate = StatementSync.prototype.iterate;
const reads = vi.spyOn(StatementSync.prototype, "iterate").mockImplementation(function (
this: StatementSync,
...args
) {
const iterator = iterate.apply(this, args);
const next = iterator.next.bind(iterator);
vi.spyOn(iterator, "next").mockImplementation((...input) => {
const result = next(...input);
if (!result.done) {
fetchedRows += 1;
}
return result;
});
return iterator;
});
await store.register("new", new Uint8Array([255]), { retained: true });
reads.mockRestore();
expect(fetchedRows).toBeLessThanOrEqual(2);
await expect(store.lookup("old-0")).resolves.toBeUndefined();
await expect(store.lookup("old-1")).resolves.toMatchObject({ bytes: new Uint8Array([1]) });
await expect(store.lookup("new")).resolves.toMatchObject({ metadata: { retained: true } });
expect(await store.entries()).toHaveLength(64);
await store.clear();
expect(await store.entries()).toEqual([]);
});
});
|