vendor/doctrine/dbal/tests/Doctrine/Tests/DBAL/Schema/SchemaDiffTest.php

   1 <?php
   2
   3 namespace Doctrine\Tests\DBAL\Schema;
   4
   5 require_once __DIR__ . '/../../TestInit.php';
   6
   7 use Doctrine\DBAL\Schema\Schema,
   8     Doctrine\DBAL\Schema\Table,
   9     Doctrine\DBAL\Schema\Column,
  10     Doctrine\DBAL\Schema\Index,
  11     Doctrine\DBAL\Schema\Sequence,
  12     Doctrine\DBAL\Schema\SchemaDiff,
  13     Doctrine\DBAL\Schema\TableDiff,
  14     Doctrine\DBAL\Schema\Comparator,
  15     Doctrine\DBAL\Types\Type;
  16
  17 class SchemaDiffTest extends \PHPUnit_Framework_TestCase
  18 {
  19     public function testSchemaDiffToSql()
  20     {
  21         $diff = $this->createSchemaDiff();
  22         $platform = $this->createPlatform(true);
  23
  24         $sql = $diff->toSql($platform);
  25
  26         $expected = array('drop_orphan_fk', 'alter_seq', 'drop_seq', 'create_seq', 'create_table', 'create_foreign_key', 'drop_table', 'alter_table');
  27
  28         $this->assertEquals($expected, $sql);
  29     }
  30
  31     public function testSchemaDiffToSaveSql()
  32     {
  33         $diff = $this->createSchemaDiff();
  34         $platform = $this->createPlatform(false);
  35
  36         $sql = $diff->toSaveSql($platform);
  37
  38         $expected = array('alter_seq', 'create_seq', 'create_table', 'create_foreign_key', 'alter_table');
  39
  40         $this->assertEquals($expected, $sql);
  41     }
  42
  43     public function createPlatform($unsafe = false)
  44     {
  45         $platform = $this->getMock('Doctrine\Tests\DBAL\Mocks\MockPlatform');
  46         if ($unsafe) {
  47             $platform->expects($this->exactly(1))
  48                  ->method('getDropSequenceSql')
  49                  ->with($this->isInstanceOf('Doctrine\DBAL\Schema\Sequence'))
  50                  ->will($this->returnValue('drop_seq'));
  51         }
  52         $platform->expects($this->exactly(1))
  53                  ->method('getAlterSequenceSql')
  54                  ->with($this->isInstanceOf('Doctrine\DBAL\Schema\Sequence'))
  55                  ->will($this->returnValue('alter_seq'));
  56         $platform->expects($this->exactly(1))
  57                  ->method('getCreateSequenceSql')
  58                  ->with($this->isInstanceOf('Doctrine\DBAL\Schema\Sequence'))
  59                  ->will($this->returnValue('create_seq'));
  60         if ($unsafe) {
  61             $platform->expects($this->exactly(1))
  62                      ->method('getDropTableSql')
  63                      ->with($this->isInstanceof('Doctrine\DBAL\Schema\Table'))
  64                      ->will($this->returnValue('drop_table'));
  65         }
  66         $platform->expects($this->exactly(1))
  67                  ->method('getCreateTableSql')
  68                  ->with($this->isInstanceof('Doctrine\DBAL\Schema\Table'))
  69                  ->will($this->returnValue(array('create_table')));
  70         $platform->expects($this->exactly(1))
  71                  ->method('getCreateForeignKeySQL')
  72                  ->with($this->isInstanceOf('Doctrine\DBAL\Schema\ForeignKeyConstraint'))
  73                  ->will($this->returnValue('create_foreign_key'));
  74         $platform->expects($this->exactly(1))
  75                  ->method('getAlterTableSql')
  76                  ->with($this->isInstanceOf('Doctrine\DBAL\Schema\TableDiff'))
  77                  ->will($this->returnValue(array('alter_table')));
  78         if ($unsafe) {
  79             $platform->expects($this->exactly(1))
  80                      ->method('getDropForeignKeySql')
  81                      ->with($this->isInstanceof('Doctrine\DBAL\Schema\ForeignKeyConstraint'), $this->equalTo('local_table'))
  82                      ->will($this->returnValue('drop_orphan_fk'));
  83         }
  84         $platform->expects($this->exactly(1))
  85                 ->method('supportsSequences')
  86                 ->will($this->returnValue(true));
  87         $platform->expects($this->exactly(2))
  88                 ->method('supportsForeignKeyConstraints')
  89                 ->will($this->returnValue(true));
  90         return $platform;
  91     }
  92
  93     public function createSchemaDiff()
  94     {
  95         $diff = new SchemaDiff();
  96         $diff->changedSequences['foo_seq'] = new Sequence('foo_seq');
  97         $diff->newSequences['bar_seq'] = new Sequence('bar_seq');
  98         $diff->removedSequences['baz_seq'] = new Sequence('baz_seq');
  99         $diff->newTables['foo_table'] = new Table('foo_table');
 100         $diff->removedTables['bar_table'] = new Table('bar_table');
 101         $diff->changedTables['baz_table'] = new TableDiff('baz_table');
 102         $diff->newTables['foo_table']->addColumn('foreign_id', 'integer');
 103         $diff->newTables['foo_table']->addForeignKeyConstraint('foreign_table', array('foreign_id'), array('id'));
 104         $fk = new \Doctrine\DBAL\Schema\ForeignKeyConstraint(array('id'), 'foreign_table', array('id'));
 105         $fk->setLocalTable(new Table('local_table'));
 106         $diff->orphanedForeignKeys[] = $fk;
 107         return $diff;
 108     }
 109 }