5 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
6 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
7 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
8 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
9 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
10 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
11 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
12 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
13 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
14 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
15 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
17 * This software consists of voluntary contributions made by many individuals
18 * and is licensed under the LGPL. For more information, see
19 * <http://www.doctrine-project.org>.
22 namespace Doctrine\Tests\Mocks;
24 use Doctrine\ORM\Proxy\ProxyFactory;
27 * Special EntityManager mock used for testing purposes.
29 class EntityManagerMock extends \Doctrine\ORM\EntityManager
32 private $_proxyFactoryMock;
33 private $_idGenerators = array();
38 public function getUnitOfWork()
40 return isset($this->_uowMock) ? $this->_uowMock : parent::getUnitOfWork();
46 * Sets a (mock) UnitOfWork that will be returned when getUnitOfWork() is called.
50 public function setUnitOfWork($uow)
52 $this->_uowMock = $uow;
55 public function setProxyFactory($proxyFactory)
57 $this->_proxyFactoryMock = $proxyFactory;
60 public function getProxyFactory()
62 return isset($this->_proxyFactoryMock) ? $this->_proxyFactoryMock : parent::getProxyFactory();
66 * Mock factory method to create an EntityManager.
68 * @param unknown_type $conn
69 * @param unknown_type $name
70 * @param Doctrine_Configuration $config
71 * @param Doctrine_EventManager $eventManager
72 * @return Doctrine\ORM\EntityManager
74 public static function create($conn, \Doctrine\ORM\Configuration $config = null,
75 \Doctrine\Common\EventManager $eventManager = null)
77 if (is_null($config)) {
78 $config = new \Doctrine\ORM\Configuration();
79 $config->setProxyDir(__DIR__ . '/../Proxies');
80 $config->setProxyNamespace('Doctrine\Tests\Proxies');
81 $config->setMetadataDriverImpl($config->newDefaultAnnotationDriver(array(), true));
83 if (is_null($eventManager)) {
84 $eventManager = new \Doctrine\Common\EventManager();
87 return new EntityManagerMock($conn, $config, $eventManager);
90 public function setIdGenerator($className, $generator)
92 $this->_idGenerators[$className] = $generator;
96 /* public function getIdGenerator($className)
99 if (isset($this->_idGenerators[$className])) {
100 return $this->_idGenerators[$className];
103 return parent::getIdGenerator($className);