@fragno-dev/db 0.1.12 → 0.1.14

package/src/adapters/kysely/kysely-query.ts

@@ -14,6 +14,7 @@ import type { CompiledQuery, Kysely } from "kysely";
 import type { TableNameMapper } from "./kysely-shared";
 import type { ConnectionPool } from "../../shared/connection-pool";
 import type { SQLProvider } from "../../shared/providers";
+import { createCursorFromRecord, Cursor, type CursorResult } from "../../query/cursor";
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 type KyselyAny = Kysely<any>;
@@ -76,6 +77,8 @@ class UpdateManySpecialBuilder<TTable extends AnyTable> {
  * @param pool - Connection pool for acquiring database connections
  * @param provider - SQL provider (postgresql, mysql, sqlite, etc.)
  * @param mapper - Optional table name mapper for namespace prefixing
+ * @param uowConfig - Optional UOW configuration
+ * @param schemaNamespaceMap - Optional WeakMap for schema-to-namespace lookups
  * @returns An AbstractQuery instance for performing database operations
  *
  * @example
@@ -95,9 +98,10 @@ export function fromKysely<T extends AnySchema>(
   provider: SQLProvider,
   mapper?: TableNameMapper,
   uowConfig?: KyselyUOWConfig,
+  schemaNamespaceMap?: WeakMap<AnySchema, string>,
 ): AbstractQuery<T, KyselyUOWConfig> {
   function createUOW(opts: { name?: string; config?: KyselyUOWConfig }) {
-    const uowCompiler = createKyselyUOWCompiler(schema, pool, provider, mapper);
+    const uowCompiler = createKyselyUOWCompiler(pool, provider, mapper);
 
     const executor: UOWExecutor<CompiledQuery, unknown> = {
       async executeRetrievalPhase(retrievalBatch: CompiledQuery[]) {
@@ -132,7 +136,7 @@ export function fromKysely<T extends AnySchema>(
     };
 
     // Create a decoder function to transform raw results into application format
-    const decoder: UOWDecoder<T> = (rawResults, ops) => {
+    const decoder: UOWDecoder<unknown> = (rawResults, ops) => {
       if (rawResults.length !== ops.length) {
         throw new Error("rawResults and ops must have the same length");
       }
@@ -159,26 +163,72 @@ export function fromKysely<T extends AnySchema>(
 
         // Each result is an array of rows - decode each row
         const rowArray = rows as Record<string, unknown>[];
-        return rowArray.map((row) => decodeResult(row, op.table, provider));
+        const decodedRows = rowArray.map((row) => decodeResult(row, op.table, provider));
+
+        // If cursor generation is requested, wrap in CursorResult
+        if (op.withCursor) {
+          let cursor: Cursor | undefined;
+
+          // Generate cursor from last item if results exist
+          if (decodedRows.length > 0 && op.options.orderByIndex && op.options.pageSize) {
+            const lastItem = decodedRows[decodedRows.length - 1];
+            const indexName = op.options.orderByIndex.indexName;
+
+            // Get index columns
+            let indexColumns;
+            if (indexName === "_primary") {
+              indexColumns = [op.table.getIdColumn()];
+            } else {
+              const index = op.table.indexes[indexName];
+              if (index) {
+                indexColumns = index.columns;
+              }
+            }
+
+            if (indexColumns && lastItem) {
+              cursor = createCursorFromRecord(lastItem as Record<string, unknown>, indexColumns, {
+                indexName: op.options.orderByIndex.indexName,
+                orderDirection: op.options.orderByIndex.direction,
+                pageSize: op.options.pageSize,
+              });
+            }
+          }
+
+          const result: CursorResult<unknown> = {
+            items: decodedRows,
+            cursor,
+          };
+          return result;
+        }
+
+        return decodedRows;
       });
     };
 
     const { onQuery, ...restUowConfig } = opts.config ?? {};
 
-    return new UnitOfWork(schema, uowCompiler, executor, decoder, opts.name, {
-      ...restUowConfig,
-      onQuery: (query) => {
-        // CompiledMutation has { query: CompiledQuery, expectedAffectedRows: number | null }
-        // CompiledQuery has { query: QueryAST, sql: string, parameters: unknown[] }
-        // Check for expectedAffectedRows to distinguish CompiledMutation from CompiledQuery
-        const actualQuery =
-          query && typeof query === "object" && "expectedAffectedRows" in query
-            ? (query as CompiledMutation<CompiledQuery>).query
-            : (query as CompiledQuery);
-
-        opts.config?.onQuery?.(actualQuery);
+    return new UnitOfWork(
+      schema,
+      uowCompiler,
+      executor,
+      decoder,
+      opts.name,
+      {
+        ...restUowConfig,
+        onQuery: (query) => {
+          // CompiledMutation has { query: CompiledQuery, expectedAffectedRows: number | null }
+          // CompiledQuery has { query: QueryAST, sql: string, parameters: unknown[] }
+          // Check for expectedAffectedRows to distinguish CompiledMutation from CompiledQuery
+          const actualQuery =
+            query && typeof query === "object" && "expectedAffectedRows" in query
+              ? (query as CompiledMutation<CompiledQuery>).query
+              : (query as CompiledQuery);
+
+          opts.config?.onQuery?.(actualQuery);
+        },
       },
-    });
+      schemaNamespaceMap,
+    );
   }
 
   return {
@@ -193,6 +243,16 @@ export function fromKysely<T extends AnySchema>(
       return result ?? [];
     },
 
+    async findWithCursor(tableName, builderFn) {
+      // Safe: builderFn returns a FindBuilder, which matches UnitOfWork signature
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const uow = createUOW({ config: uowConfig }).findWithCursor(tableName, builderFn as any);
+      // executeRetrieve returns an array of results (one per find operation)
+      // Since we only have one findWithCursor, unwrap the first result
+      const [result] = await uow.executeRetrieve();
+      return result as CursorResult<unknown>;
+    },
+
     async findFirst(tableName, builderFn) {
       const uow = createUOW({ config: uowConfig });
       if (builderFn) {

package/src/adapters/kysely/kysely-uow-compiler.test.ts

@@ -5,6 +5,7 @@ import { UnitOfWork, type UOWDecoder } from "../../query/unit-of-work";
 import { createKyselyUOWCompiler } from "./kysely-uow-compiler";
 import type { ConnectionPool } from "../../shared/connection-pool";
 import { createKyselyConnectionPool } from "./kysely-connection-pool";
+import { Cursor } from "../../query/cursor";
 
 describe("kysely-uow-compiler", () => {
   const testSchema = schema((s) => {
@@ -113,12 +114,12 @@ describe("kysely-uow-compiler", () => {
 
   // Helper to create UnitOfWork for testing
   function createTestUOW(name?: string) {
-    const mockCompiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+    const mockCompiler = createKyselyUOWCompiler(pool, "postgresql");
     const mockExecutor = {
       executeRetrievalPhase: async () => [],
       executeMutationPhase: async () => ({ success: true, createdInternalIds: [] }),
     };
-    const mockDecoder: UOWDecoder<typeof testSchema> = (rawResults, operations) => {
+    const mockDecoder: UOWDecoder = (rawResults, operations) => {
       if (rawResults.length !== operations.length) {
         throw new Error("rawResults and ops must have the same length");
       }
@@ -135,7 +136,7 @@ describe("kysely-uow-compiler", () => {
         b.whereIndex("idx_email", (eb) => eb("email", "=", "test@example.com")),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -151,7 +152,7 @@ describe("kysely-uow-compiler", () => {
         b.whereIndex("idx_name", (eb) => eb("name", "=", "Alice")).select(["id", "name"]),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -165,7 +166,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("primary").pageSize(10));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -179,7 +180,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("primary").orderByIndex("primary", "desc"));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -192,7 +193,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("idx_name").orderByIndex("idx_name", "desc"));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -208,7 +209,7 @@ describe("kysely-uow-compiler", () => {
       );
       uow.find("posts", (b) => b.whereIndex("idx_title", (eb) => eb("title", "contains", "test")));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(2);
@@ -227,7 +228,7 @@ describe("kysely-uow-compiler", () => {
         return b;
       });
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -240,7 +241,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("idx_age", (eb) => eb("age", ">", 25)).selectCount());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -252,12 +253,17 @@ describe("kysely-uow-compiler", () => {
 
     it("should compile find operation with cursor pagination using after", () => {
       const uow = createTestUOW();
-      const cursor = "eyJpbmRleFZhbHVlcyI6eyJuYW1lIjoiQWxpY2UifSwiZGlyZWN0aW9uIjoiZm9yd2FyZCJ9"; // {"indexValues":{"name":"Alice"},"direction":"forward"}
+      const cursor = new Cursor({
+        indexName: "idx_name",
+        orderDirection: "asc",
+        pageSize: 10,
+        indexValues: { name: "Alice" },
+      });
       uow.find("users", (b) =>
         b.whereIndex("idx_name").orderByIndex("idx_name", "asc").after(cursor).pageSize(10),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -269,12 +275,17 @@ describe("kysely-uow-compiler", () => {
 
     it("should compile find operation with cursor pagination using before", () => {
       const uow = createTestUOW();
-      const cursor = "eyJpbmRleFZhbHVlcyI6eyJuYW1lIjoiQm9iIn0sImRpcmVjdGlvbiI6ImJhY2t3YXJkIn0="; // {"indexValues":{"name":"Bob"},"direction":"backward"}
+      const cursor = new Cursor({
+        indexName: "idx_name",
+        orderDirection: "desc",
+        pageSize: 10,
+        indexValues: { name: "Bob" },
+      });
       uow.find("users", (b) =>
         b.whereIndex("idx_name").orderByIndex("idx_name", "desc").before(cursor).pageSize(10),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -286,7 +297,12 @@ describe("kysely-uow-compiler", () => {
 
     it("should compile find operation with cursor pagination and additional where conditions", () => {
       const uow = createTestUOW();
-      const cursor = "eyJpbmRleFZhbHVlcyI6eyJuYW1lIjoiQWxpY2UifSwiZGlyZWN0aW9uIjoiZm9yd2FyZCJ9";
+      const cursor = new Cursor({
+        indexName: "idx_name",
+        orderDirection: "asc",
+        pageSize: 5,
+        indexValues: { name: "Alice" },
+      });
       uow.find("users", (b) =>
         b
           .whereIndex("idx_name", (eb) => eb("name", "starts with", "John"))
@@ -295,7 +311,7 @@ describe("kysely-uow-compiler", () => {
           .pageSize(5),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -315,7 +331,7 @@ describe("kysely-uow-compiler", () => {
         age: 30,
       });
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -341,7 +357,7 @@ describe("kysely-uow-compiler", () => {
         }),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -357,7 +373,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user123", 5);
       uow.update("users", userId, (b) => b.set({ age: 18 }).check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -373,7 +389,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user123", 0);
       uow.delete("users", userId);
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -389,7 +405,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user123", 3);
       uow.delete("users", userId, (b) => b.check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -409,7 +425,7 @@ describe("kysely-uow-compiler", () => {
         }),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -424,7 +440,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.delete("users", "user123");
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -464,7 +480,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user456", 0);
       uow.delete("posts", userId);
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [createBatch, updateBatch, deleteBatch] = compiled.mutationBatch;
 
@@ -503,7 +519,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user123", 3);
       uow.update("users", userId, (b) => b.set({ age: 31 }).check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.name).toBe("update-user-balance");
@@ -535,7 +551,7 @@ describe("kysely-uow-compiler", () => {
         ),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -549,7 +565,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("primary", () => false));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       // When condition is false, the operation should return null and not be added to batch
@@ -560,7 +576,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users", (b) => b.whereIndex("primary", () => true));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -577,7 +593,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user123", 5);
       uow.update("users", userId, (b) => b.set({ age: 31 }).check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -594,7 +610,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user456", 3);
       uow.delete("users", userId, (b) => b.check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -614,7 +630,7 @@ describe("kysely-uow-compiler", () => {
       uow.update("users", userId, (b) => b.set({ age: 30 }).check());
       uow.update("posts", postId, (b) => b.set({ viewCount: 100 }).check());
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [userBatch, postBatch] = compiled.mutationBatch;
 
@@ -641,7 +657,7 @@ describe("kysely-uow-compiler", () => {
       const userId = FragnoId.fromExternal("user1", 0);
       uow.update("users", userId, (b) => b.set({ age: 25 }));
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
       const [batch] = compiled.mutationBatch;
       assert(batch);
@@ -660,7 +676,7 @@ describe("kysely-uow-compiler", () => {
         b.whereIndex("primary").join((jb) => jb.author((ab) => ab.select(["name", "email"]))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -684,7 +700,7 @@ describe("kysely-uow-compiler", () => {
           ),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -703,7 +719,7 @@ describe("kysely-uow-compiler", () => {
           .join((jb) => jb.author((ab) => ab.select(["name"]).orderByIndex("idx_name", "desc"))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -719,7 +735,7 @@ describe("kysely-uow-compiler", () => {
         b.whereIndex("primary").join((jb) => jb.author((ab) => ab.select(["name"]).pageSize(5))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -750,7 +766,7 @@ describe("kysely-uow-compiler", () => {
         ),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -769,7 +785,7 @@ describe("kysely-uow-compiler", () => {
           .join((jb) => jb.post((pb) => pb.select(["title"])).author((ab) => ab.select(["name"]))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -787,7 +803,7 @@ describe("kysely-uow-compiler", () => {
         b.whereIndex("primary").join((jb) => jb.inviter((ib) => ib.select(["name", "email"]))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -817,7 +833,7 @@ describe("kysely-uow-compiler", () => {
           ),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -840,7 +856,7 @@ describe("kysely-uow-compiler", () => {
           .join((jb) => jb.post((pb) => pb.select(["title"])).tag((tb) => tb.select(["name"]))),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -865,7 +881,7 @@ describe("kysely-uow-compiler", () => {
           ),
       );
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -881,7 +897,7 @@ describe("kysely-uow-compiler", () => {
     it("should handle UOW with no operations", () => {
       const uow = createTestUOW();
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(0);
@@ -892,7 +908,7 @@ describe("kysely-uow-compiler", () => {
       const uow = createTestUOW();
       uow.find("users");
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(1);
@@ -906,7 +922,7 @@ describe("kysely-uow-compiler", () => {
         email: "test@example.com",
       });
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       expect(compiled.retrievalBatch).toHaveLength(0);
@@ -933,7 +949,7 @@ describe("kysely-uow-compiler", () => {
         content: "This is my first post",
       });
 
-      const compiler = createKyselyUOWCompiler(testSchema, pool, "postgresql");
+      const compiler = createKyselyUOWCompiler(pool, "postgresql");
       const compiled = uow.compile(compiler);
 
       // Should have no retrieval operations