Add way to call hooks for tests running in coroutines (#7151)

hyperf · Nov 20, 2024 · d703d28 · d703d28
1 parent 3cba517
commit d703d28
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/Concerns/RunTestsInCoroutine.php b/src/Concerns/RunTestsInCoroutine.php
@@ -37,10 +37,12 @@ final protected function runTestsInCoroutine(...$arguments)
         /* @phpstan-ignore-next-line */
         \Swoole\Coroutine\run(function () use (&$testResult, &$exception, $arguments) {
             try {
+                $this->invokeBeforeHookMethods();
                 $testResult = $this->{$this->realTestName}(...$arguments);
             } catch (Throwable $e) {
                 $exception = $e;
             } finally {
+                $this->invokeAfterHookMethods();
                 Timer::clearAll();
                 CoordinatorManager::until(Constants::WORKER_EXIT)->resume();
             }
@@ -62,4 +64,18 @@ final protected function runTest(): mixed
 
         return parent::runTest();
     }
+
+    private function invokeBeforeHookMethods(): void
+    {
+        if (method_exists($this, 'beforeTestInCoroutine')) {
+            call_user_func([$this, 'beforeTestInCoroutine']);
+        }
+    }
+
+    private function invokeAfterHookMethods(): void
+    {
+        if (method_exists($this, 'afterTestInCoroutine')) {
+            call_user_func([$this, 'afterTestInCoroutine']);
+        }
+    }
 }