Improve requireSpaceAfterBinaryOperators rule

Catch nodes with "VariableDeclarator" type

Author: markelog

lib/rules/require-space-after-binary-operators.js

@@ -71,34 +71,36 @@ module.exports.prototype = {
 
         // For everything else
         file.iterateNodesByType(
-            ['BinaryExpression', 'AssignmentExpression', 'LogicalExpression'],
+            ['BinaryExpression', 'AssignmentExpression', 'VariableDeclarator', 'LogicalExpression'],
             function(node) {
-                if (operators[node.operator]) {
-                    var indent;
-                    var range = node.right.range[0];
-
-                    if (tokenHelper.isTokenParenthesis(file, range - 1, true)) {
-                        indent = node.operator.length + 1;
-                    } else {
-                        indent = node.operator.length;
-                    }
-
-                    var part = tokenHelper.getTokenByRangeStartIfPunctuator(
-                        file,
-                        range - indent,
-                        node.operator,
-                        true
-                    );
+                var isDec = node.type === 'VariableDeclarator';
+                var operator = isDec ? '=' : node.operator;
+
+                if (!operators[operator] || node.init === null) {
+                    return;
+                }
+
+                var range = (isDec ? node.init : node.right).range[0];
 
-                    if (part) {
-                        var loc = part.loc.start;
+                var indent = tokenHelper.isTokenParenthesis(file, range - 1, true) ?
+                    operator.length + 1 :
+                    operator.length;
 
-                        errors.add(
-                            'Operator ' + node.operator + ' should not stick to following expression',
-                            loc.line,
-                            tokenHelper.getPointerEntities(loc.column, node.operator.length)
-                        );
-                    }
+                var part = tokenHelper.getTokenByRangeStartIfPunctuator(
+                    file,
+                    range - indent,
+                    operator,
+                    true
+                );
+
+                if (part) {
+                    var loc = part.loc.start;
+
+                    errors.add(
+                        'Operator ' + operator + ' should not stick to following expression',
+                        loc.line,
+                        tokenHelper.getPointerEntities(loc.column, operator.length)
+                    );
                 }
             }
         );

test/rules/require-space-after-binary-operators.js

@@ -73,4 +73,20 @@ describe('rules/require-space-after-binary-operators', function() {
         checker.configure({ requireSpaceAfterBinaryOperators: [','] });
         assert(checker.checkString('function test(a,b){}').getErrorCount() === 1);
     });
+    it('should report for assignment expression', function() {
+        checker.configure({ requireSpaceAfterBinaryOperators: ['='] });
+        assert(checker.checkString('var x=1').getErrorCount() === 1);
+    });
+    it('should report for assignment expressions', function() {
+        checker.configure({ requireSpaceAfterBinaryOperators: ['='] });
+        assert(checker.checkString('var x=1, t=2').getErrorCount() === 2);
+    });
+    it('should not report for assignment expressions without "=" sign', function() {
+        checker.configure({ requireSpaceAfterBinaryOperators: ['='] });
+        assert(checker.checkString('var x,z;').isEmpty());
+    });
+    it('should not report for assignment expressions if "=" is not specified', function() {
+        checker.configure({ requireSpaceAfterBinaryOperators: [','] });
+        assert(checker.checkString('var x=1;').isEmpty());
+    });
 });