feat: random int

ItMaga · Nov 29, 2023 · 25aecdb · 25aecdb
1 parent d5975e4
commit 25aecdb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/generators/NumberGenerator.ts b/lib/generators/NumberGenerator.ts
@@ -3,14 +3,14 @@ import type BaseGenerator from './BaseGenerator';
 
 export default class NumberGenerator<T extends z.ZodNumber> implements BaseGenerator<T> {
   public generate(schema: T) {
-    let number: z.infer<T> = 1;
+    let number: z.infer<T> = this.randomInt();
 
     if (schema._def.checks && schema._def.checks.length > 0) {
       for (const check of schema._def.checks) {
         switch (check.kind) {
           case 'int':
           case 'finite':
-            number = 1;
+            number = this.randomInt();
             break;
           case 'min':
             number = check.value + 1;
@@ -27,4 +27,8 @@ export default class NumberGenerator<T extends z.ZodNumber> implements BaseGener
 
     return number;
   }
+
+  private randomInt() {
+    return Math.floor(Math.random() * 1000);
+  }
 }