vendor/doctrine/orm/tests/Doctrine/Tests/ORM/Functional/DatabaseDriverTest.php

   1 <?php
   2
   3 namespace Doctrine\Tests\ORM\Functional;
   4
   5 require_once __DIR__ . '/../../TestInit.php';
   6
   7 use Doctrine\ORM\Mapping\ClassMetadataInfo,
   8     Doctrine\Common\Util\Inflector;
   9
  10 class DatabaseDriverTest extends \Doctrine\Tests\OrmFunctionalTestCase
  11 {
  12     /**
  13      * @var \Doctrine\DBAL\Schema\AbstractSchemaManager
  14      */
  15     protected $_sm = null;
  16
  17     public function setUp()
  18     {
  19         $this->useModelSet('cms');
  20         parent::setUp();
  21
  22         $this->_sm = $this->_em->getConnection()->getSchemaManager();
  23     }
  24
  25     public function testLoadMetadataFromDatabase()
  26     {
  27         if (!$this->_em->getConnection()->getDatabasePlatform()->supportsForeignKeyConstraints()) {
  28             $this->markTestSkipped('Platform does not support foreign keys.');
  29         }
  30
  31         $table = new \Doctrine\DBAL\Schema\Table("dbdriver_foo");
  32         $table->addColumn('id', 'integer');
  33         $table->setPrimaryKey(array('id'));
  34         $table->addColumn('bar', 'string', array('notnull' => false, 'length' => 200));
  35
  36         $this->_sm->dropAndCreateTable($table);
  37
  38         $metadatas = $this->extractClassMetadata(array("DbdriverFoo"));
  39
  40         $this->assertArrayHasKey('DbdriverFoo', $metadatas);
  41         $metadata = $metadatas['DbdriverFoo'];
  42
  43         $this->assertArrayHasKey('id',          $metadata->fieldMappings);
  44         $this->assertEquals('id',               $metadata->fieldMappings['id']['fieldName']);
  45         $this->assertEquals('id',               strtolower($metadata->fieldMappings['id']['columnName']));
  46         $this->assertEquals('integer',          (string)$metadata->fieldMappings['id']['type']);
  47
  48         $this->assertArrayHasKey('bar',         $metadata->fieldMappings);
  49         $this->assertEquals('bar',              $metadata->fieldMappings['bar']['fieldName']);
  50         $this->assertEquals('bar',              strtolower($metadata->fieldMappings['bar']['columnName']));
  51         $this->assertEquals('string',           (string)$metadata->fieldMappings['bar']['type']);
  52         $this->assertEquals(200,                $metadata->fieldMappings['bar']['length']);
  53         $this->assertTrue($metadata->fieldMappings['bar']['nullable']);
  54     }
  55
  56     public function testLoadMetadataWithForeignKeyFromDatabase()
  57     {
  58         if (!$this->_em->getConnection()->getDatabasePlatform()->supportsForeignKeyConstraints()) {
  59             $this->markTestSkipped('Platform does not support foreign keys.');
  60         }
  61
  62         $tableB = new \Doctrine\DBAL\Schema\Table("dbdriver_bar");
  63         $tableB->addColumn('id', 'integer');
  64         $tableB->setPrimaryKey(array('id'));
  65
  66         $this->_sm->dropAndCreateTable($tableB);
  67
  68         $tableA = new \Doctrine\DBAL\Schema\Table("dbdriver_baz");
  69         $tableA->addColumn('id', 'integer');
  70         $tableA->setPrimaryKey(array('id'));
  71         $tableA->addColumn('bar_id', 'integer');
  72         $tableA->addForeignKeyConstraint('dbdriver_bar', array('bar_id'), array('id'));
  73
  74         $this->_sm->dropAndCreateTable($tableA);
  75
  76         $metadatas = $this->extractClassMetadata(array("DbdriverBar", "DbdriverBaz"));
  77
  78         $this->assertArrayHasKey('DbdriverBaz', $metadatas);
  79         $bazMetadata = $metadatas['DbdriverBaz'];
  80
  81         $this->assertArrayNotHasKey('barId', $bazMetadata->fieldMappings, "The foreign Key field should not be inflected as 'barId' field, its an association.");
  82         $this->assertArrayHasKey('id', $bazMetadata->fieldMappings);
  83
  84         $bazMetadata->associationMappings = \array_change_key_case($bazMetadata->associationMappings, \CASE_LOWER);
  85
  86         $this->assertArrayHasKey('bar', $bazMetadata->associationMappings);
  87         $this->assertEquals(ClassMetadataInfo::MANY_TO_ONE, $bazMetadata->associationMappings['bar']['type']);
  88     }
  89
  90     public function testDetectManyToManyTables()
  91     {
  92         if (!$this->_em->getConnection()->getDatabasePlatform()->supportsForeignKeyConstraints()) {
  93             $this->markTestSkipped('Platform does not support foreign keys.');
  94         }
  95
  96         $metadatas = $this->extractClassMetadata(array("CmsUsers", "CmsGroups"));
  97
  98         $this->assertArrayHasKey('CmsUsers', $metadatas, 'CmsUsers entity was not detected.');
  99         $this->assertArrayHasKey('CmsGroups', $metadatas, 'CmsGroups entity was not detected.');
 100
 101         $this->assertEquals(2, count($metadatas['CmsUsers']->associationMappings));
 102         $this->assertArrayHasKey('group', $metadatas['CmsUsers']->associationMappings);
 103         $this->assertEquals(1, count($metadatas['CmsGroups']->associationMappings));
 104         $this->assertArrayHasKey('user', $metadatas['CmsGroups']->associationMappings);
 105     }
 106
 107     public function testIgnoreManyToManyTableWithoutFurtherForeignKeyDetails()
 108     {
 109         $tableB = new \Doctrine\DBAL\Schema\Table("dbdriver_bar");
 110         $tableB->addColumn('id', 'integer');
 111         $tableB->setPrimaryKey(array('id'));
 112
 113         $tableA = new \Doctrine\DBAL\Schema\Table("dbdriver_baz");
 114         $tableA->addColumn('id', 'integer');
 115         $tableA->setPrimaryKey(array('id'));
 116
 117         $tableMany = new \Doctrine\DBAL\Schema\Table("dbdriver_bar_baz");
 118         $tableMany->addColumn('bar_id', 'integer');
 119         $tableMany->addColumn('baz_id', 'integer');
 120         $tableMany->addForeignKeyConstraint('dbdriver_bar', array('bar_id'), array('id'));
 121
 122         $metadatas = $this->convertToClassMetadata(array($tableA, $tableB), array($tableMany));
 123
 124         $this->assertEquals(0, count($metadatas['DbdriverBaz']->associationMappings), "no association mappings should be detected.");
 125     }
 126
 127     protected function convertToClassMetadata(array $entityTables, array $manyTables = array())
 128     {
 129         $driver = new \Doctrine\ORM\Mapping\Driver\DatabaseDriver($this->_sm);
 130         $driver->setTables($entityTables, $manyTables);
 131
 132         $metadatas = array();
 133         foreach ($driver->getAllClassNames() AS $className) {
 134             $class = new ClassMetadataInfo($className);
 135             $driver->loadMetadataForClass($className, $class);
 136             $metadatas[$className] = $class;
 137         }
 138
 139         return $metadatas;
 140     }
 141
 142     /**
 143      * @param  string $className
 144      * @return ClassMetadata
 145      */
 146     protected function extractClassMetadata(array $classNames)
 147     {
 148         $classNames = array_map('strtolower', $classNames);
 149         $metadatas = array();
 150
 151         $driver = new \Doctrine\ORM\Mapping\Driver\DatabaseDriver($this->_sm);
 152         foreach ($driver->getAllClassNames() as $className) {
 153             if (!in_array(strtolower($className), $classNames)) {
 154                 continue;
 155             }
 156             $class = new ClassMetadataInfo($className);
 157             $driver->loadMetadataForClass($className, $class);
 158             $metadatas[$className] = $class;
 159         }
 160
 161         if (count($metadatas) != count($classNames)) {
 162             $this->fail("Have not found all classes matching the names '" . implode(", ", $classNames) . "' only tables " . implode(", ", array_keys($metadatas)));
 163         }
 164         return $metadatas;
 165     }
 166 }