[jest] wait for beforeAll() to complete

fixes #1688

Author: alexlamsl

src/bun.js/test/jest.zig

@@ -1492,10 +1492,16 @@ pub const DescribeScope = struct {
         for (hooks) |cb, i| {
             if (cb.isEmpty()) continue;
 
-            const err = cb.call(ctx, &.{});
-            if (err.isAnyError(ctx)) {
-                return err;
+            const pending_test = Jest.runner.?.pending_test;
+            // forbid `expect()` within hooks
+            Jest.runner.?.pending_test = null;
+            var result = cb.call(ctx, &.{});
+            if (result.asPromise()) |promise| {
+                VirtualMachine.get().waitForPromise(promise);
+                result = promise.result(ctx.vm());
             }
+            Jest.runner.?.pending_test = pending_test;
+            if (result.isAnyError(ctx)) return result;
 
             if (comptime hook == .beforeAll or hook == .afterAll) {
                 hooks[i] = JSC.JSValue.zero;

test/bun.js/bun-test/jest-hooks.test.ts

@@ -73,4 +73,41 @@ describe("test jest hooks in bun-test", () => {
       expect(animal).toEqual("lion");
     });
   });
+
+  describe("test async hooks", async () => {
+    let beforeAllCalled = 0;
+    let beforeEachCalled = 0;
+    let afterAllCalled = 0;
+    let afterEachCalled = 0;
+
+    beforeAll(async () => {
+      beforeAllCalled += await 1;
+    });
+
+    beforeEach(async () => {
+      beforeEachCalled += await 1;
+    });
+
+    afterAll(async () => {
+      afterAllCalled += await 1;
+    });
+
+    afterEach(async () => {
+      afterEachCalled += await 1;
+    });
+
+    it("should run after beforeAll()", () => {
+      expect(beforeAllCalled).toBe(1);
+      expect(beforeEachCalled).toBe(1);
+      expect(afterAllCalled).toBe(0);
+      expect(afterEachCalled).toBe(0);
+    });
+
+    it("should run after beforeEach()", () => {
+      expect(beforeAllCalled).toBe(1);
+      expect(beforeEachCalled).toBe(2);
+      expect(afterAllCalled).toBe(0);
+      expect(afterEachCalled).toBe(1);
+    });
+  });
 });