wip: next ver

README.md

@@ -21,6 +21,7 @@ Prisma generator for model factories.
   - [has-many / has-one relation](#has-many--has-one-relation)
   - [Custom scalar field generation](#custom-scalar-field-generation)
   - [Traits](#traits)
+  - [Callbacks](#callbacks)
   - [Field value precedence](#field-value-precedence)
   - [More examples](#more-examples)
 - [Generator configuration](#generator-configuration)
@@ -375,6 +376,64 @@ Multiple traits are also available:
 await UserFactory.use("someTrait", "anotherTrait").create();
 ```
 
+### Callbacks
+
+You can set callback function before or after factory execution.
+
+```ts
+const UserFactory = defineUserFactory({
+  onAfterCreate: async user => {
+    await PostFactory.craete({
+      author: { connect: uesr },
+    });
+  },
+});
+
+await UserFactory.create();
+```
+
+Callback functions are also available within trait definition.
+
+```ts
+const UserFactory = defineUserFactory({
+  traits: {
+    withComment: {
+      onAfterCreate: async user => {
+        await PostFactory.craete({
+          author: { connect: uesr },
+        });
+      },
+    },
+  },
+});
+
+await UserFactory.create();
+await UserFactory.use("withComment").create();
+```
+
+Note: The above code is to explain the callback. If you want to create association, first consider to use `defaultData` and `trait.data` option as in [has-many / has-one relation](#has-many--has-one-relation).
+
+The following three types are available as callback function:
+
+```ts
+const UserFactory = defineUserFactory({
+  onAfterBuild: async createInput => {
+    // do something
+  },
+  onBeforeCreate: async createInput => {
+    // do something
+  },
+  onAfterCreate: async createdData => {
+    // do something
+  },
+});
+```
+
+And here, the parameter types are:
+
+- `createInput` is assignable to model create function parameter (e.g. `Prsima.UserCreateInput`).
+- `createdData` is resolved object by model create function (e.g. `User` model type)
+
 ### Field value precedence
 
 Each field is determined in the following priority order(lower numbers have higher priority):

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

@@ -2,12 +2,28 @@ import {
   defineUserFactory,
   definePostFactory,
   defineCommentFactory,
-  CommentFactoryInterface,
+  type PostFactoryInterface,
+  type CommentFactoryInterface,
 } from "./__generated__/fabbrica";
 
 const prisma = jestPrisma.client;
 
-export const UserFactory = defineUserFactory();
+export const UserFactory = defineUserFactory({
+  traits: {
+    withSelfCommentedPost: {
+      onAfterCreate: async user => {
+        await getPostFactory().create({
+          author: {
+            connect: user,
+          },
+          comments: {
+            create: [await getCommentFactory().build({ author: { connect: user } })],
+          },
+        });
+      },
+    },
+  },
+});
 
 export const PostFactory = definePostFactory({
   defaultData: {
@@ -43,6 +59,10 @@ export const CommentFactory = defineCommentFactory({
   },
 });
 
+function getPostFactory(): PostFactoryInterface {
+  return PostFactory;
+}
+
 function getCommentFactory(): CommentFactoryInterface {
   return CommentFactory;
 }
@@ -77,5 +97,12 @@ describe("factories", () => {
         await expect(prisma.comment.count({ where: { postId: post2.id } })).resolves.toBe(1);
       });
     });
+
+    describe("trait and callback", () => {
+      test("Execute other factory in callback", async () => {
+        const { id: userId } = await UserFactory.use("withSelfCommentedPost").create();
+        await expect(prisma.comment.count({ where: { authorId: userId } })).resolves.toBe(1);
+      });
+    });
   });
 });