fix: calling shorthand method with string (#50)

maximilian-krauss · fent · commit 15eb60a6a723 · 2018-02-23T09:29:59.000-05:00
* randexp: Convert string to regexp * Removes additional local variable fixes #49
diff --git a/lib/randexp.js b/lib/randexp.js
@@ -236,6 +236,10 @@ module.exports = class RandExp {
    */
   static randexp(regexp, m) {
     var randexp;
+    if(typeof regexp === 'string') {
+      regexp = new RegExp(regexp, m);
+    }
+
     if (regexp._randexp === undefined) {
       randexp = new RandExp(regexp, m);
       regexp._randexp = randexp;
diff --git a/test/main-test.js b/test/main-test.js
@@ -53,6 +53,13 @@ describe('Call with a string', () => {
   });
 });
 
+describe('Call shorthand randexp method with a string', () => {
+  it('Returns a correctly generated string', () => {
+    var r = randexp('\\d{4}');
+    assert.equal(r.length, 4);
+  });
+});
+
 describe('Call without a string or regular expression', () => {
   it('Throws an error', () => {
     assert.throws(() => {
