From 632382b0699f77f1d03280cf0f2281f6ef461209 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C3=ABl=20Gallego?= <mic.gallego@gmail.com>
Date: Sat, 4 Jan 2014 19:11:55 +0100
Subject: [PATCH] Add missing methods to the mock

---
 .../ORM/Cache/Persister/AbstractEntityPersister.php  |  2 +-
 lib/Doctrine/ORM/LazyCriteriaCollection.php          | 12 +++++++++++-
 .../Cache/Persister/AbstractEntityPersisterTest.php  |  4 +++-
 3 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/lib/Doctrine/ORM/Cache/Persister/AbstractEntityPersister.php b/lib/Doctrine/ORM/Cache/Persister/AbstractEntityPersister.php
index 2f6ba34c8..b95becba2 100644
--- a/lib/Doctrine/ORM/Cache/Persister/AbstractEntityPersister.php
+++ b/lib/Doctrine/ORM/Cache/Persister/AbstractEntityPersister.php
@@ -161,7 +161,7 @@ abstract class AbstractEntityPersister implements CachedEntityPersister
      */
     public function getCountSQL($criteria = array())
     {
-        return $this->persister->count($criteria);
+        return $this->persister->getCountSQL($criteria);
     }
 
     /**
diff --git a/lib/Doctrine/ORM/LazyCriteriaCollection.php b/lib/Doctrine/ORM/LazyCriteriaCollection.php
index 8deb4b72f..b807885cd 100644
--- a/lib/Doctrine/ORM/LazyCriteriaCollection.php
+++ b/lib/Doctrine/ORM/LazyCriteriaCollection.php
@@ -23,6 +23,7 @@ use Closure;
 use Doctrine\Common\Collections\ArrayCollection;
 use Doctrine\Common\Collections\Collection;
 use Doctrine\Common\Collections\Criteria;
+use Doctrine\Common\Collections\Selectable;
 use Doctrine\ORM\Persisters\BasicEntityPersister;
 
 /**
@@ -31,7 +32,7 @@ use Doctrine\ORM\Persisters\BasicEntityPersister;
  * @since   2.5
  * @author  Michaël Gallego <mic.gallego@gmail.com>
  */
-class LazyCriteriaCollection implements Collection
+class LazyCriteriaCollection implements Collection, Selectable
 {
     /**
      * @var BasicEntityPersister
@@ -354,4 +355,13 @@ class LazyCriteriaCollection implements Collection
         $this->collection  = new ArrayCollection($elements);
         $this->initialized = true;
     }
+
+    /**
+     * {@inheritDoc}
+     */
+    function matching(Criteria $criteria)
+    {
+        $this->initialize();
+        return $this->collection->matching($criteria);
+    }
 }
diff --git a/tests/Doctrine/Tests/ORM/Cache/Persister/AbstractEntityPersisterTest.php b/tests/Doctrine/Tests/ORM/Cache/Persister/AbstractEntityPersisterTest.php
index 341db5ba6..4f81d0680 100644
--- a/tests/Doctrine/Tests/ORM/Cache/Persister/AbstractEntityPersisterTest.php
+++ b/tests/Doctrine/Tests/ORM/Cache/Persister/AbstractEntityPersisterTest.php
@@ -55,6 +55,7 @@ abstract class AbstractEntityPersisterTest extends OrmTestCase
         'getInserts',
         'getInsertSQL',
         'getSelectSQL',
+        'getCountSQL',
         'expandParameters',
         'expandCriteriaParameters',
         'getSelectConditionStatementSQL',
@@ -66,6 +67,7 @@ abstract class AbstractEntityPersisterTest extends OrmTestCase
         'load',
         'loadById',
         'loadOneToOneEntity',
+        'count',
         'refresh',
         'loadCriteria',
         'loadAll',
@@ -435,4 +437,4 @@ abstract class AbstractEntityPersisterTest extends OrmTestCase
 
         $this->assertNull($persister->exists($entity, array()));
     }
-}
\ No newline at end of file
+}