From 462481ebbe1d57eff6bb39314704e788816f87f8 Mon Sep 17 00:00:00 2001
From: Marco Pivetta <ocramius@gmail.com>
Date: Wed, 21 Jun 2017 05:47:19 +0200
Subject: [PATCH] #6167 #6168 - better connection mock
 documentation/docblocks/return-types

---
 tests/Doctrine/Tests/Mocks/ConnectionMock.php | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tests/Doctrine/Tests/Mocks/ConnectionMock.php b/tests/Doctrine/Tests/Mocks/ConnectionMock.php
index 0b5685d2a..af8fc252e 100644
--- a/tests/Doctrine/Tests/Mocks/ConnectionMock.php
+++ b/tests/Doctrine/Tests/Mocks/ConnectionMock.php
@@ -1,6 +1,7 @@
 <?php
 
 namespace Doctrine\Tests\Mocks;
+
 use Doctrine\DBAL\Connection;
 use Doctrine\DBAL\Driver\Statement;
 
@@ -15,12 +16,12 @@ class ConnectionMock extends Connection
     private $_fetchOneResult;
 
     /**
-     * @var \Exception
+     * @var \Exception|null
      */
     private $_fetchOneException;
 
     /**
-     * @var Statement
+     * @var Statement|null
      */
     private $_queryResult;
 
@@ -107,7 +108,7 @@ class ConnectionMock extends Connection
     /**
      * {@inheritdoc}
      */
-    public function query()
+    public function query() : Statement
     {
         return $this->_queryResult;
     }