fixed linesOfCode extension

Author: EduardTrutsyk

extensions/assert-linesOfCode.js

@@ -1,17 +1,19 @@
 import assert from 'assert';
 
+const exclude = ['', ');'];
+
 function linesOfCode(fn, expected, recommended) {
-  const actualLinesOfCode = fn.toString().split('\n').filter(s => s.trim()).length - 2;
+  const actualLinesOfCode = fn.toString().split('\n').filter(s => !exclude.includes(s.trim())).length - 2;
   const isValid = actualLinesOfCode <= expected;
-  if (actualLinesOfCode <= expected && actualLinesOfCode > recommended) {
-    console.log(`Recommended lines of code for ${fn.name} is ${recommended}`); 
-  }
-  const message = `
+
+  const messages = [`
         "${fn.name}" should have maximum ${expected} line(s) of code but it ${actualLinesOfCode}.
-        Recommended lines of code for ${fn.name} is ${recommended}
-  `;
+  `];
+  if (recommended && !isValid && actualLinesOfCode > recommended) {
+    messages.push(`Recommended lines of code for ${fn.name} is ${recommended}`);
+  }
 
-  assert.ok(isValid, message);
+  assert.ok(isValid, messages.join('\n'));
 }
 
 assert.linesOfCode = linesOfCode;