@fragno-dev/db 0.1.13 → 0.1.15

package/src/query/unit-of-work-types.test.ts

@@ -1,6 +1,6 @@
 import { describe, expectTypeOf, it } from "vitest";
 import { column, idColumn, referenceColumn, schema } from "../schema/create";
-import { createUnitOfWork, JoinFindBuilder } from "./unit-of-work";
+import { JoinFindBuilder, UnitOfWork } from "./unit-of-work";
 import type { FragnoId, FragnoReference } from "../schema/create";
 
 type Prettify<T> = {
@@ -95,7 +95,7 @@ describe("UnitOfWork type tests", () => {
       executeMutationPhase: async () => ({ success: true, createdInternalIds: [] }),
     };
     const mockDecoder = () => [];
-    return createUnitOfWork(testSchema, mockCompiler, mockExecutor, mockDecoder);
+    return new UnitOfWork(mockCompiler, mockExecutor, mockDecoder).forSchema(testSchema);
   }
 
   it("should type find without joins correctly", async () => {
@@ -114,6 +114,19 @@ describe("UnitOfWork type tests", () => {
     }>();
   });
 
+  it("should type find with select clause correctly", async () => {
+    const uow = createTestUOW();
+
+    const uow1 = uow.find("users", (b) => b.whereIndex("primary").select(["id", "name"]));
+    const [_userResult] = await uow1.executeRetrieve();
+    type UserResult = RecursivePrettify<(typeof _userResult)[number]>;
+
+    expectTypeOf<UserResult>().toEqualTypeOf<{
+      id: FragnoId;
+      name: string;
+    }>();
+  });
+
   it("should type find with joins correctly", async () => {
     const uow = createTestUOW();