vendor/doctrine/orm/tests/Doctrine/Tests/ORM/Functional/SchemaTool/PostgreSqlSchemaToolTest.php

   1 <?php
   2
   3 namespace Doctrine\Tests\ORM\Functional\SchemaTool;
   4
   5 use Doctrine\ORM\Tools\SchemaTool,
   6     Doctrine\ORM\Mapping\ClassMetadata;
   7
   8 require_once __DIR__ . '/../../../TestInit.php';
   9
  10 class PostgreSqlSchemaToolTest extends \Doctrine\Tests\OrmFunctionalTestCase
  11 {
  12     protected function setUp() {
  13         parent::setUp();
  14         if ($this->_em->getConnection()->getDatabasePlatform()->getName() !== 'postgresql') {
  15             $this->markTestSkipped('The ' . __CLASS__ .' requires the use of postgresql.');
  16         }
  17     }
  18
  19     public function testPostgresMetadataSequenceIncrementedBy10()
  20     {
  21         $address = $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsAddress');
  22         $this->assertEquals(1, $address->sequenceGeneratorDefinition['allocationSize']);
  23     }
  24
  25     public function testGetCreateSchemaSql()
  26     {
  27         $classes = array(
  28             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsAddress'),
  29             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsUser'),
  30             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsPhonenumber'),
  31         );
  32
  33         $tool = new SchemaTool($this->_em);
  34         $sql = $tool->getCreateSchemaSql($classes);
  35         $sqlCount = count($sql);
  36
  37         $this->assertEquals("CREATE TABLE cms_addresses (id INT NOT NULL, user_id INT DEFAULT NULL, country VARCHAR(50) NOT NULL, zip VARCHAR(50) NOT NULL, city VARCHAR(50) NOT NULL, PRIMARY KEY(id))", array_shift($sql));
  38         $this->assertEquals("CREATE UNIQUE INDEX UNIQ_ACAC157BA76ED395 ON cms_addresses (user_id)", array_shift($sql));
  39         $this->assertEquals("CREATE TABLE cms_users (id INT NOT NULL, email_id INT DEFAULT NULL, status VARCHAR(50) DEFAULT NULL, username VARCHAR(255) NOT NULL, name VARCHAR(255) NOT NULL, PRIMARY KEY(id))", array_shift($sql));
  40         $this->assertEquals("CREATE UNIQUE INDEX UNIQ_3AF03EC5F85E0677 ON cms_users (username)", array_shift($sql));
  41         $this->assertEquals("CREATE UNIQUE INDEX UNIQ_3AF03EC5A832C1C9 ON cms_users (email_id)", array_shift($sql));
  42         $this->assertEquals("CREATE TABLE cms_users_groups (user_id INT NOT NULL, group_id INT NOT NULL, PRIMARY KEY(user_id, group_id))", array_shift($sql));
  43         $this->assertEquals("CREATE INDEX IDX_7EA9409AA76ED395 ON cms_users_groups (user_id)", array_shift($sql));
  44         $this->assertEquals("CREATE INDEX IDX_7EA9409AFE54D947 ON cms_users_groups (group_id)", array_shift($sql));
  45         $this->assertEquals("CREATE TABLE cms_phonenumbers (phonenumber VARCHAR(50) NOT NULL, user_id INT DEFAULT NULL, PRIMARY KEY(phonenumber))", array_shift($sql));
  46         $this->assertEquals("CREATE INDEX IDX_F21F790FA76ED395 ON cms_phonenumbers (user_id)", array_shift($sql));
  47         $this->assertEquals("CREATE SEQUENCE cms_addresses_id_seq INCREMENT BY 1 MINVALUE 1 START 1", array_shift($sql));
  48         $this->assertEquals("CREATE SEQUENCE cms_users_id_seq INCREMENT BY 1 MINVALUE 1 START 1", array_shift($sql));
  49         $this->assertEquals("ALTER TABLE cms_addresses ADD CONSTRAINT FK_ACAC157BA76ED395 FOREIGN KEY (user_id) REFERENCES cms_users (id) NOT DEFERRABLE INITIALLY IMMEDIATE", array_shift($sql));
  50         $this->assertEquals("ALTER TABLE cms_users ADD CONSTRAINT FK_3AF03EC5A832C1C9 FOREIGN KEY (email_id) REFERENCES cms_emails (id) NOT DEFERRABLE INITIALLY IMMEDIATE", array_shift($sql));
  51         $this->assertEquals("ALTER TABLE cms_users_groups ADD CONSTRAINT FK_7EA9409AA76ED395 FOREIGN KEY (user_id) REFERENCES cms_users (id) NOT DEFERRABLE INITIALLY IMMEDIATE", array_shift($sql));
  52         $this->assertEquals("ALTER TABLE cms_users_groups ADD CONSTRAINT FK_7EA9409AFE54D947 FOREIGN KEY (group_id) REFERENCES cms_groups (id) NOT DEFERRABLE INITIALLY IMMEDIATE", array_shift($sql));
  53         $this->assertEquals("ALTER TABLE cms_phonenumbers ADD CONSTRAINT FK_F21F790FA76ED395 FOREIGN KEY (user_id) REFERENCES cms_users (id) NOT DEFERRABLE INITIALLY IMMEDIATE", array_shift($sql));
  54
  55         $this->assertEquals(array(), $sql, "SQL Array should be empty now.");
  56         $this->assertEquals(17, $sqlCount, "Total of 17 queries should be executed");
  57     }
  58
  59     public function testGetCreateSchemaSql2()
  60     {
  61         $classes = array(
  62             $this->_em->getClassMetadata('Doctrine\Tests\Models\Generic\DecimalModel')
  63         );
  64
  65         $tool = new SchemaTool($this->_em);
  66         $sql = $tool->getCreateSchemaSql($classes);
  67
  68         $this->assertEquals(2, count($sql));
  69
  70         $this->assertEquals('CREATE TABLE decimal_model (id INT NOT NULL, "decimal" NUMERIC(5, 2) NOT NULL, "high_scale" NUMERIC(14, 4) NOT NULL, PRIMARY KEY(id))', $sql[0]);
  71         $this->assertEquals("CREATE SEQUENCE decimal_model_id_seq INCREMENT BY 1 MINVALUE 1 START 1", $sql[1]);
  72     }
  73
  74     public function testGetCreateSchemaSql3()
  75     {
  76         $classes = array(
  77             $this->_em->getClassMetadata('Doctrine\Tests\Models\Generic\BooleanModel')
  78         );
  79
  80         $tool = new SchemaTool($this->_em);
  81         $sql = $tool->getCreateSchemaSql($classes);
  82
  83         $this->assertEquals(2, count($sql));
  84         $this->assertEquals("CREATE TABLE boolean_model (id INT NOT NULL, booleanField BOOLEAN NOT NULL, PRIMARY KEY(id))", $sql[0]);
  85         $this->assertEquals("CREATE SEQUENCE boolean_model_id_seq INCREMENT BY 1 MINVALUE 1 START 1", $sql[1]);
  86     }
  87
  88     public function testGetDropSchemaSql()
  89     {
  90         $classes = array(
  91             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsAddress'),
  92             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsUser'),
  93             $this->_em->getClassMetadata('Doctrine\Tests\Models\CMS\CmsPhonenumber'),
  94         );
  95
  96         $tool = new SchemaTool($this->_em);
  97         $sql = $tool->getDropSchemaSQL($classes);
  98
  99         $this->assertEquals(14, count($sql));
 100         $dropSequenceSQLs = 0;
 101         foreach ($sql AS $stmt) {
 102             if (strpos($stmt, "DROP SEQUENCE") === 0) {
 103                 $dropSequenceSQLs++;
 104             }
 105         }
 106         $this->assertEquals(4, $dropSequenceSQLs, "Expect 4 sequences to be dropped.");
 107     }
 108
 109     /**
 110      * @group DDC-1657
 111      */
 112     public function testUpdateSchemaWithPostgreSQLSchema()
 113     {
 114         $classes = array(
 115             $this->_em->getClassMetadata(__NAMESPACE__ . '\\DDC1657Screen'),
 116             $this->_em->getClassMetadata(__NAMESPACE__ . '\\DDC1657Avatar'),
 117         );
 118         try {
 119             $this->_em->getConnection()->exec("CREATE SCHEMA stonewood");
 120         } catch(\Exception $e) {
 121         }
 122
 123         $tool = new SchemaTool($this->_em);
 124         $tool->createSchema($classes);
 125
 126         $sql = $tool->getUpdateSchemaSql($classes);
 127         $sql = array_filter($sql, function($sql) { return (strpos($sql, "DROP SEQUENCE stonewood.") === 0); });
 128
 129         $this->assertCount(0, $sql, implode("\n", $sql));
 130     }
 131 }
 132
 133 /**
 134  * @Entity
 135  * @Table(name="stonewood.screen")
 136  */
 137 class DDC1657Screen
 138 {
 139     /**
 140      * Identifier
 141      * @var integer
 142      *
 143      * @Id
 144      * @GeneratedValue(strategy="IDENTITY")
 145      * @Column(name="pk", type="integer", nullable=false)
 146      */
 147     private $pk;
 148
 149     /**
 150      * Title
 151      * @var string
 152      *
 153      * @Column(name="title", type="string", length=255, nullable=false)
 154      */
 155     private $title;
 156
 157     /**
 158      * Path
 159      * @var string
 160      *
 161      * @Column(name="path", type="string", length=255, nullable=false)
 162      */
 163     private $path;
 164
 165     /**
 166      * Register date
 167      * @var Date
 168      *
 169      * @Column(name="ddate", type="date", nullable=false)
 170      */
 171     private $ddate;
 172
 173     /**
 174      * Avatar
 175      * @var Stonewood\Model\Entity\Avatar
 176      *
 177      * @ManyToOne(targetEntity="DDC1657Avatar")
 178      * @JoinColumn(name="pk_avatar", referencedColumnName="pk", nullable=true, onDelete="CASCADE")
 179      */
 180     private $avatar;
 181 }
 182
 183 /**
 184  * @Entity
 185  * @Table(name="stonewood.avatar")
 186  */
 187 class DDC1657Avatar
 188 {
 189     /**
 190      * Identifier
 191      * @var integer
 192      *
 193      * @Id
 194      * @GeneratedValue(strategy="IDENTITY")
 195      * @Column(name="pk", type="integer", nullable=false)
 196      */
 197     private $pk;
 198 }