feat: add primitive lazy support

README.md

@@ -95,7 +95,7 @@ export const api = {
 ## TODO
 
 - ZodNever
-- ZodLazy
+- ZodLazy (recursive types)
 
 ## License
 

lib/MockGenerator.ts

@@ -31,6 +31,7 @@ import RecordGenerator from './generators/RecordGenerator';
 import PipelineGenerator from './generators/PipelineGenerator';
 import BrandedGenerator from './generators/BrandedGenerator';
 import FunctionGenerator from './generators/FunctionGenerator';
+import LazyGenerator from './generators/LazyGenerator';
 
 export default class MockGenerator<T extends z.ZodTypeAny> {
   private generator: BaseGenerator<TypeOf<T>>;
@@ -73,6 +74,7 @@ export default class MockGenerator<T extends z.ZodTypeAny> {
       ZodPipeline: PipelineGenerator,
       ZodBranded: BrandedGenerator,
       ZodFunction: FunctionGenerator,
+      ZodLazy: LazyGenerator,
     };
 
     if (this.schema._def.typeName in generatorMap) {

lib/generators/LazyGenerator.ts

@@ -0,0 +1,10 @@
+import type { z } from 'zod';
+import MockGenerator from '../MockGenerator';
+import type BaseGenerator from './BaseGenerator';
+
+export default class LazyGenerator<T extends z.ZodLazy<z.ZodTypeAny>> implements BaseGenerator<T> {
+  public generate(schema: T): z.infer<typeof schema> {
+    const mockGenerator = new MockGenerator(schema._def.getter());
+    return mockGenerator.generate();
+  }
+}

tests/lazy.test.ts

@@ -0,0 +1,11 @@
+import { describe, test } from 'vitest';
+import { z } from 'zod';
+import { expect } from './utils/expect';
+
+describe('Lazy', () => {
+  test('primitive', () => {
+    const schema = z.lazy(() => z.string());
+
+    expect(schema);
+  });
+});