indexer: add test for sqlite persistence

packages/indexer/src/plugins/persistence.test.ts

@@ -0,0 +1,61 @@
+import type { Cursor } from "@apibara/protocol";
+import { type Database, open } from "sqlite";
+import sqlite3 from "sqlite3";
+import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { SqlitePersistence } from "./persistence";
+
+describe("Sqlite Persistence", () => {
+  let db: Database<sqlite3.Database, sqlite3.Statement>;
+  let store: SqlitePersistence;
+
+  beforeAll(async () => {
+    db = await open({ driver: sqlite3.Database, filename: ":memory:" });
+    await SqlitePersistence.initialize(db);
+    store = new SqlitePersistence(db);
+  });
+
+  afterAll(async () => {
+    await db.close();
+  });
+
+  it("should store a cursor", async () => {
+    const cursor: Cursor = {
+      orderKey: 5_000_000n,
+    };
+    await store.put(cursor);
+  });
+
+  it("should get the last cursor", async () => {
+    const latest = await store.get();
+
+    expect(latest).toEqual({
+      orderKey: 5_000_000n,
+      uniqueKey: null,
+    });
+  });
+
+  it("should update last cursor", async () => {
+    const cursor: Cursor = {
+      orderKey: 5_000_010n,
+      uniqueKey: "0x1234567890",
+    };
+    await store.put(cursor);
+  });
+
+  it("should get updated cursor", async () => {
+    const latest = await store.get();
+
+    expect(latest).toEqual({
+      orderKey: 5_000_010n,
+      uniqueKey: "0x1234567890",
+    });
+  });
+
+  it("should delete persisted cursor", async () => {
+    await store.del();
+
+    const latest = await store.get();
+
+    expect(latest).toBeUndefined();
+  });
+});

packages/indexer/src/plugins/persistence.ts

@@ -39,7 +39,7 @@ export class SqlitePersistence {
         id TEXT NOT NULL PRIMARY KEY,
         order_key INTEGER NOT NULL,
         unique_key TEXT,
-        PRIMARY KEY (order_key, unique_key)
+        UNIQUE(order_key, unique_key)
       );
     `);
   }