Merge pull request #274 from sidetracklabs/unique-key-tests

test(sidetrack): add tests for unique key functionality
sidetracklabs · Feb 18, 2025 · 3cdb28e · 3cdb28e
2 parents 6e900c5 + 2b7db6b
commit 3cdb28e
Showing 1 changed file with 69 additions and 0 deletions.
diff --git a/packages/sidetrack/test/index.test.ts b/packages/sidetrack/test/index.test.ts
@@ -306,6 +306,75 @@ describe.concurrent("jobs", () => {
     });
   });
 
+  it("unique key prevents duplicate jobs", async () => {
+    await runInTransaction(pool, async (client) => {
+      const sidetrack = new SidetrackTest<{
+        test: { id: string };
+      }>({
+        dbClient: usePg(client),
+        queues: {
+          test: {
+            run: async (payload) => {
+              return payload;
+            },
+          },
+        },
+      });
+
+      await sidetrack.insertJob(
+        "test",
+        { id: "unique job" },
+        { uniqueKey: "test-unique-key" },
+      );
+
+      // Attempting to insert another job with same unique key should throw
+      await expect(
+        sidetrack.insertJob(
+          "test",
+          { id: "unique job 2" },
+          { uniqueKey: "test-unique-key" },
+        ),
+      ).rejects.toThrow();
+    });
+  });
+
+  it("can suppress unique key errors", async () => {
+    await runInTransaction(pool, async (client) => {
+      const sidetrack = new SidetrackTest<{
+        test: { id: string };
+      }>({
+        dbClient: usePg(client),
+        queues: {
+          test: {
+            run: async (payload) => {
+              return payload;
+            },
+          },
+        },
+      });
+
+      await sidetrack.insertJob(
+        "test",
+        { id: "suppress unique 1" },
+        { uniqueKey: "suppress-key" },
+      );
+
+      // Should not throw when suppressDuplicateUniqueKeyErrors is true
+      await sidetrack.insertJob(
+        "test",
+        { id: "suppress unique 2" },
+        {
+          suppressDuplicateUniqueKeyErrors: true,
+          uniqueKey: "suppress-key",
+        },
+      );
+
+      const jobs = await sidetrack.listJobs({ queue: ["test"] });
+      expect(jobs.length).toBe(1);
+      expect(jobs[0].payload).toEqual({ id: "suppress unique 1" });
+    });
+  });
+
   it("payload transformer works", async () => {
     await runInTransaction(pool, async (client) => {
       const sidetrack = new SidetrackTest<{