@fragno-dev/db 0.1.15 → 0.2.1

package/src/hooks/hooks.test.ts

@@ -0,0 +1,587 @@
+import SQLite from "better-sqlite3";
+import { SqliteDialect } from "kysely";
+import { beforeAll, describe, expect, it, vi } from "vitest";
+import { instantiate } from "@fragno-dev/core";
+import { prepareHookMutations, processHooks, type HooksMap, type HookContext } from "./hooks";
+import { internalFragmentDef, internalSchema } from "../fragments/internal-fragment";
+import type { FragnoPublicConfigWithDatabase } from "../db-fragment-definition-builder";
+import { DrizzleAdapter } from "../adapters/drizzle/drizzle-adapter";
+import { BetterSQLite3DriverConfig } from "../adapters/generic-sql/driver-config";
+import { ExponentialBackoffRetryPolicy, NoRetryPolicy } from "../query/unit-of-work/retry-policy";
+import type { FragnoId } from "../schema/create";
+
+describe("Hook System", () => {
+  let sqliteDatabase: SQLite.Database;
+  let adapter: DrizzleAdapter;
+  let internalFragment: ReturnType<typeof instantiateFragment>;
+
+  function instantiateFragment(options: FragnoPublicConfigWithDatabase) {
+    return instantiate(internalFragmentDef).withConfig({}).withOptions(options).build();
+  }
+
+  beforeAll(async () => {
+    sqliteDatabase = new SQLite(":memory:");
+
+    const dialect = new SqliteDialect({
+      database: sqliteDatabase,
+    });
+
+    adapter = new DrizzleAdapter({
+      dialect,
+      driverConfig: new BetterSQLite3DriverConfig(),
+    });
+
+    {
+      const migrations = adapter.prepareMigrations(internalSchema, "");
+      await migrations.executeWithDriver(adapter.driver, 0);
+    }
+
+    const options: FragnoPublicConfigWithDatabase = {
+      databaseAdapter: adapter,
+    };
+
+    internalFragment = instantiateFragment(options);
+
+    return async () => {
+      await adapter.close();
+    };
+  }, 12000);
+
+  describe("prepareHookMutations", () => {
+    it("should create hook records for triggered hooks", async () => {
+      const namespace = "test-namespace";
+      const hooks: HooksMap = {
+        onTest: vi.fn(),
+      };
+      const onSuccess = vi.fn();
+      const onBeforeMutate = vi.fn();
+
+      await internalFragment.inContext(async function () {
+        await this.uow(
+          async ({ forSchema, executeMutate }) => {
+            const uow = forSchema(internalSchema, hooks);
+
+            // Trigger a hook
+            uow.triggerHook("onTest", { data: "test" });
+
+            // Prepare hook mutations
+            prepareHookMutations(uow, {
+              hooks,
+              namespace,
+              internalFragment,
+              defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 5 }),
+            });
+
+            await executeMutate();
+          },
+          {
+            onSuccess,
+            onBeforeMutate,
+          },
+        );
+      });
+
+      // Verify callbacks were executed
+      expect(onSuccess).toHaveBeenCalledOnce();
+      expect(onBeforeMutate).toHaveBeenCalledOnce();
+
+      // Verify hook was created
+      const events = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ executeRetrieve }) => {
+          const result = internalFragment.services.hookService.getPendingHookEvents(namespace);
+          await executeRetrieve();
+          return result;
+        });
+      });
+
+      expect(events).toHaveLength(1);
+      expect(events[0]).toMatchObject({
+        hookName: "onTest",
+        payload: { data: "test" },
+        attempts: 0,
+        maxAttempts: 5,
+      });
+    });
+
+    it("should set maxAttempts to 1 when retry policy does not retry", async () => {
+      const namespace = "test-no-retry";
+      const hooks: HooksMap = {
+        onNoRetry: vi.fn(),
+      };
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema, hooks);
+
+          uow.triggerHook("onNoRetry", { data: "test" });
+
+          prepareHookMutations(uow, {
+            hooks,
+            namespace,
+            internalFragment,
+            defaultRetryPolicy: new NoRetryPolicy(),
+          });
+
+          await executeMutate();
+        });
+      });
+
+      const events = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ executeRetrieve }) => {
+          const result = internalFragment.services.hookService.getPendingHookEvents(namespace);
+          await executeRetrieve();
+          return result;
+        });
+      });
+
+      expect(events).toHaveLength(1);
+      expect(events[0]?.maxAttempts).toBe(1);
+    });
+
+    it("should use custom retry policy from trigger options", async () => {
+      const namespace = "test-custom-retry";
+      const hooks: HooksMap = {
+        onCustomRetry: vi.fn(),
+      };
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema, hooks);
+
+          uow.triggerHook(
+            "onCustomRetry",
+            { data: "test" },
+            {
+              retryPolicy: new NoRetryPolicy(),
+            },
+          );
+
+          prepareHookMutations(uow, {
+            hooks,
+            namespace,
+            internalFragment,
+            defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 10 }),
+          });
+
+          await executeMutate();
+        });
+      });
+
+      const events = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ executeRetrieve }) => {
+          const result = internalFragment.services.hookService.getPendingHookEvents(namespace);
+          await executeRetrieve();
+          return result;
+        });
+      });
+
+      expect(events[0]?.maxAttempts).toBe(1);
+    });
+  });
+
+  describe("processHooks", () => {
+    it("should execute hooks and mark them as completed", async () => {
+      const namespace = "test-success";
+      const hookFn = vi.fn();
+      const hooks: HooksMap = {
+        onSuccess: hookFn,
+      };
+
+      let eventId: FragnoId;
+
+      // Create a pending hook event
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          eventId = uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onSuccess",
+            payload: { email: "test@example.com" },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "test-nonce",
+          });
+          await executeMutate();
+        });
+      });
+
+      // Process hooks
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 3 }),
+      });
+
+      // Verify hook was called
+      expect(hookFn).toHaveBeenCalledOnce();
+      expect(hookFn).toHaveBeenCalledWith({ email: "test@example.com" });
+
+      // Verify hook context (this)
+      const hookContext = hookFn.mock.contexts[0] as HookContext;
+      expect(hookContext.nonce).toBe("test-nonce");
+
+      // Verify event was marked as completed
+      const result = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("primary", (eb) => eb("id", "=", eventId)),
+          );
+          await executeRetrieve();
+          const [events] = await findUow.retrievalPhase;
+          return events?.[0];
+        });
+      });
+
+      expect(result?.status).toBe("completed");
+      expect(result?.lastAttemptAt).toBeInstanceOf(Date);
+    });
+
+    it("should mark failed hooks for retry", async () => {
+      const namespace = "test-failure";
+      const hookFn = vi.fn().mockRejectedValue(new Error("Hook failed"));
+      const hooks: HooksMap = {
+        onFailure: hookFn,
+      };
+
+      let eventId: FragnoId;
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          eventId = uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onFailure",
+            payload: { data: "test" },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "test-nonce",
+          });
+          await executeMutate();
+        });
+      });
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 3 }),
+      });
+
+      expect(hookFn).toHaveBeenCalledOnce();
+
+      const result = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("primary", (eb) => eb("id", "=", eventId)),
+          );
+          await executeRetrieve();
+          const [events] = await findUow.retrievalPhase;
+          return events?.[0];
+        });
+      });
+
+      expect(result?.status).toBe("pending");
+      expect(result?.attempts).toBe(1);
+      expect(result?.error).toBe("Hook failed");
+      expect(result?.nextRetryAt).toBeInstanceOf(Date);
+    });
+
+    it("should mark failed hooks as permanently failed when max retries exceeded", async () => {
+      const namespace = "test-max-retries";
+      const hookFn = vi.fn().mockRejectedValue(new Error("Permanent failure"));
+      const hooks: HooksMap = {
+        onMaxRetries: hookFn,
+      };
+
+      let eventId: FragnoId;
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          eventId = uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onMaxRetries",
+            payload: { data: "test" },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 1,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "test-nonce",
+          });
+          await executeMutate();
+        });
+      });
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new NoRetryPolicy(),
+      });
+
+      const result = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("primary", (eb) => eb("id", "=", eventId)),
+          );
+          await executeRetrieve();
+          const [events] = await findUow.retrievalPhase;
+          return events?.[0];
+        });
+      });
+
+      expect(result?.status).toBe("failed");
+      expect(result?.attempts).toBe(1);
+      expect(result?.error).toBe("Permanent failure");
+    });
+
+    it("should handle missing hooks gracefully", async () => {
+      const namespace = "test-missing-hook";
+      const hooks: HooksMap = {
+        onExisting: vi.fn(),
+      };
+
+      let eventId: FragnoId;
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          eventId = uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onMissing",
+            payload: { data: "test" },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 1,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "test-nonce",
+          });
+          await executeMutate();
+        });
+      });
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new NoRetryPolicy(),
+      });
+
+      const result = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("primary", (eb) => eb("id", "=", eventId)),
+          );
+          await executeRetrieve();
+          const [events] = await findUow.retrievalPhase;
+          return events?.[0];
+        });
+      });
+
+      expect(result?.status).toBe("failed");
+      expect(result?.error).toBe("Hook 'onMissing' not found in hooks map");
+    });
+
+    it("should process multiple hooks in parallel", async () => {
+      const namespace = "test-parallel";
+      const hook1 = vi.fn();
+      const hook2 = vi.fn();
+      const hook3 = vi.fn();
+      const hooks: HooksMap = {
+        onHook1: hook1,
+        onHook2: hook2,
+        onHook3: hook3,
+      };
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook1",
+            payload: { id: 1 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-1",
+          });
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook2",
+            payload: { id: 2 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-2",
+          });
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook3",
+            payload: { id: 3 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-3",
+          });
+          await executeMutate();
+        });
+      });
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 3 }),
+      });
+
+      expect(hook1).toHaveBeenCalledWith({ id: 1 });
+      expect(hook2).toHaveBeenCalledWith({ id: 2 });
+      expect(hook3).toHaveBeenCalledWith({ id: 3 });
+
+      // Verify all were marked as completed
+      const events = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("idx_namespace_status_retry", (eb) => eb("namespace", "=", namespace)),
+          );
+          await executeRetrieve();
+          const [results] = await findUow.retrievalPhase;
+          return results;
+        });
+      });
+
+      const completed = events.filter((e) => e.status === "completed");
+      expect(completed).toHaveLength(3);
+    });
+
+    it("should continue processing other hooks when one fails", async () => {
+      const namespace = "test-partial-failure";
+      const hook1 = vi.fn();
+      const hook2 = vi.fn().mockRejectedValue(new Error("Hook 2 failed"));
+      const hook3 = vi.fn();
+      const hooks: HooksMap = {
+        onHook1: hook1,
+        onHook2: hook2,
+        onHook3: hook3,
+      };
+
+      await internalFragment.inContext(async function () {
+        await this.uow(async ({ forSchema, executeMutate }) => {
+          const uow = forSchema(internalSchema);
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook1",
+            payload: { id: 1 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-1",
+          });
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook2",
+            payload: { id: 2 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-2",
+          });
+          uow.create("fragno_hooks", {
+            namespace,
+            hookName: "onHook3",
+            payload: { id: 3 },
+            status: "pending",
+            attempts: 0,
+            maxAttempts: 5,
+            lastAttemptAt: null,
+            nextRetryAt: null,
+            error: null,
+            nonce: "nonce-3",
+          });
+          await executeMutate();
+        });
+      });
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 3 }),
+      });
+
+      expect(hook1).toHaveBeenCalledOnce();
+      expect(hook2).toHaveBeenCalledOnce();
+      expect(hook3).toHaveBeenCalledOnce();
+
+      // Verify hook1 and hook3 were completed, hook2 was marked for retry
+      const events = await internalFragment.inContext(async function () {
+        return await this.uow(async ({ forSchema, executeRetrieve }) => {
+          const uow = forSchema(internalSchema);
+          const findUow = uow.find("fragno_hooks", (b) =>
+            b.whereIndex("idx_namespace_status_retry", (eb) => eb("namespace", "=", namespace)),
+          );
+          await executeRetrieve();
+          const [results] = await findUow.retrievalPhase;
+          return results;
+        });
+      });
+
+      const completed = events.filter((e) => e.status === "completed");
+      const pending = events.filter((e) => e.status === "pending" && e.attempts === 1);
+
+      expect(completed).toHaveLength(2);
+      expect(pending).toHaveLength(1);
+      expect(pending[0]?.error).toBe("Hook 2 failed");
+    });
+
+    it("should do nothing when no pending events exist", async () => {
+      const namespace = "test-no-events";
+      const hookFn = vi.fn();
+      const hooks: HooksMap = {
+        onTest: hookFn,
+      };
+
+      await processHooks({
+        hooks,
+        namespace,
+        internalFragment,
+        defaultRetryPolicy: new ExponentialBackoffRetryPolicy({ maxRetries: 3 }),
+      });
+
+      expect(hookFn).not.toHaveBeenCalled();
+    });
+  });
+});