Added error throwing tests to runlock

Author: matt-aitken

internal-packages/run-engine/src/engine/tests/locking.test.ts

@@ -45,4 +45,58 @@ describe("RunLocker", () => {
       await redis.quit();
     }
   });
+
+  redisTest(
+    "Test lock throws when callback throws",
+    { timeout: 15_000 },
+    async ({ redisOptions }) => {
+      const redis = createRedisClient(redisOptions);
+      try {
+        const runLock = new RunLocker({ redis });
+
+        expect(runLock.isInsideLock()).toBe(false);
+
+        await expect(
+          runLock.lock(["test-1"], 5000, async () => {
+            throw new Error("Test error");
+          })
+        ).rejects.toThrow("Test error");
+
+        // Verify the lock was released
+        expect(runLock.isInsideLock()).toBe(false);
+      } finally {
+        await redis.quit();
+      }
+    }
+  );
+
+  redisTest(
+    "Test nested lock throws when inner callback throws",
+    { timeout: 15_000 },
+    async ({ redisOptions }) => {
+      const redis = createRedisClient(redisOptions);
+      try {
+        const runLock = new RunLocker({ redis });
+
+        expect(runLock.isInsideLock()).toBe(false);
+
+        await expect(
+          runLock.lock(["test-1"], 5000, async () => {
+            expect(runLock.isInsideLock()).toBe(true);
+
+            // Nested lock with same resource
+            await runLock.lock(["test-1"], 5000, async () => {
+              expect(runLock.isInsideLock()).toBe(true);
+              throw new Error("Inner lock error");
+            });
+          })
+        ).rejects.toThrow("Inner lock error");
+
+        // Verify all locks were released
+        expect(runLock.isInsideLock()).toBe(false);
+      } finally {
+        await redis.quit();
+      }
+    }
+  );
 });