NativeDriverConnectionInterface

Author: simPod

tests/Tracing/Doctrine/DBAL/Fixture/NativeDriverConnectionInterface.php

@@ -0,0 +1,15 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sentry\SentryBundle\Tests\Tracing\Doctrine\DBAL\Fixture;
+
+use Sentry\SentryBundle\Tracing\Doctrine\DBAL\TracingDriverConnectionInterface;
+
+interface NativeDriverConnectionInterface extends TracingDriverConnectionInterface
+{
+    /**
+     * @return object|resource
+     */
+    public function getNativeConnection();
+}

tests/Tracing/Doctrine/DBAL/TracingDriverConnectionTest.php

@@ -8,9 +8,9 @@
 use Doctrine\DBAL\Driver\Result as DriverResultInterface;
 use Doctrine\DBAL\Driver\Statement as DriverStatementInterface;
 use Doctrine\DBAL\ParameterType;
-use PDO;
 use PHPUnit\Framework\MockObject\MockObject;
 use Sentry\SentryBundle\Tests\DoctrineTestCase;
+use Sentry\SentryBundle\Tests\Tracing\Doctrine\DBAL\Fixture\NativeDriverConnectionInterface;
 use Sentry\SentryBundle\Tracing\Doctrine\DBAL\TracingDriverConnection;
 use Sentry\SentryBundle\Tracing\Doctrine\DBAL\TracingStatement;
 use Sentry\State\HubInterface;
@@ -399,13 +399,16 @@ public function testGetWrappedConnection(): void
 
     public function testGetNativeConnection(): void
     {
-        $connection = new TracingDriverConnection($this->hub, $this->decoratedConnection, 'foo_platform', []);
+        $nativeConnection = new class() {
+        };
 
-        $this->decoratedConnection->expects($this->once())
-            ->method('getNativeConnection')
-            ->willReturn($this->createMock(PDO::class));
+        $decoratedConnection = $this->createMock(NativeDriverConnectionInterface::class);
+        $decoratedConnection->method('getNativeConnection')
+            ->willReturn($nativeConnection);
+
+        $connection = new TracingDriverConnection($this->hub, $decoratedConnection, 'foo_platform', []);
 
-        $this->assertSame($this->decoratedConnection, $connection->getNativeConnection());
+        $this->assertSame($nativeConnection, $connection->getNativeConnection());
 
     }
 

tests/Tracing/Doctrine/DBAL/TracingServerInfoAwareDriverConnectionTest.php

@@ -8,9 +8,9 @@
 use Doctrine\DBAL\Driver\Result;
 use Doctrine\DBAL\Driver\Statement;
 use Doctrine\DBAL\ParameterType;
-use PDO;
 use PHPUnit\Framework\MockObject\MockObject;
 use Sentry\SentryBundle\Tests\DoctrineTestCase;
+use Sentry\SentryBundle\Tests\Tracing\Doctrine\DBAL\Fixture\NativeDriverConnectionInterface;
 use Sentry\SentryBundle\Tracing\Doctrine\DBAL\TracingDriverConnectionInterface;
 use Sentry\SentryBundle\Tracing\Doctrine\DBAL\TracingServerInfoAwareDriverConnection;
 
@@ -260,12 +260,15 @@ public function testGetWrappedConnection(): void
 
     public function testGetNativeConnection(): void
     {
-        $nativeConnection = $this->createMock(PDO::class);
+        $nativeConnection = new class() {
+        };
 
-        $this->decoratedConnection->expects($this->once())
-            ->method('getNativeConnection')
+        $decoratedConnection = $this->createMock(NativeDriverConnectionInterface::class);
+        $decoratedConnection->method('getNativeConnection')
             ->willReturn($nativeConnection);
 
-        $this->assertSame($nativeConnection, $this->connection->getWrappedConnection());
+        $connection = new TracingServerInfoAwareDriverConnection($decoratedConnection);
+
+        $this->assertSame($nativeConnection, $connection->getNativeConnection());
     }
 }