Always expand first argument if the second argument is also a call expression (#4657)

src/language-js/printer-estree.js

@@ -3415,6 +3415,8 @@ function shouldGroupFirstArg(args) {
     (firstArg.type === "FunctionExpression" ||
       (firstArg.type === "ArrowFunctionExpression" &&
         firstArg.body.type === "BlockStatement")) &&
+    secondArg.type !== "FunctionExpression" &&
+    secondArg.type !== "ArrowFunctionExpression" &&
     !couldGroupArg(secondArg)
   );
 }

tests/first_argument_expansion/__snapshots__/jsfmt.spec.js.snap

@@ -88,6 +88,10 @@ setTimeout(
 setTimeout(/* blip */ function() {
   thing();
 }, 500);
+
+func((args) => {
+  execute(args);
+}, result => result && console.log("success"))
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 setTimeout(function() {
   thing();
@@ -200,4 +204,11 @@ setTimeout(
   500
 );
 
+func(
+  args => {
+    execute(args);
+  },
+  result => result && console.log("success")
+);
+
 `;

tests/first_argument_expansion/jsfmt.spec.js

@@ -1 +1 @@
-run_spec(__dirname, ["flow", "typescript"]);
+run_spec(__dirname, ["babylon", "flow", "typescript"]);

tests/first_argument_expansion/test.js

@@ -85,3 +85,7 @@ setTimeout(
 setTimeout(/* blip */ function() {
   thing();
 }, 500);
+
+func((args) => {
+  execute(args);
+}, result => result && console.log("success"))