Check static template strings in valid-test-description and valid-sui…

…te-description
lo1tuma · Feb 18, 2020 · 6255546 · 6255546
1 parent 7eea93d
commit 6255546
Show file tree

Hide file tree

Showing 7 changed files with 90 additions and 20 deletions.
diff --git a/lib/rules/valid-suite-description.js b/lib/rules/valid-suite-description.js
@@ -1,11 +1,12 @@
 'use strict';
 
+const { getStringIfConstant } = require('eslint-utils');
+
 /**
  * @fileoverview Match suite descriptions to match a pre-configured regular expression
  * @author Alexander Afanasyev
  */
 
-const astUtils = require('../util/ast');
 const defaultSuiteNames = [ 'describe', 'context', 'suite' ];
 
 function inlineOptions(options) {
@@ -76,10 +77,11 @@ module.exports = {
 
         function hasValidSuiteDescription(mochaCallExpression) {
             const args = mochaCallExpression.arguments;
-            const description = args[0];
+            const descriptionArgument = args[0];
+            const description = getStringIfConstant(descriptionArgument, context.getScope());
 
-            if (astUtils.isStringLiteral(description)) {
-                return pattern.test(description.value);
+            if (description) {
+                return pattern.test(description);
             }
 
             return true;

diff --git a/lib/rules/valid-test-description.js b/lib/rules/valid-test-description.js
@@ -1,12 +1,12 @@
 'use strict';
 
+const { getStringIfConstant } = require('eslint-utils');
+
 /**
  * @fileoverview Match test descriptions to match a pre-configured regular expression
  * @author Alexander Afanasyev
  */
 
-const astUtils = require('../util/ast');
-
 const defaultTestNames = [ 'it', 'test', 'specify' ];
 
 function inlineOptions(options) {
@@ -77,9 +77,10 @@ module.exports = {
         function hasValidTestDescription(mochaCallExpression) {
             const args = mochaCallExpression.arguments;
             const testDescriptionArgument = args[0];
+            const description = getStringIfConstant(testDescriptionArgument, context.getScope());
 
-            if (astUtils.isStringLiteral(testDescriptionArgument)) {
-                return pattern.test(testDescriptionArgument.value);
+            if (description) {
+                return pattern.test(description);
             }
 
             return true;

diff --git a/lib/util/ast.js b/lib/util/ast.js
@@ -100,10 +100,6 @@ function isMochaFunctionCall(node, scope) {
     return isTestCase(node) || isDescribe(node) || isHookCall(node);
 }
 
-function isStringLiteral(node) {
-    return node && node.type === 'Literal' && typeof node.value === 'string';
-}
-
 function hasParentMochaFunctionCall(functionExpression) {
     return isTestCase(functionExpression.parent) || isHookCall(functionExpression.parent);
 }
@@ -130,7 +126,6 @@ module.exports = {
     isMochaFunctionCall,
     isHookCall,
     isSuiteConfigCall,
-    isStringLiteral,
     hasParentMochaFunctionCall,
     findReturnStatement,
     isReturnOfUndefined

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -22,6 +22,7 @@
         "changelog": "pr-log"
     },
     "dependencies": {
+        "eslint-utils": "^2.0.0",
         "ramda": "^0.26.1"
     },
     "devDependencies": {

diff --git a/test/rules/valid-suite-description.js b/test/rules/valid-suite-description.js
@@ -43,6 +43,16 @@ ruleTester.run('valid-suite-description', rules['valid-suite-description'], {
             parserOptions: { ecmaVersion: 2017 },
             options: [ '^Foo' ],
             code: 'describe(`Foo with template strings`, function () {});'
+        },
+        {
+            parserOptions: { ecmaVersion: 2019 },
+            options: [ '^Foo' ],
+            code: 'describe(anyTag`with template strings`, function () {});'
+        },
+        {
+            parserOptions: { ecmaVersion: 2019 },
+            options: [ '^Foo' ],
+            code: 'describe(`${dynamicVar} with template strings`, function () {});'
         }
 
     ],
@@ -89,6 +99,26 @@ ruleTester.run('valid-suite-description', rules['valid-suite-description'], {
             errors: [
                 { message: 'some error message' }
             ]
+        },
+        {
+            options: [ '^[A-Z]' ],
+            code: 'describe(`this is a test`, function () { });',
+            parserOptions: {
+                ecmaVersion: 2019
+            },
+            errors: [
+                { message: 'Invalid "describe()" description found.', line: 1, column: 1 }
+            ]
+        },
+        {
+            options: [ '^[A-Z]' ],
+            code: 'const foo = "this"; describe(`${foo} is a test`, function () { });',
+            parserOptions: {
+                ecmaVersion: 2019
+            },
+            errors: [
+                { message: 'Invalid "describe()" description found.', line: 1, column: 21 }
+            ]
         }
     ]
 });
diff --git a/test/rules/valid-test-description.js b/test/rules/valid-test-description.js
@@ -43,6 +43,14 @@ ruleTester.run('valid-test-description', rules['valid-test-description'], {
         {
             parserOptions: { ecmaVersion: 2017 },
             code: 'it(`should work with template strings`, function () {});'
+        },
+        {
+            parserOptions: { ecmaVersion: 2019 },
+            code: 'it(foo`work with template strings`, function () {});'
+        },
+        {
+            parserOptions: { ecmaVersion: 2019 },
+            code: 'it(`${foo} work with template strings`, function () {});'
         }
     ],
 
@@ -113,6 +121,24 @@ ruleTester.run('valid-test-description', rules['valid-test-description'], {
             errors: [
                 { message: 'Invalid "it()" description found.' }
             ]
+        },
+        {
+            code: 'it(`this is a test`, function () { });',
+            parserOptions: {
+                ecmaVersion: 2019
+            },
+            errors: [
+                { message: 'Invalid "it()" description found.', line: 1, column: 1 }
+            ]
+        },
+        {
+            code: 'const foo = "this"; it(`${foo} is a test`, function () { });',
+            parserOptions: {
+                ecmaVersion: 2019
+            },
+            errors: [
+                { message: 'Invalid "it()" description found.', line: 1, column: 21 }
+            ]
         }
     ]
 });