Support modify the context of sub coroutine when using test component…

…s. (#5962)
hyperf · Jul 21, 2023 · fa9d8c1 · fa9d8c1
1 parent e660ba6
commit fa9d8c1
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/src/Client.php b/src/Client.php
@@ -145,16 +145,18 @@ public function file(string $uri, array $data = [], array $headers = [])
         return $this->packer->unpack((string) $response->getBody());
     }
 
-    public function request(string $method, string $path, array $options = [])
+    public function request(string $method, string $path, array $options = [], ?callable $callable = null)
     {
-        return wait(function () use ($method, $path, $options) {
+        return wait(function () use ($method, $path, $options, $callable) {
+            $callable && $callable();
             return $this->execute($this->initRequest($method, $path, $options));
         }, $this->waitTimeout);
     }
 
-    public function sendRequest(ServerRequestInterface $psr7Request): ResponseInterface
+    public function sendRequest(ServerRequestInterface $psr7Request, ?callable $callable = null): ResponseInterface
     {
-        return wait(function () use ($psr7Request) {
+        return wait(function () use ($psr7Request, $callable) {
+            $callable && $callable();
             return $this->execute($psr7Request);
         }, $this->waitTimeout);
     }

diff --git a/tests/ClientTest.php b/tests/ClientTest.php
@@ -14,6 +14,7 @@
 use Hyperf\Codec\Json;
 use Hyperf\Config\Config;
 use Hyperf\Context\ApplicationContext;
+use Hyperf\Context\Context;
 use Hyperf\Contract\ConfigInterface;
 use Hyperf\Contract\NormalizerInterface;
 use Hyperf\Coroutine\Coroutine;
@@ -119,6 +120,19 @@ public function testClientGetUri()
         $this->assertSame($id, $data['params']['id']);
     }
 
+    public function testClientCallable()
+    {
+        $container = $this->getContainer();
+
+        $client = new Client($container);
+
+        $id = uniqid();
+
+        $res = $client->request('GET', '/context', callable: fn () => Context::set('request_id', $id));
+
+        $this->assertSame(['request_id' => $id], Json::decode((string) $res->getBody()));
+    }
+
     public function getContainer()
     {
         $container = Mockery::mock(Container::class);
@@ -175,6 +189,7 @@ public function getContainer()
         Router::get('/', [FooController::class, 'index']);
         Router::get('/exception', [FooController::class, 'exception']);
         Router::get('/id', [FooController::class, 'id']);
+        Router::get('/context', [FooController::class, 'context']);
         Router::addRoute(['GET', 'POST'], '/request', [FooController::class, 'request']);
 
         return $container;

diff --git a/tests/Stub/FooController.php b/tests/Stub/FooController.php
@@ -33,6 +33,13 @@ public function id()
         return ['code' => 0, 'data' => Coroutine::id()];
     }
 
+    public function context()
+    {
+        return [
+            'request_id' => Context::getOrSet('request_id', uniqid()),
+        ];
+    }
+
     public function request()
     {
         /** @var ServerRequestInterface $request */