--- /dev/null
+<?php
+
+namespace Doctrine\Tests\ORM\Mapping;
+
+use Doctrine\Tests\Mocks\MetadataDriverMock;
+use Doctrine\Tests\Mocks\EntityManagerMock;
+use Doctrine\Tests\Mocks\ConnectionMock;
+use Doctrine\Tests\Mocks\DriverMock;
+use Doctrine\ORM\Mapping\ClassMetadata;
+use Doctrine\Common\EventManager;
+use Doctrine\ORM\Mapping\ClassMetadataFactory;
+
+require_once __DIR__ . '/../../TestInit.php';
+
+class ClassMetadataFactoryTest extends \Doctrine\Tests\OrmTestCase
+{
+ public function testGetMetadataForSingleClass()
+ {
+ $mockDriver = new MetadataDriverMock();
+ $entityManager = $this->_createEntityManager($mockDriver);
+
+ $conn = $entityManager->getConnection();
+ $mockPlatform = $conn->getDatabasePlatform();
+ $mockPlatform->setPrefersSequences(true);
+ $mockPlatform->setPrefersIdentityColumns(false);
+
+ $cm1 = $this->_createValidClassMetadata();
+
+ // SUT
+ $cmf = new \Doctrine\ORM\Mapping\ClassMetadataFactory();
+ $cmf->setEntityManager($entityManager);
+ $cmf->setMetadataFor($cm1->name, $cm1);
+
+ // Prechecks
+ $this->assertEquals(array(), $cm1->parentClasses);
+ $this->assertEquals(ClassMetadata::INHERITANCE_TYPE_NONE, $cm1->inheritanceType);
+ $this->assertTrue($cm1->hasField('name'));
+ $this->assertEquals(2, count($cm1->associationMappings));
+ $this->assertEquals(ClassMetadata::GENERATOR_TYPE_AUTO, $cm1->generatorType);
+ $this->assertEquals('group', $cm1->table['name']);
+
+ // Go
+ $cmMap1 = $cmf->getMetadataFor($cm1->name);
+
+ $this->assertSame($cm1, $cmMap1);
+ $this->assertEquals('group', $cmMap1->table['name']);
+ $this->assertTrue($cmMap1->table['quoted']);
+ $this->assertEquals(array(), $cmMap1->parentClasses);
+ $this->assertTrue($cmMap1->hasField('name'));
+ }
+
+ public function testGetMetadataFor_ReturnsLoadedCustomIdGenerator()
+ {
+ $cm1 = $this->_createValidClassMetadata();
+ $cm1->setIdGeneratorType(ClassMetadata::GENERATOR_TYPE_CUSTOM);
+ $cm1->customGeneratorDefinition = array(
+ "class" => "Doctrine\Tests\ORM\Mapping\CustomIdGenerator");
+ $cmf = $this->_createTestFactory();
+ $cmf->setMetadataForClass($cm1->name, $cm1);
+
+ $actual = $cmf->getMetadataFor($cm1->name);
+
+ $this->assertEquals(ClassMetadata::GENERATOR_TYPE_CUSTOM,
+ $actual->generatorType);
+ $this->assertInstanceOf("Doctrine\Tests\ORM\Mapping\CustomIdGenerator",
+ $actual->idGenerator);
+ }
+
+ public function testGetMetadataFor_ThrowsExceptionOnUnknownCustomGeneratorClass()
+ {
+ $cm1 = $this->_createValidClassMetadata();
+ $cm1->setIdGeneratorType(ClassMetadata::GENERATOR_TYPE_CUSTOM);
+ $cm1->customGeneratorDefinition = array("class" => "NotExistingGenerator");
+ $cmf = $this->_createTestFactory();
+ $cmf->setMetadataForClass($cm1->name, $cm1);
+ $this->setExpectedException("Doctrine\ORM\ORMException");
+
+ $actual = $cmf->getMetadataFor($cm1->name);
+ }
+
+ public function testGetMetadataFor_ThrowsExceptionOnMissingCustomGeneratorDefinition()
+ {
+ $cm1 = $this->_createValidClassMetadata();
+ $cm1->setIdGeneratorType(ClassMetadata::GENERATOR_TYPE_CUSTOM);
+ $cmf = $this->_createTestFactory();
+ $cmf->setMetadataForClass($cm1->name, $cm1);
+ $this->setExpectedException("Doctrine\ORM\ORMException");
+
+ $actual = $cmf->getMetadataFor($cm1->name);
+ }
+
+ public function testHasGetMetadata_NamespaceSeperatorIsNotNormalized()
+ {
+ require_once __DIR__."/../../Models/Global/GlobalNamespaceModel.php";
+
+ $metadataDriver = $this->createAnnotationDriver(array(__DIR__ . '/../../Models/Global/'));
+
+ $entityManager = $this->_createEntityManager($metadataDriver);
+
+ $mf = $entityManager->getMetadataFactory();
+ $m1 = $mf->getMetadataFor("DoctrineGlobal_Article");
+ $h1 = $mf->hasMetadataFor("DoctrineGlobal_Article");
+ $h2 = $mf->hasMetadataFor("\DoctrineGlobal_Article");
+ $m2 = $mf->getMetadataFor("\DoctrineGlobal_Article");
+
+ $this->assertNotSame($m1, $m2);
+ $this->assertFalse($h2);
+ $this->assertTrue($h1);
+ }
+
+ /**
+ * @group DDC-1512
+ */
+ public function testIsTransient()
+ {
+ $cmf = new ClassMetadataFactory();
+ $driver = $this->getMock('Doctrine\Common\Persistence\Mapping\Driver\MappingDriver');
+ $driver->expects($this->at(0))
+ ->method('isTransient')
+ ->with($this->equalTo('Doctrine\Tests\Models\CMS\CmsUser'))
+ ->will($this->returnValue(true));
+ $driver->expects($this->at(1))
+ ->method('isTransient')
+ ->with($this->equalTo('Doctrine\Tests\Models\CMS\CmsArticle'))
+ ->will($this->returnValue(false));
+
+ $em = $this->_createEntityManager($driver);
+
+ $this->assertTrue($em->getMetadataFactory()->isTransient('Doctrine\Tests\Models\CMS\CmsUser'));
+ $this->assertFalse($em->getMetadataFactory()->isTransient('Doctrine\Tests\Models\CMS\CmsArticle'));
+ }
+
+ /**
+ * @group DDC-1512
+ */
+ public function testIsTransientEntityNamespace()
+ {
+ $cmf = new ClassMetadataFactory();
+ $driver = $this->getMock('Doctrine\Common\Persistence\Mapping\Driver\MappingDriver');
+ $driver->expects($this->at(0))
+ ->method('isTransient')
+ ->with($this->equalTo('Doctrine\Tests\Models\CMS\CmsUser'))
+ ->will($this->returnValue(true));
+ $driver->expects($this->at(1))
+ ->method('isTransient')
+ ->with($this->equalTo('Doctrine\Tests\Models\CMS\CmsArticle'))
+ ->will($this->returnValue(false));
+
+ $em = $this->_createEntityManager($driver);
+ $em->getConfiguration()->addEntityNamespace('CMS', 'Doctrine\Tests\Models\CMS');
+
+ $this->assertTrue($em->getMetadataFactory()->isTransient('CMS:CmsUser'));
+ $this->assertFalse($em->getMetadataFactory()->isTransient('CMS:CmsArticle'));
+ }
+
+ public function testAddDefaultDiscriminatorMap()
+ {
+ $cmf = new ClassMetadataFactory();
+ $driver = $this->createAnnotationDriver(array(__DIR__ . '/../../Models/JoinedInheritanceType/'));
+ $em = $this->_createEntityManager($driver);
+ $cmf->setEntityManager($em);
+
+ $rootMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\JoinedInheritanceType\RootClass');
+ $childMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\JoinedInheritanceType\ChildClass');
+ $anotherChildMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\JoinedInheritanceType\AnotherChildClass');
+ $rootDiscriminatorMap = $rootMetadata->discriminatorMap;
+ $childDiscriminatorMap = $childMetadata->discriminatorMap;
+ $anotherChildDiscriminatorMap = $anotherChildMetadata->discriminatorMap;
+
+ $rootClass = 'Doctrine\Tests\Models\JoinedInheritanceType\RootClass';
+ $childClass = 'Doctrine\Tests\Models\JoinedInheritanceType\ChildClass';
+ $anotherChildClass = 'Doctrine\Tests\Models\JoinedInheritanceType\AnotherChildClass';
+
+ $rootClassKey = array_search($rootClass, $rootDiscriminatorMap);
+ $childClassKey = array_search($childClass, $rootDiscriminatorMap);
+ $anotherChildClassKey = array_search($anotherChildClass, $rootDiscriminatorMap);
+
+ $this->assertEquals('rootclass', $rootClassKey);
+ $this->assertEquals('childclass', $childClassKey);
+ $this->assertEquals('anotherchildclass', $anotherChildClassKey);
+
+ $this->assertEquals($childDiscriminatorMap, $rootDiscriminatorMap);
+ $this->assertEquals($anotherChildDiscriminatorMap, $rootDiscriminatorMap);
+
+ // ClassMetadataFactory::addDefaultDiscriminatorMap shouldn't be called again, because the
+ // discriminator map is already cached
+ $cmf = $this->getMock('Doctrine\ORM\Mapping\ClassMetadataFactory', array('addDefaultDiscriminatorMap'));
+ $cmf->setEntityManager($em);
+ $cmf->expects($this->never())
+ ->method('addDefaultDiscriminatorMap');
+
+ $rootMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\JoinedInheritanceType\RootClass');
+ }
+
+ protected function _createEntityManager($metadataDriver)
+ {
+ $driverMock = new DriverMock();
+ $config = new \Doctrine\ORM\Configuration();
+ $config->setProxyDir(__DIR__ . '/../../Proxies');
+ $config->setProxyNamespace('Doctrine\Tests\Proxies');
+ $eventManager = new EventManager();
+ $conn = new ConnectionMock(array(), $driverMock, $config, $eventManager);
+ $mockDriver = new MetadataDriverMock();
+ $config->setMetadataDriverImpl($metadataDriver);
+
+ return EntityManagerMock::create($conn, $config, $eventManager);
+ }
+
+ /**
+ * @return ClassMetadataFactoryTestSubject
+ */
+ protected function _createTestFactory()
+ {
+ $mockDriver = new MetadataDriverMock();
+ $entityManager = $this->_createEntityManager($mockDriver);
+ $cmf = new ClassMetadataFactoryTestSubject();
+ $cmf->setEntityManager($entityManager);
+ return $cmf;
+ }
+
+ /**
+ * @param string $class
+ * @return ClassMetadata
+ */
+ protected function _createValidClassMetadata()
+ {
+ // Self-made metadata
+ $cm1 = new ClassMetadata('Doctrine\Tests\ORM\Mapping\TestEntity1');
+ $cm1->initializeReflection(new \Doctrine\Common\Persistence\Mapping\RuntimeReflectionService);
+ $cm1->setPrimaryTable(array('name' => '`group`'));
+ // Add a mapped field
+ $cm1->mapField(array('fieldName' => 'name', 'type' => 'varchar'));
+ // Add a mapped field
+ $cm1->mapField(array('fieldName' => 'id', 'type' => 'integer', 'id' => true));
+ // and a mapped association
+ $cm1->mapOneToOne(array('fieldName' => 'other', 'targetEntity' => 'TestEntity1', 'mappedBy' => 'this'));
+ // and an association on the owning side
+ $joinColumns = array(
+ array('name' => 'other_id', 'referencedColumnName' => 'id')
+ );
+ $cm1->mapOneToOne(array('fieldName' => 'association', 'targetEntity' => 'TestEntity1', 'joinColumns' => $joinColumns));
+ // and an id generator type
+ $cm1->setIdGeneratorType(ClassMetadata::GENERATOR_TYPE_AUTO);
+ return $cm1;
+ }
+
+ /**
+ * @group DDC-1845
+ */
+ public function testQuoteMetadata()
+ {
+ $cmf = new ClassMetadataFactory();
+ $driver = $this->createAnnotationDriver(array(__DIR__ . '/../../Models/Quote/'));
+ $em = $this->_createEntityManager($driver);
+ $cmf->setEntityManager($em);
+
+
+ $userMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\Quote\User');
+ $phoneMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\Quote\Phone');
+ $groupMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\Quote\Group');
+ $addressMetadata = $cmf->getMetadataFor('Doctrine\Tests\Models\Quote\Address');
+
+
+ // Phone Class Metadata
+ $this->assertTrue($phoneMetadata->fieldMappings['number']['quoted']);
+ $this->assertEquals('phone-number', $phoneMetadata->fieldMappings['number']['columnName']);
+
+ $user = $phoneMetadata->associationMappings['user'];
+ $this->assertTrue($user['joinColumns'][0]['quoted']);
+ $this->assertEquals('user-id', $user['joinColumns'][0]['name']);
+ $this->assertEquals('user-id', $user['joinColumns'][0]['referencedColumnName']);
+
+
+
+ // User Group Metadata
+ $this->assertTrue($groupMetadata->fieldMappings['id']['quoted']);
+ $this->assertTrue($groupMetadata->fieldMappings['name']['quoted']);
+
+ $this->assertEquals('user-id', $userMetadata->fieldMappings['id']['columnName']);
+ $this->assertEquals('user-name', $userMetadata->fieldMappings['name']['columnName']);
+
+ $user = $groupMetadata->associationMappings['parent'];
+ $this->assertTrue($user['joinColumns'][0]['quoted']);
+ $this->assertEquals('parent-id', $user['joinColumns'][0]['name']);
+ $this->assertEquals('group-id', $user['joinColumns'][0]['referencedColumnName']);
+
+
+ // Address Class Metadata
+ $this->assertTrue($addressMetadata->fieldMappings['id']['quoted']);
+ $this->assertTrue($addressMetadata->fieldMappings['zip']['quoted']);
+
+ $this->assertEquals('address-id', $addressMetadata->fieldMappings['id']['columnName']);
+ $this->assertEquals('address-zip', $addressMetadata->fieldMappings['zip']['columnName']);
+
+ $user = $addressMetadata->associationMappings['user'];
+ $this->assertTrue($user['joinColumns'][0]['quoted']);
+ $this->assertEquals('user-id', $user['joinColumns'][0]['name']);
+ $this->assertEquals('user-id', $user['joinColumns'][0]['referencedColumnName']);
+
+
+
+ // User Class Metadata
+ $this->assertTrue($userMetadata->fieldMappings['id']['quoted']);
+ $this->assertTrue($userMetadata->fieldMappings['name']['quoted']);
+
+ $this->assertEquals('user-id', $userMetadata->fieldMappings['id']['columnName']);
+ $this->assertEquals('user-name', $userMetadata->fieldMappings['name']['columnName']);
+
+
+ $address = $userMetadata->associationMappings['address'];
+ $this->assertTrue($address['joinColumns'][0]['quoted']);
+ $this->assertEquals('address-id', $address['joinColumns'][0]['name']);
+ $this->assertEquals('address-id', $address['joinColumns'][0]['referencedColumnName']);
+
+ $groups = $userMetadata->associationMappings['groups'];
+ $this->assertTrue($groups['joinTable']['quoted']);
+ $this->assertTrue($groups['joinTable']['joinColumns'][0]['quoted']);
+ $this->assertEquals('quote-users-groups', $groups['joinTable']['name']);
+ $this->assertEquals('user-id', $groups['joinTable']['joinColumns'][0]['name']);
+ $this->assertEquals('user-id', $groups['joinTable']['joinColumns'][0]['referencedColumnName']);
+
+ $this->assertTrue($groups['joinTable']['inverseJoinColumns'][0]['quoted']);
+ $this->assertEquals('group-id', $groups['joinTable']['inverseJoinColumns'][0]['name']);
+ $this->assertEquals('group-id', $groups['joinTable']['inverseJoinColumns'][0]['referencedColumnName']);
+ }
+}
+
+/* Test subject class with overriden factory method for mocking purposes */
+class ClassMetadataFactoryTestSubject extends \Doctrine\ORM\Mapping\ClassMetadataFactory
+{
+ private $mockMetadata = array();
+ private $requestedClasses = array();
+
+ /** @override */
+ protected function newClassMetadataInstance($className)
+ {
+ $this->requestedClasses[] = $className;
+ if ( ! isset($this->mockMetadata[$className])) {
+ throw new InvalidArgumentException("No mock metadata found for class $className.");
+ }
+ return $this->mockMetadata[$className];
+ }
+
+ public function setMetadataForClass($className, $metadata)
+ {
+ $this->mockMetadata[$className] = $metadata;
+ }
+
+ public function getRequestedClasses()
+ {
+ return $this->requestedClasses;
+ }
+}
+
+class TestEntity1
+{
+ private $id;
+ private $name;
+ private $other;
+ private $association;
+}
+
+class CustomIdGenerator extends \Doctrine\ORM\Id\AbstractIdGenerator
+{
+ public function generate(\Doctrine\ORM\EntityManager $em, $entity)
+ {
+ }
+}