override afterEach

Author: adrien2p

packages/medusa-test-utils/src/medusa-test-runner.ts

@@ -95,6 +95,29 @@ class MedusaTestRunner {
     }
 
     this.setupProcessHandlers()
+
+    /**
+     * Override the afterEach hook to wait for workflow executions to finish.
+     * This is necessary to ensure that the test are not considered done while
+     * background workflow executions might still be running and can interfere with
+     * the other tests.
+     */
+    // @ts-ignore
+    if (!global.afterEach.medusaTestRunnerAfterEachOverridden) {
+      const originalAfterEach = global.afterEach
+      global.afterEach = (fn, timeout) => {
+        originalAfterEach(async (...args: any[]) => {
+          await waitWorkflowExecutions(this.globalContainer as MedusaContainer)
+          // @ts-ignore
+          return fn(...args)
+        }, timeout)
+      }
+      Object.defineProperty(global, "medusaTestRunnerAfterEachOverridden", {
+        value: true,
+        configurable: true,
+        enumerable: false,
+      })
+    }
   }
 
   private setupProcessHandlers(): void {
@@ -272,7 +295,6 @@ class MedusaTestRunner {
 
   public async afterEach(): Promise<void> {
     try {
-      await waitWorkflowExecutions(this.globalContainer as MedusaContainer)
       await this.dbUtils.teardown({ schema: this.schema })
     } catch (error) {
       logger.error("Error tearing down database:", error?.message)