fix(sampler): introduce max-length support for string pattern format …

…sample closes #208
NeuraLegion · Sep 14, 2023 · 34bd1ef · 34bd1ef
1 parent 5d15e91
commit 34bd1ef
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 2 deletions.
diff --git a/packages/openapi-sampler/src/samplers/StringSampler.ts b/packages/openapi-sampler/src/samplers/StringSampler.ts
@@ -33,7 +33,7 @@ export class StringSampler implements Sampler {
       _min: number,
       _max: number,
       { pattern }: { pattern: string | RegExp }
-    ) => this.patternSample(pattern),
+    ) => this.patternSample(pattern, _max),
     'default': (min: number, max: number) =>
       this.adjustLength('lorem', min, max)
   };
@@ -47,8 +47,10 @@ export class StringSampler implements Sampler {
     return this.checkLength(sampler(min || 0, max, schema), format, min, max);
   }
 
-  private patternSample(pattern: string | RegExp): string {
+  private patternSample(pattern: string | RegExp, max: number): string {
     const randExp = new RandExp(pattern);
+
+    randExp.max = max != null ? max : randExp.max;
     randExp.randInt = (a, b) => Math.floor((a + b) / 2);
 
     return randExp.gen();

diff --git a/packages/openapi-sampler/tests/string.spec.ts b/packages/openapi-sampler/tests/string.spec.ts
@@ -24,6 +24,33 @@ describe('StringSampler', () => {
       },
       expected: '44'
     },
+    {
+      input: {
+        type: 'string',
+        format: 'pattern',
+        pattern: '^[a-z\\u05D0-\\u05EAA-Z0-9-/\\\\.\\u05F4,\'":&#;+()\\s]*$',
+        maxLength: 50
+      },
+      expected: 'MMMMMMMMMMMMMMMMMMMMMMMMM'
+    },
+    {
+      input: {
+        type: 'string',
+        format: 'pattern',
+        pattern: '^[A-Z0-9]*$',
+        maxLength: 10
+      },
+      expected: 'RRRRR'
+    },
+    {
+      input: {
+        type: 'string',
+        format: 'pattern',
+        pattern: '^[A-Z0-9]*$',
+        maxLength: 0
+      },
+      expected: ''
+    },
     {
       input: {
         type: 'string',