add a test for the listen

goetas · goetas · commit b462665a13c9 · 2022-08-07T21:42:05.000+02:00
diff --git a/src/Symfony/Component/Messenger/Bridge/Doctrine/Tests/Transport/PostgreSqlConnectionTest.php b/src/Symfony/Component/Messenger/Bridge/Doctrine/Tests/Transport/PostgreSqlConnectionTest.php
@@ -11,6 +11,11 @@
 
 namespace Symfony\Component\Messenger\Bridge\Doctrine\Tests\Transport;
 
+use Doctrine\DBAL\Cache\ArrayResult;
+use Doctrine\DBAL\Cache\ArrayStatement;
+use Doctrine\DBAL\Platforms\PostgreSQLPlatform;
+use Doctrine\DBAL\Query\QueryBuilder;
+use Doctrine\DBAL\Result;
 use Doctrine\DBAL\Schema\Table;
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Messenger\Bridge\Doctrine\Transport\PostgreSqlConnection;
@@ -42,6 +47,68 @@ public function testUnserialize()
         $connection->__wakeup();
     }
 
+    public function testListenOnConnection()
+    {
+        $driverConnection = $this->createMock(\Doctrine\DBAL\Connection::class);
+
+        $driverConnection
+            ->expects(self::any())
+            ->method('getDatabasePlatform')
+            ->willReturn(new PostgreSQLPlatform());
+
+        $driverConnection
+            ->expects(self::any())
+            ->method('createQueryBuilder')
+            ->willReturn(new QueryBuilder($driverConnection));
+
+        $wrappedConnection = new class() {
+            private $notifyCalls = 0;
+
+            public function pgsqlGetNotify()
+            {
+                ++$this->notifyCalls;
+
+                return false;
+            }
+
+            public function countNotifyCalls()
+            {
+                return $this->notifyCalls;
+            }
+        };
+
+        // dbal 2.x
+        if (interface_exists(Result::class)) {
+            $driverConnection
+                ->expects(self::exactly(2))
+                ->method('getWrappedConnection')
+                ->willReturn($wrappedConnection);
+
+            $driverConnection
+                ->expects(self::any())
+                ->method('executeQuery')
+                ->willReturn(new ArrayStatement([]));
+        } else {
+            // dbal 3.x
+            $driverConnection
+                ->expects(self::exactly(2))
+                ->method('getNativeConnection')
+                ->willReturn($wrappedConnection);
+
+            $driverConnection
+                ->expects(self::any())
+                ->method('executeQuery')
+                ->willReturn(new Result(new ArrayResult([]), $driverConnection));
+        }
+        $connection = new PostgreSqlConnection(['table_name' => 'queue_table'], $driverConnection);
+
+        $connection->get(); // first time we have queueEmptiedAt === null, fallback on the parent implementation
+        $connection->get();
+        $connection->get();
+
+        $this->assertSame(2, $wrappedConnection->countNotifyCalls());
+    }
+
     public function testGetExtraSetupSql()
     {
         $driverConnection = $this->createMock(\Doctrine\DBAL\Connection::class);
