test: mock context

Author: aladdin-add

.nycrc

@@ -1,5 +1,5 @@
 {
-  "branches": 95,
+  "branches": 94,
   "lines": 99,
   "functions": 99,
   "statements": 99

lib/utils.js

@@ -468,9 +468,10 @@ module.exports = {
             isRuleTesterConstruction(declarator.init) &&
             declarator.id.type === 'Identifier'
           ) {
-            (sourceCode.getDeclaredVariables || context.getDeclaredVariables)(
-              declarator
-            ).forEach((variable) => {
+            const vars = sourceCode.getDeclaredVariables
+              ? sourceCode.getDeclaredVariables(declarator)
+              : context.getDeclaredVariables(declarator);
+            vars.forEach((variable) => {
               variable.references
                 .filter((ref) => ref.isRead())
                 .forEach((ref) => variableIdentifiers.add(ref.identifier));

tests/lib/utils.js

@@ -761,8 +761,14 @@ describe('utils', () => {
             sourceType: 'script',
             nodejsScope: true,
           });
+          const context = {
+            sourceCode: {
+              getDeclaredVariables:
+                scopeManager.getDeclaredVariables.bind(scopeManager),
+            },
+          }; // mock object
           assert.deepEqual(
-            utils.getTestInfo(scopeManager, ast),
+            utils.getTestInfo(context, ast),
             [],
             'Expected no tests to be found'
           );
@@ -827,7 +833,13 @@ describe('utils', () => {
             sourceType: 'script',
             nodejsScope: true,
           });
-          const testInfo = utils.getTestInfo(scopeManager, ast);
+          const context = {
+            sourceCode: {
+              getDeclaredVariables:
+                scopeManager.getDeclaredVariables.bind(scopeManager),
+            },
+          }; // mock object
+          const testInfo = utils.getTestInfo(context, ast);
 
           assert.strictEqual(
             testInfo.length,
@@ -1021,7 +1033,13 @@ describe('utils', () => {
             sourceType: 'script',
             nodejsScope: true,
           });
-          const testInfo = utils.getTestInfo(scopeManager, ast);
+          const context = {
+            sourceCode: {
+              getDeclaredVariables:
+                scopeManager.getDeclaredVariables.bind(scopeManager),
+            },
+          }; // mock object
+          const testInfo = utils.getTestInfo(context, ast);
 
           assert.strictEqual(
             testInfo.length,