Test with RuntimeException as well

Author: dominikzogg

tests/Unit/Adapter/BlackfireOnRequestAdapterTest.php

@@ -6,6 +6,7 @@
 
 use Blackfire\Client;
 use Blackfire\Exception\LogicException;
+use Blackfire\Exception\RuntimeException;
 use Blackfire\Probe;
 use Blackfire\Profile\Configuration;
 use Chubbyphp\Mock\MockMethod\WithCallback;
@@ -121,7 +122,7 @@ public function testInvokeWithConfigAndWithLogger(): void
     }
 
     #[DoesNotPerformAssertions]
-    public function testInvokeWithExceptionOnCreateProbe(): void
+    public function testInvokeWithLogicExceptionOnCreateProbe(): void
     {
         $builder = new MockObjectBuilder();
 
@@ -157,7 +158,43 @@ public function testInvokeWithExceptionOnCreateProbe(): void
     }
 
     #[DoesNotPerformAssertions]
-    public function testInvokeWithExceptionOnProbeEnd(): void
+    public function testInvokeWithRuntimeExceptionOnCreateProbe(): void
+    {
+        $builder = new MockObjectBuilder();
+
+        /** @var SwooleRequest $swooleRequest */
+        $swooleRequest = $builder->create(SwooleRequest::class, []);
+        $swooleRequest->header['x-blackfire-query'] = 'swoole';
+
+        /** @var SwooleResponse $swooleResponse */
+        $swooleResponse = $builder->create(SwooleResponse::class, []);
+
+        /** @var OnRequestInterface $onRequest */
+        $onRequest = $builder->create(OnRequestInterface::class, [
+            new WithoutReturn('__invoke', [$swooleRequest, $swooleResponse]),
+        ]);
+
+        /** @var Configuration $config */
+        $config = $builder->create(Configuration::class, []);
+
+        $exception = new RuntimeException('Something went wrong');
+
+        /** @var Client $client */
+        $client = $builder->create(Client::class, [
+            new WithException('createProbe', [$config, true], $exception),
+        ]);
+
+        /** @var LoggerInterface $logger */
+        $logger = $builder->create(LoggerInterface::class, [
+            new WithoutReturn('error', ['Blackfire exception: Something went wrong', []]),
+        ]);
+
+        $adapter = new BlackfireOnRequestAdapter($onRequest, $client, $config, $logger);
+        $adapter($swooleRequest, $swooleResponse);
+    }
+
+    #[DoesNotPerformAssertions]
+    public function testInvokeWithLogicExceptionOnProbeEnd(): void
     {
         $builder = new MockObjectBuilder();
 
@@ -195,4 +232,44 @@ public function testInvokeWithExceptionOnProbeEnd(): void
         $adapter = new BlackfireOnRequestAdapter($onRequest, $client, $config, $logger);
         $adapter($swooleRequest, $swooleResponse);
     }
+
+    #[DoesNotPerformAssertions]
+    public function testInvokeWithRuntimeExceptionOnProbeEnd(): void
+    {
+        $builder = new MockObjectBuilder();
+
+        /** @var SwooleRequest $swooleRequest */
+        $swooleRequest = $builder->create(SwooleRequest::class, []);
+        $swooleRequest->header['x-blackfire-query'] = 'swoole';
+
+        /** @var SwooleResponse $swooleResponse */
+        $swooleResponse = $builder->create(SwooleResponse::class, []);
+
+        /** @var OnRequestInterface $onRequest */
+        $onRequest = $builder->create(OnRequestInterface::class, [
+            new WithoutReturn('__invoke', [$swooleRequest, $swooleResponse]),
+        ]);
+
+        /** @var Configuration $config */
+        $config = $builder->create(Configuration::class, []);
+
+        /** @var Probe $probe */
+        $probe = $builder->create(Probe::class, []);
+
+        $exception = new RuntimeException('Something went wrong');
+
+        /** @var Client $client */
+        $client = $builder->create(Client::class, [
+            new WithReturn('createProbe', [$config, true], $probe),
+            new WithException('endProbe', [$probe], $exception),
+        ]);
+
+        /** @var LoggerInterface $logger */
+        $logger = $builder->create(LoggerInterface::class, [
+            new WithoutReturn('error', ['Blackfire exception: Something went wrong', []]),
+        ]);
+
+        $adapter = new BlackfireOnRequestAdapter($onRequest, $client, $config, $logger);
+        $adapter($swooleRequest, $swooleResponse);
+    }
 }