Merge pull request #188 from APshenkin/master

adamgruber · Sep 18, 2017 · 6265395 · 6265395
2 parents 9a022a4 + 95f3221
commit 6265395
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/utils.js b/src/utils.js
@@ -67,7 +67,7 @@ function removeAllPropsFromObjExcept(obj, propsToKeep) {
 function cleanCode(str) {
   str = str
     .replace(/\r\n?|[\n\u2028\u2029]/g, '\n').replace(/^\uFEFF/, '')
-    .replace(/^function\s*\(.*\)\s*{|\(.*\)\s*=>\s*{?/, '')
+    .replace(/^function\*?\s*\(.*\)\s*{|\(.*\)\s*=>\s*{?/, '')
     .replace(/\s*\}$/, '');
 
   const spaces = str.match(/^\n?( *)/)[1].length;

diff --git a/test/utils.test.js b/test/utils.test.js
@@ -120,6 +120,13 @@ describe('Mochawesome Utils', () => {
       cleanCode(fnStr).should.equal(expected);
     });
 
+    it('should clean generator function syntax', () => {
+      fnStr = `function* () {
+        return true;
+      }`;
+      cleanCode(fnStr).should.equal(expected);
+    });
+
     it('should clean non-standard arrow function syntax', () => {
       fnStr = `() => 
       {