SchemaDiffTest.php 4.79 KB
Newer Older
1 2 3 4
<?php

namespace Doctrine\Tests\DBAL\Schema;

Sergei Morozov's avatar
Sergei Morozov committed
5
use Doctrine\DBAL\Schema\ForeignKeyConstraint;
jeroendedauw's avatar
jeroendedauw committed
6 7 8 9
use Doctrine\DBAL\Schema\SchemaDiff;
use Doctrine\DBAL\Schema\Sequence;
use Doctrine\DBAL\Schema\Table;
use Doctrine\DBAL\Schema\TableDiff;
Sergei Morozov's avatar
Sergei Morozov committed
10
use Doctrine\Tests\DBAL\Mocks\MockPlatform;
Sergei Morozov's avatar
Sergei Morozov committed
11
use PHPUnit\Framework\TestCase;
12

Sergei Morozov's avatar
Sergei Morozov committed
13
class SchemaDiffTest extends TestCase
14 15 16
{
    public function testSchemaDiffToSql()
    {
Sergei Morozov's avatar
Sergei Morozov committed
17
        $diff     = $this->createSchemaDiff();
18
        $platform = $this->createPlatform(true);
19 20 21

        $sql = $diff->toSql($platform);

Sergei Morozov's avatar
Sergei Morozov committed
22
        $expected = ['create_schema', 'drop_orphan_fk', 'alter_seq', 'drop_seq', 'create_seq', 'create_table', 'create_foreign_key', 'drop_table', 'alter_table'];
23

24
        self::assertEquals($expected, $sql);
25 26 27 28
    }

    public function testSchemaDiffToSaveSql()
    {
Sergei Morozov's avatar
Sergei Morozov committed
29
        $diff     = $this->createSchemaDiff();
30
        $platform = $this->createPlatform(false);
31 32 33

        $sql = $diff->toSaveSql($platform);

Sergei Morozov's avatar
Sergei Morozov committed
34
        $expected = ['create_schema', 'alter_seq', 'create_seq', 'create_table', 'create_foreign_key', 'alter_table'];
35

36
        self::assertEquals($expected, $sql);
37 38
    }

39
    public function createPlatform($unsafe = false)
40
    {
Sergei Morozov's avatar
Sergei Morozov committed
41
        $platform = $this->createMock(MockPlatform::class);
42 43 44 45
        $platform->expects($this->exactly(1))
            ->method('getCreateSchemaSQL')
            ->with('foo_ns')
            ->will($this->returnValue('create_schema'));
46 47
        if ($unsafe) {
            $platform->expects($this->exactly(1))
48
                 ->method('getDropSequenceSql')
Sergei Morozov's avatar
Sergei Morozov committed
49
                 ->with($this->isInstanceOf(Sequence::class))
50
                 ->will($this->returnValue('drop_seq'));
51 52 53
        }
        $platform->expects($this->exactly(1))
                 ->method('getAlterSequenceSql')
Sergei Morozov's avatar
Sergei Morozov committed
54
                 ->with($this->isInstanceOf(Sequence::class))
55 56
                 ->will($this->returnValue('alter_seq'));
        $platform->expects($this->exactly(1))
57
                 ->method('getCreateSequenceSql')
Sergei Morozov's avatar
Sergei Morozov committed
58
                 ->with($this->isInstanceOf(Sequence::class))
59
                 ->will($this->returnValue('create_seq'));
60 61
        if ($unsafe) {
            $platform->expects($this->exactly(1))
62
                     ->method('getDropTableSql')
Sergei Morozov's avatar
Sergei Morozov committed
63
                     ->with($this->isInstanceOf(Table::class))
64 65 66 67
                     ->will($this->returnValue('drop_table'));
        }
        $platform->expects($this->exactly(1))
                 ->method('getCreateTableSql')
Sergei Morozov's avatar
Sergei Morozov committed
68
                 ->with($this->isInstanceOf(Table::class))
Sergei Morozov's avatar
Sergei Morozov committed
69
                 ->will($this->returnValue(['create_table']));
70 71
        $platform->expects($this->exactly(1))
                 ->method('getCreateForeignKeySQL')
Sergei Morozov's avatar
Sergei Morozov committed
72
                 ->with($this->isInstanceOf(ForeignKeyConstraint::class))
73
                 ->will($this->returnValue('create_foreign_key'));
74 75
        $platform->expects($this->exactly(1))
                 ->method('getAlterTableSql')
Sergei Morozov's avatar
Sergei Morozov committed
76
                 ->with($this->isInstanceOf(TableDiff::class))
Sergei Morozov's avatar
Sergei Morozov committed
77
                 ->will($this->returnValue(['alter_table']));
78 79
        if ($unsafe) {
            $platform->expects($this->exactly(1))
80
                     ->method('getDropForeignKeySql')
81
                     ->with(
Sergei Morozov's avatar
Sergei Morozov committed
82 83
                         $this->isInstanceOf(ForeignKeyConstraint::class),
                         $this->isInstanceOf(Table::class)
84
                     )
85 86
                     ->will($this->returnValue('drop_orphan_fk'));
        }
87 88 89
        $platform->expects($this->exactly(1))
                ->method('supportsSchemas')
                ->will($this->returnValue(true));
90 91 92
        $platform->expects($this->exactly(1))
                ->method('supportsSequences')
                ->will($this->returnValue(true));
93
        $platform->expects($this->exactly(2))
94 95 96 97 98 99 100
                ->method('supportsForeignKeyConstraints')
                ->will($this->returnValue(true));
        return $platform;
    }

    public function createSchemaDiff()
    {
Sergei Morozov's avatar
Sergei Morozov committed
101 102
        $diff                              = new SchemaDiff();
        $diff->newNamespaces['foo_ns']     = 'foo_ns';
103
        $diff->removedNamespaces['bar_ns'] = 'bar_ns';
104
        $diff->changedSequences['foo_seq'] = new Sequence('foo_seq');
Sergei Morozov's avatar
Sergei Morozov committed
105
        $diff->newSequences['bar_seq']     = new Sequence('bar_seq');
106
        $diff->removedSequences['baz_seq'] = new Sequence('baz_seq');
Sergei Morozov's avatar
Sergei Morozov committed
107 108 109
        $diff->newTables['foo_table']      = new Table('foo_table');
        $diff->removedTables['bar_table']  = new Table('bar_table');
        $diff->changedTables['baz_table']  = new TableDiff('baz_table');
110
        $diff->newTables['foo_table']->addColumn('foreign_id', 'integer');
Sergei Morozov's avatar
Sergei Morozov committed
111 112
        $diff->newTables['foo_table']->addForeignKeyConstraint('foreign_table', ['foreign_id'], ['id']);
        $fk = new ForeignKeyConstraint(['id'], 'foreign_table', ['id']);
113 114 115 116
        $fk->setLocalTable(new Table('local_table'));
        $diff->orphanedForeignKeys[] = $fk;
        return $diff;
    }
117
}