SQLFilters enahancements

lib/Doctrine/ORM/Query/Filter/SQLFilter.php

@@ -19,7 +19,7 @@
 
 namespace Doctrine\ORM\Query\Filter;
 
-use Doctrine\ORM\EntityManager;
+use Doctrine\ORM\EntityManagerInterface;
 use Doctrine\ORM\Mapping\ClassMetadata;
 use Doctrine\ORM\Query\ParameterTypeInferer;
 
@@ -37,7 +37,7 @@ abstract class SQLFilter
     /**
      * The entity manager.
      *
-     * @var EntityManager
+     * @var EntityManagerInterface
      */
     private $em;
 
@@ -51,9 +51,9 @@ abstract class SQLFilter
     /**
      * Constructs the SQLFilter object.
      *
-     * @param EntityManager $em The entity manager.
+     * @param EntityManagerInterface $em The entity manager.
      */
-    final public function __construct(EntityManager $em)
+    final public function __construct(EntityManagerInterface $em)
     {
         $this->em = $em;
     }
@@ -133,6 +133,16 @@ final public function __toString()
         return serialize($this->parameters);
     }
 
+    /**
+     * Returns the database connection used by the entity manager
+     *
+     * @return \Doctrine\DBAL\Connection
+     */
+    final protected function getConnection()
+    {
+        return $this->em->getConnection();
+    }
+
     /**
      * Gets the SQL query part to add to a query.
      *

tests/Doctrine/Tests/ORM/Functional/SQLFilterTest.php

@@ -243,6 +243,24 @@ public function testSQLFilterGetSetParameter()
         $this->assertEquals("'en'", $filter->getParameter('locale'));
     }
 
+    public function testSQLFilterGetConnection()
+    {
+        // Setup mock connection
+        $conn = $this->getMockConnection();
+
+        $em = $this->getMockEntityManager($conn);
+        $em->expects($this->once())
+            ->method('getConnection')
+            ->will($this->returnValue($conn));
+
+        $filter = new MyLocaleFilter($em);
+
+        $reflMethod = new \ReflectionMethod('Doctrine\ORM\Query\Filter\SQLFilter', 'getConnection');
+        $reflMethod->setAccessible(true);
+
+        $this->assertTrue($reflMethod->invoke($filter) === $conn);
+    }
+
     public function testSQLFilterSetParameterInfersType()
     {
         // Setup mock connection