feat: Screening disturbing fields before create

examples/example-prj/src/sample.test.ts

@@ -23,6 +23,16 @@ describe("factories", () => {
       const user = await prisma.user.findUnique({ where: { id: "user001" } });
       expect(user).toEqual({ id: "user001", name: "Quramy" });
     });
+
+    it("creates record with children relation", async () => {
+      await UserFactory.create({
+        posts: {
+          create: [await PostFactory.buildCreateInput()],
+        },
+      });
+      const created = await prisma.user.findFirst({ include: { posts: true } });
+      expect(created?.posts.length).toBe(1);
+    });
   });
 
   describe("PostFactory", () => {

packages/prisma-fabbrica/src/relations/index.ts

@@ -0,0 +1 @@
+export * from "./screen";

packages/prisma-fabbrica/src/templates/__snapshots__/getSourceFile.test.ts.snap

@@ -5,12 +5,17 @@ exports[`getSourceFile generates TypeScript AST 1`] = `
 import { Prisma } from "@prisma/client";
 import type { PrismaClient } from "@prisma/client";
 import { getClient } from "@quramy/prisma-fabbrica/lib/clientHolder";
+import { ModelWithFields, createScreener } from "@quramy/prisma-fabbrica/lib/relations";
 import scalarFieldValueGenerator from "@quramy/prisma-fabbrica/lib/scalar/gen";
 import { Resolver, normalizeResolver, getSequenceCounter } from "@quramy/prisma-fabbrica/lib/helpers";
 export { initialize, resetSequence } from "@quramy/prisma-fabbrica";
 type BuildDataOptions = {
     readonly seq: number;
 };
+const modelFieldDefinitions: ModelWithFields[] = [{
+        name: "User",
+        fields: []
+    }];
 type UserScalarOrEnumFields = {
     id: number;
     name: string;
@@ -33,6 +38,7 @@ function autoGenerateUserScalarsOrEnums({ seq }: {
 function defineUserFactoryInternal({ defaultData: defaultDataResolver }: UserFactoryDefineOptions) {
     const seqKey = {};
     const getSeq = () => getSequenceCounter(seqKey);
+    const screen = createScreener("User", modelFieldDefinitions);
     const buildCreateInput = async (inputData: Partial<Prisma.UserCreateInput> = {}) => {
         const seq = getSeq();
         const requiredScalarData = autoGenerateUserScalarsOrEnums({ seq });
@@ -46,7 +52,7 @@ function defineUserFactoryInternal({ defaultData: defaultDataResolver }: UserFac
         id: inputData.id
     });
     const create = async (inputData: Partial<Prisma.UserCreateInput> = {}) => {
-        const data = await buildCreateInput(inputData);
+        const data = await buildCreateInput(inputData).then(screen);
         return await getClient<PrismaClient>().user.create({ data });
     };
     const createForConnect = (inputData: Partial<Prisma.UserCreateInput> = {}) => create(inputData).then(pickForConnect);

packages/prisma-fabbrica/src/templates/index.ts

@@ -1,7 +1,8 @@
 import { DMMF } from "@prisma/generator-helper";
 import ts from "typescript";
 import { template } from "talt";
-import { camelize, ast, byName } from "../helpers";
+import { camelize, ast, byName, createJSONLiteral } from "../helpers";
+import { createFieldDefinitions } from "../relations";
 
 export function findPrsimaCreateInputTypeFromModelName(document: DMMF.Document, modelName: string) {
   const search = `${modelName}CreateInput`;
@@ -64,6 +65,7 @@ export const header = (prismaClientModuleSpecifier: string) =>
     import { Prisma } from ${() => ast.stringLiteral(prismaClientModuleSpecifier)};
     import type { PrismaClient } from ${() => ast.stringLiteral(prismaClientModuleSpecifier)};
     import { getClient } from "@quramy/prisma-fabbrica/lib/clientHolder";
+    import { ModelWithFields, createScreener } from "@quramy/prisma-fabbrica/lib/relations";
     import scalarFieldValueGenerator from "@quramy/prisma-fabbrica/lib/scalar/gen";
     import { Resolver, normalizeResolver, getSequenceCounter } from "@quramy/prisma-fabbrica/lib/helpers";
     export { initialize, resetSequence } from "@quramy/prisma-fabbrica";
@@ -78,6 +80,11 @@ export const importStatement = (specifier: string, prismaClientModuleSpecifier:
     import { ${() => ast.identifier(specifier)} } from ${() => ast.stringLiteral(prismaClientModuleSpecifier)};
   `();
 
+export const modelFieldDefinitions = (models: DMMF.Model[]) =>
+  template.statement`
+  const modelFieldDefinitions: ModelWithFields[] = ${() => createJSONLiteral(createFieldDefinitions(models))}
+`();
+
 export const scalarFieldType = (
   model: DMMF.Model,
   fieldName: string,
@@ -263,6 +270,7 @@ export const defineModelFactoryInernal = (model: DMMF.Model, inputType: DMMF.Inp
 
       const seqKey = {};
       const getSeq = () => getSequenceCounter(seqKey);
+      const screen = createScreener(${() => ast.stringLiteral(model.name)}, modelFieldDefinitions);
 
       const buildCreateInput = async (
         inputData: Partial<Prisma.MODEL_CREATE_INPUT> = {}
@@ -305,7 +313,7 @@ export const defineModelFactoryInernal = (model: DMMF.Model, inputType: DMMF.Inp
       const create = async (
         inputData: Partial<Prisma.MODEL_CREATE_INPUT> = {}
       ) => {
-        const data = await buildCreateInput(inputData);
+        const data = await buildCreateInput(inputData).then(screen);
         return await getClient<PrismaClient>().MODEL_KEY.create({ data });
       };
 
@@ -369,6 +377,7 @@ export function getSourceFile({
     ...modelNames.map(modelName => importStatement(modelName, prismaClientModuleSpecifier)),
     ...enums.map(enumName => importStatement(enumName, prismaClientModuleSpecifier)),
     ...header(prismaClientModuleSpecifier).statements,
+    modelFieldDefinitions(document.datamodel.models),
     ...document.datamodel.models
       .map(model => ({ model, createInputType: findPrsimaCreateInputTypeFromModelName(document, model.name) }))
       .flatMap(({ model, createInputType }) => [