feat: brand support

README.md

@@ -37,5 +37,4 @@ createMock(schema); // { name: string, age: number }
 
 - ZodNever
 - ZodFunction
-- ZodLazy
-- ZodBranded
+- ZodLazy

lib/MockGenerator.ts

@@ -29,6 +29,7 @@ import MapGenerator from './generators/MapGenerator';
 import SetGenerator from './generators/SetGenerator';
 import RecordGenerator from './generators/RecordGenerator';
 import PipelineGenerator from './generators/PipelineGenerator';
+import BrandedGenerator from './generators/BrandedGenerator';
 
 export default class MockGenerator<T extends z.ZodTypeAny> {
   private generator: BaseGenerator<TypeOf<T>>;
@@ -69,6 +70,7 @@ export default class MockGenerator<T extends z.ZodTypeAny> {
       ZodSet: SetGenerator,
       ZodRecord: RecordGenerator,
       ZodPipeline: PipelineGenerator,
+      ZodBranded: BrandedGenerator,
     };
     if (this.schema._def.typeName in generatorMap) {
       this.generator = new generatorMap[this.schema._def.typeName as ZodFirstPartyTypeKind]();

lib/generators/BrandedGenerator.ts

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

tests/branded.test.ts

@@ -0,0 +1,18 @@
+import { describe, test } from 'vitest';
+import { z } from 'zod';
+import { expect } from './utils/expect';
+
+describe('Branded', () => {
+  test('string', () => {
+    const schema = z.string().brand('myBrand');
+
+    expect(schema);
+  });
+  test('object', () => {
+    const schema = z.object({
+      foo: z.string().brand('brandKey'),
+    }).brand('brandObject');
+
+    expect(schema);
+  });
+});