From 2b7db6b6b2af5cfbe67e25f2f2f9e77e8b169b79 Mon Sep 17 00:00:00 2001
From: Ani Ravi <5902976+aniravi24@users.noreply.github.com>
Date: Mon, 17 Feb 2025 22:21:00 -0500
Subject: [PATCH] test(sidetrack): add tests for unique key functionality

---
 packages/sidetrack/test/index.test.ts | 69 +++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/packages/sidetrack/test/index.test.ts b/packages/sidetrack/test/index.test.ts
index cd889834..f4ee36ce 100644
--- 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<{