refactor: decorators

TomerAberbach · Dec 7, 2024 · c43e975 · c43e975
1 parent ffc6c3a
commit c43e975
Show file tree

Hide file tree

Showing 17 changed files with 630 additions and 130 deletions.
diff --git a/src/arbitrary.ts b/src/arbitrary.ts
@@ -16,6 +16,7 @@ export type Arbitrary = BaseArbitrary &
     | IntegerArbitrary
     | BigIntegerArbitrary
     | FloatArbitrary
+    | DoubleArbitrary
     | StringArbitrary
   )
 
@@ -71,6 +72,12 @@ export type FloatArbitrary = {
   max?: number
 }
 
+export type DoubleArbitrary = {
+  type: `double`
+  min?: number
+  max?: number
+}
+
 export type StringArbitrary = {
   type: `string`
   minLength?: number

diff --git a/src/components.ts b/src/components.ts
@@ -10,6 +10,7 @@ import type {
   ArrayArbitrary,
   BigIntegerArbitrary,
   DictionaryArbitrary,
+  DoubleArbitrary,
   EnumArbitrary,
   FloatArbitrary,
   IntegerArbitrary,
@@ -145,6 +146,8 @@ const ArbitraryDefinition = ({
       return BigIntegerArbitrary({ arbitrary })
     case `float`:
       return FloatArbitrary({ arbitrary })
+    case `double`:
+      return DoubleArbitrary({ arbitrary })
     case `string`:
       return StringArbitrary({ arbitrary })
   }
@@ -243,6 +246,18 @@ const FloatArbitrary = ({ arbitrary }: { arbitrary: FloatArbitrary }): Child =>
     ]),
   })})`
 
+const DoubleArbitrary = ({
+  arbitrary,
+}: {
+  arbitrary: DoubleArbitrary
+}): Child =>
+  code`fc.double(${Options({
+    properties: new Map([
+      [`min`, arbitrary.min],
+      [`max`, arbitrary.max],
+    ]),
+  })})`
+
 const StringArbitrary = ({
   arbitrary,
 }: {