format jest calls like regular calls (#5011)

src/language-js/printer-estree.js

@@ -5800,10 +5800,7 @@ function isTestCall(n, parent) {
     }
 
     if (isUnitTestSetUp(n)) {
-      return (
-        isFunctionOrArrowExpression(n.arguments[0].type) ||
-        isAngularTestWrapper(n.arguments[0])
-      );
+      return isAngularTestWrapper(n.arguments[0]);
     }
   } else if (n.arguments.length === 2) {
     if (

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

@@ -5,10 +5,22 @@ beforeEach(async(() => {
   // code
 }));
 
+beforeEach(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 afterAll(async(() => {
   console.log('Hello');
 }));
 
+afterAll(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 it('should create the app', async(() => {
   //code
 }));
@@ -28,10 +40,22 @@ beforeEach(async(() => {
   // code
 }));
 
+beforeEach(done =>
+  foo()
+    .bar()
+    .bar()
+);
+
 afterAll(async(() => {
   console.log("Hello");
 }));
 
+afterAll(done =>
+  foo()
+    .bar()
+    .bar()
+);
+
 it("should create the app", async(() => {
   //code
 }));
@@ -56,10 +80,22 @@ beforeEach(async(() => {
   // code
 }));
 
+beforeEach(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 afterAll(async(() => {
   console.log('Hello');
 }));
 
+afterAll(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 it('should create the app', async(() => {
   //code
 }));
@@ -79,10 +115,22 @@ beforeEach(async(() => {
   // code
 }));
 
+beforeEach((done) =>
+  foo()
+    .bar()
+    .bar()
+);
+
 afterAll(async(() => {
   console.log("Hello");
 }));
 
+afterAll((done) =>
+  foo()
+    .bar()
+    .bar()
+);
+
 it("should create the app", async(() => {
   //code
 }));

tests/test_declarations/angular_async.js

@@ -2,10 +2,22 @@ beforeEach(async(() => {
   // code
 }));
 
+beforeEach(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 afterAll(async(() => {
   console.log('Hello');
 }));
 
+afterAll(done =>
+  foo()
+    .bar()
+    .bar(),
+);
+
 it('should create the app', async(() => {
   //code
 }));