added tests for variables

Author: Code-Hex

tests/myzod.spec.ts

@@ -803,6 +803,38 @@ describe('myzod', () => {
         expect(result.content).toContain(wantContain);
       }
     });
+
+    it('with object arguments', async () => {
+      const schema = buildSchema(/* GraphQL */ `
+        type MyType {
+          foo(a: String, b: Int!, c: Boolean, d: Float!, e: Text): String
+        }
+        scalar Text
+      `);
+      const result = await plugin(
+        schema,
+        [],
+        {
+          schema: 'myzod',
+          withObjectType: true,
+          scalars: {
+            Text: 'string',
+          },
+        },
+        {}
+      );
+      const wantContain = dedent`
+      export function MyTypeFooArgsSchema(): myzod.Type<MyTypeFooArgs> {
+        return myzod.object({
+          a: myzod.string().optional().nullable(),
+          b: myzod.number(),
+          c: myzod.boolean().optional().nullable(),
+          d: myzod.number(),
+          e: myzod.string().optional().nullable()
+        })
+      }`;
+      expect(result.content).toContain(wantContain);
+    });
   });
 
   it('properly generates custom directive values', async () => {

tests/yup.spec.ts

@@ -717,6 +717,38 @@ describe('yup', () => {
         expect(result.content).toContain(wantContain);
       }
     });
+
+    it('with object arguments', async () => {
+      const schema = buildSchema(/* GraphQL */ `
+        type MyType {
+          foo(a: String, b: Int!, c: Boolean, d: Float!, e: Text): String
+        }
+        scalar Text
+      `);
+      const result = await plugin(
+        schema,
+        [],
+        {
+          schema: 'yup',
+          withObjectType: true,
+          scalars: {
+            Text: 'string',
+          },
+        },
+        {}
+      );
+      const wantContain = dedent`
+      export function MyTypeFooArgsSchema(): yup.ObjectSchema<MyTypeFooArgs> {
+        return yup.object({
+          a: yup.string().defined().nullable(),
+          b: yup.number().defined().nonNullable(),
+          c: yup.boolean().defined().nullable(),
+          d: yup.number().defined().nonNullable(),
+          e: yup.string().defined().nullable()
+        })
+      }`;
+      expect(result.content).toContain(wantContain);
+    });
   });
 
   it('properly generates custom directive values', async () => {

tests/zod.spec.ts

@@ -871,6 +871,38 @@ describe('zod', () => {
         expect(result.content).toContain(wantContain);
       }
     });
+
+    it('with object arguments', async () => {
+      const schema = buildSchema(/* GraphQL */ `
+        type MyType {
+          foo(a: String, b: Int!, c: Boolean, d: Float!, e: Text): String
+        }
+        scalar Text
+      `);
+      const result = await plugin(
+        schema,
+        [],
+        {
+          schema: 'zod',
+          withObjectType: true,
+          scalars: {
+            Text: 'string',
+          },
+        },
+        {}
+      );
+      const wantContain = dedent`
+      export function MyTypeFooArgsSchema(): z.ZodObject<Properties<MyTypeFooArgs>> {
+        return z.object({
+          a: z.string().nullish(),
+          b: z.number(),
+          c: z.boolean().nullish(),
+          d: z.number(),
+          e: z.string().nullish()
+        })
+      }`;
+      expect(result.content).toContain(wantContain);
+    });
   });
 
   it('properly generates custom directive values', async () => {