Fix jsx-wrap-multilines ternaries handling (fixes #916)

Author: Yannick Croissant

lib/rules/jsx-wrap-multilines.js

@@ -92,15 +92,27 @@ module.exports = {
     return {
 
       VariableDeclarator: function(node) {
-        if (isEnabled('declaration')) {
-          check(node.init);
+        if (!isEnabled('declaration')) {
+          return;
         }
+        if (node.init && node.init.type === 'ConditionalExpression') {
+          check(node.init.consequent);
+          check(node.init.alternate);
+          return;
+        }
+        check(node.init);
       },
 
       AssignmentExpression: function(node) {
-        if (isEnabled('assignment')) {
-          check(node.right);
+        if (!isEnabled('assignment')) {
+          return;
+        }
+        if (node.right.type === 'ConditionalExpression') {
+          check(node.right.consequent);
+          check(node.right.alternate);
+          return;
         }
+        check(node.right);
       },
 
       ReturnStatement: function(node) {

tests/lib/rules/jsx-wrap-multilines.js

@@ -47,6 +47,40 @@ var RETURN_NO_PAREN = '\
     }\
   });';
 
+var DECLARATION_TERNARY_SINGLE_LINE = 'var hello = foo ? <p>Hello</p> : <p>Hi</p>;';
+
+var DECLARATION_TERNARY_PAREN = '\
+  var hello = foo ? (<div>\n\
+    <p>Hello</p>\n\
+  </div>) : (<div>\n\
+    <p>Hi</p>\n\
+  </div>);';
+
+var DECLARATION_TERNARY_NO_PAREN = '\
+  var hello = foo ? <div>\n\
+    <p>Hello</p>\n\
+  </div> : <div>\n\
+    <p>Hi</p>\n\
+  </div>;';
+
+var ASSIGNMENT_TERNARY_SINGLE_LINE = 'var hello; hello = foo ? <p>Hello</p> : <p>Hi</p>;';
+
+var ASSIGNMENT_TERNARY_PAREN = '\
+  var hello;\n\
+  hello = foo ? (<div>\n\
+    <p>Hello</p>\n\
+  </div>) : (<div>\n\
+    <p>Hi</p>\n\
+  </div>);';
+
+var ASSIGNMENT_TERNARY_NO_PAREN = '\
+  var hello;\n\
+  hello = foo ? <div>\n\
+    <p>Hello</p>\n\
+  </div> : <div>\n\
+    <p>Hi</p>\n\
+  </div>;';
+
 var DECLARATION_SINGLE_LINE = 'var hello = <p>Hello</p>;';
 
 var DECLARATION_PAREN = '\
@@ -91,6 +125,26 @@ ruleTester.run('jsx-wrap-multilines', rule, {
       code: RETURN_NO_PAREN,
       options: [{return: false}],
       parserOptions: parserOptions
+    }, {
+      code: DECLARATION_TERNARY_SINGLE_LINE,
+      parserOptions: parserOptions
+    }, {
+      code: DECLARATION_TERNARY_PAREN,
+      parserOptions: parserOptions
+    }, {
+      code: DECLARATION_TERNARY_NO_PAREN,
+      options: [{declaration: false}],
+      parserOptions: parserOptions
+    }, {
+      code: ASSIGNMENT_TERNARY_SINGLE_LINE,
+      parserOptions: parserOptions
+    }, {
+      code: ASSIGNMENT_TERNARY_PAREN,
+      parserOptions: parserOptions
+    }, {
+      code: ASSIGNMENT_TERNARY_NO_PAREN,
+      options: [{assignment: false}],
+      parserOptions: parserOptions
     }, {
       code: DECLARATION_SINGLE_LINE,
       parserOptions: parserOptions
@@ -127,6 +181,22 @@ ruleTester.run('jsx-wrap-multilines', rule, {
       parserOptions: parserOptions,
       options: [{return: true}],
       errors: [{message: 'Missing parentheses around multilines JSX'}]
+    }, {
+      code: DECLARATION_TERNARY_NO_PAREN,
+      output: DECLARATION_TERNARY_PAREN,
+      parserOptions: parserOptions,
+      errors: [
+        {message: 'Missing parentheses around multilines JSX'},
+        {message: 'Missing parentheses around multilines JSX'}
+      ]
+    }, {
+      code: ASSIGNMENT_TERNARY_NO_PAREN,
+      output: ASSIGNMENT_TERNARY_PAREN,
+      parserOptions: parserOptions,
+      errors: [
+        {message: 'Missing parentheses around multilines JSX'},
+        {message: 'Missing parentheses around multilines JSX'}
+      ]
     }, {
       code: DECLARATION_NO_PAREN,
       output: DECLARATION_PAREN,