nodejs
diff --git a/Collapse file
‎test/parallel/test-eslint-prefer-assert-iferror.js‎
Copy file name to clipboardExpand all lines: test/parallel/test-eslint-prefer-assert-iferror.js
+10-4Lines changed: 10 additions & 4 deletions b/Collapse file
‎test/parallel/test-eslint-prefer-assert-iferror.js‎
Copy file name to clipboardExpand all lines: test/parallel/test-eslint-prefer-assert-iferror.js
+10-4Lines changed: 10 additions & 4 deletions
diff --git a/Collapse file
‎tools/eslint-rules/prefer-assert-iferror.js‎
Copy file name to clipboardExpand all lines: tools/eslint-rules/prefer-assert-iferror.js
+21-2Lines changed: 21 additions & 2 deletions b/Collapse file
‎tools/eslint-rules/prefer-assert-iferror.js‎
Copy file name to clipboardExpand all lines: tools/eslint-rules/prefer-assert-iferror.js
+21-2Lines changed: 21 additions & 2 deletions
@@ -14,12 +14,18 @@ new RuleTester().run('prefer-assert-iferror', rule, {
   ],
   invalid: [
     {
-      code: 'if (err) throw err;',
-      errors: [{ message: 'Use assert.ifError(err) instead.' }]
+      code: 'require("assert");\n' +
+            'if (err) throw err;',
+      errors: [{ message: 'Use assert.ifError(err) instead.' }],
+      output: 'require("assert");\n' +
+              'assert.ifError(err);'
     },
     {
-      code: 'if (error) { throw error; }',
-      errors: [{ message: 'Use assert.ifError(error) instead.' }]
+      code: 'require("assert");\n' +
+            'if (error) { throw error; }',
+      errors: [{ message: 'Use assert.ifError(error) instead.' }],
+      output: 'require("assert");\n' +
+              'assert.ifError(error);'
     }
   ]
 });
@@ -5,9 +5,12 @@
 
 'use strict';
 
+const utils = require('./rules-utils.js');
+
 module.exports = {
   create(context) {
     const sourceCode = context.getSourceCode();
+    var assertImported = false;
 
     function hasSameTokens(nodeA, nodeB) {
       const aTokens = sourceCode.getTokens(nodeA);
@@ -20,8 +23,15 @@ module.exports = {
         });
     }
 
+    function checkAssertNode(node) {
+      if (utils.isRequired(node, ['assert'])) {
+        assertImported = true;
+      }
+    }
+
     return {
-      IfStatement(node) {
+      'CallExpression': (node) => checkAssertNode(node),
+      'IfStatement': (node) => {
         const firstStatement = node.consequent.type === 'BlockStatement' ?
           node.consequent.body[0] :
           node.consequent;
@@ -30,10 +40,19 @@ module.exports = {
           firstStatement.type === 'ThrowStatement' &&
           hasSameTokens(node.test, firstStatement.argument)
         ) {
+          const argument = sourceCode.getText(node.test);
           context.report({
             node: firstStatement,
             message: 'Use assert.ifError({{argument}}) instead.',
-            data: { argument: sourceCode.getText(node.test) }
+            data: { argument },
+            fix: (fixer) => {
+              if (assertImported) {
+                return fixer.replaceText(
+                  node,
+                  `assert.ifError(${argument});`
+                );
+              }
+            }
           });
         }
       }
Original file line number	Diff line number	Diff line change
`@@ -14,12 +14,18 @@ new RuleTester().run('prefer-assert-iferror', rule, {`
`14`	`14`	`],`
`15`	`15`	`invalid: [`
`16`	`16`	`{`
`17`		`- code: 'if (err) throw err;',`
`18`		`- errors: [{ message: 'Use assert.ifError(err) instead.' }]`
	`17`	`+ code: 'require("assert");\n' +`
	`18`	`+ 'if (err) throw err;',`
	`19`	`+ errors: [{ message: 'Use assert.ifError(err) instead.' }],`
	`20`	`+ output: 'require("assert");\n' +`
	`21`	`+ 'assert.ifError(err);'`
`19`	`22`	`},`
`20`	`23`	`{`
`21`		`- code: 'if (error) { throw error; }',`
`22`		`- errors: [{ message: 'Use assert.ifError(error) instead.' }]`
	`24`	`+ code: 'require("assert");\n' +`
	`25`	`+ 'if (error) { throw error; }',`
	`26`	`+ errors: [{ message: 'Use assert.ifError(error) instead.' }],`
	`27`	`+ output: 'require("assert");\n' +`
	`28`	`+ 'assert.ifError(error);'`
`23`	`29`	`}`
`24`	`30`	`]`
`25`	`31`	`});`