2014-01-29 18:48:12 +04:00
|
|
|
<?php
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Copyright (c) 2014 Robin Appelman <icewind@owncloud.com>
|
|
|
|
* This file is licensed under the Affero General Public License version 3 or
|
|
|
|
* later.
|
|
|
|
* See the COPYING-README file.
|
|
|
|
*/
|
|
|
|
|
|
|
|
namespace Test\DB;
|
|
|
|
|
2019-11-22 22:52:10 +03:00
|
|
|
use Doctrine\DBAL\DBALException;
|
2014-10-23 16:32:37 +04:00
|
|
|
use Doctrine\DBAL\Platforms\OraclePlatform;
|
2019-11-22 22:52:10 +03:00
|
|
|
use Doctrine\DBAL\Schema\Schema;
|
|
|
|
use Doctrine\DBAL\Schema\SchemaConfig;
|
2015-07-29 19:19:31 +03:00
|
|
|
use OCP\IConfig;
|
2014-01-29 18:48:12 +04:00
|
|
|
|
2015-11-03 03:52:41 +03:00
|
|
|
/**
|
2016-05-20 16:38:20 +03:00
|
|
|
* Class MigratorTest
|
2015-11-03 03:52:41 +03:00
|
|
|
*
|
|
|
|
* @group DB
|
|
|
|
*
|
|
|
|
* @package Test\DB
|
|
|
|
*/
|
2016-05-20 16:38:20 +03:00
|
|
|
class MigratorTest extends \Test\TestCase {
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* @var \Doctrine\DBAL\Connection $connection
|
|
|
|
*/
|
|
|
|
private $connection;
|
|
|
|
|
2014-07-01 14:54:35 +04:00
|
|
|
/**
|
|
|
|
* @var \OC\DB\MDB2SchemaManager
|
|
|
|
*/
|
|
|
|
private $manager;
|
|
|
|
|
2015-05-18 17:57:59 +03:00
|
|
|
/**
|
|
|
|
* @var IConfig
|
|
|
|
**/
|
|
|
|
private $config;
|
|
|
|
|
2015-07-29 19:19:31 +03:00
|
|
|
/** @var string */
|
2014-01-29 18:48:12 +04:00
|
|
|
private $tableName;
|
|
|
|
|
2017-10-02 13:32:21 +03:00
|
|
|
/** @var string */
|
|
|
|
private $tableNameTmp;
|
|
|
|
|
2019-11-21 18:40:38 +03:00
|
|
|
protected function setUp(): void {
|
2014-11-11 01:30:38 +03:00
|
|
|
parent::setUp();
|
|
|
|
|
2015-05-18 17:57:59 +03:00
|
|
|
$this->config = \OC::$server->getConfig();
|
2016-01-07 12:26:00 +03:00
|
|
|
$this->connection = \OC::$server->getDatabaseConnection();
|
2014-10-23 16:32:37 +04:00
|
|
|
if ($this->connection->getDatabasePlatform() instanceof OraclePlatform) {
|
|
|
|
$this->markTestSkipped('DB migration tests are not supported on OCI');
|
|
|
|
}
|
2014-07-01 14:54:35 +04:00
|
|
|
$this->manager = new \OC\DB\MDB2SchemaManager($this->connection);
|
2017-10-02 13:32:21 +03:00
|
|
|
$this->tableName = $this->getUniqueTableName();
|
|
|
|
$this->tableNameTmp = $this->getUniqueTableName();
|
|
|
|
}
|
|
|
|
|
|
|
|
private function getUniqueTableName() {
|
|
|
|
return strtolower($this->getUniqueID($this->config->getSystemValue('dbtableprefix', 'oc_') . 'test_'));
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
|
|
|
|
2019-11-21 18:40:38 +03:00
|
|
|
protected function tearDown(): void {
|
2017-10-02 13:32:21 +03:00
|
|
|
// Try to delete if exists (IF EXISTS NOT SUPPORTED IN ORACLE)
|
|
|
|
try {
|
|
|
|
$this->connection->exec('DROP TABLE ' . $this->connection->quoteIdentifier($this->tableNameTmp));
|
|
|
|
} catch (\Doctrine\DBAL\DBALException $e) {}
|
|
|
|
|
|
|
|
try {
|
|
|
|
$this->connection->exec('DROP TABLE ' . $this->connection->quoteIdentifier($this->tableName));
|
|
|
|
} catch (\Doctrine\DBAL\DBALException $e) {}
|
2014-11-11 01:30:38 +03:00
|
|
|
parent::tearDown();
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
/**
|
|
|
|
* @return \Doctrine\DBAL\Schema\Schema[]
|
|
|
|
*/
|
|
|
|
private function getDuplicateKeySchemas() {
|
|
|
|
$startSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $startSchema->createTable($this->tableName);
|
2014-01-29 18:48:12 +04:00
|
|
|
$table->addColumn('id', 'integer');
|
|
|
|
$table->addColumn('name', 'string');
|
2014-01-30 17:21:16 +04:00
|
|
|
$table->addIndex(array('id'), $this->tableName . '_id');
|
2014-01-29 18:48:12 +04:00
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
$endSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $endSchema->createTable($this->tableName);
|
2014-01-29 18:48:12 +04:00
|
|
|
$table->addColumn('id', 'integer');
|
|
|
|
$table->addColumn('name', 'string');
|
2014-01-30 17:21:16 +04:00
|
|
|
$table->addUniqueIndex(array('id'), $this->tableName . '_id');
|
2014-03-27 17:44:19 +04:00
|
|
|
|
|
|
|
return array($startSchema, $endSchema);
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
private function getSchemaConfig() {
|
|
|
|
$config = new SchemaConfig();
|
|
|
|
$config->setName($this->connection->getDatabase());
|
|
|
|
return $config;
|
|
|
|
}
|
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
private function isSQLite() {
|
|
|
|
return $this->connection->getDriver() instanceof \Doctrine\DBAL\Driver\PDOSqlite\Driver;
|
|
|
|
}
|
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* @expectedException \OC\DB\MigrationException
|
|
|
|
*/
|
|
|
|
public function testDuplicateKeyUpgrade() {
|
2014-03-27 17:44:19 +04:00
|
|
|
if ($this->isSQLite()) {
|
2014-10-23 16:32:37 +04:00
|
|
|
$this->markTestSkipped('sqlite does not throw errors when creating a new key on existing data');
|
2014-03-27 17:44:19 +04:00
|
|
|
}
|
|
|
|
list($startSchema, $endSchema) = $this->getDuplicateKeySchemas();
|
2014-07-01 14:54:35 +04:00
|
|
|
$migrator = $this->manager->getMigrator();
|
2014-03-27 17:44:19 +04:00
|
|
|
$migrator->migrate($startSchema);
|
2014-01-29 18:48:12 +04:00
|
|
|
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 1, 'name' => 'foo'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'bar'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'qwerty'));
|
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
$migrator->checkMigrate($endSchema);
|
2014-01-29 18:48:12 +04:00
|
|
|
$this->fail('checkMigrate should have failed');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUpgrade() {
|
2014-03-27 17:44:19 +04:00
|
|
|
list($startSchema, $endSchema) = $this->getDuplicateKeySchemas();
|
2014-07-01 14:54:35 +04:00
|
|
|
$migrator = $this->manager->getMigrator();
|
2014-03-27 17:44:19 +04:00
|
|
|
$migrator->migrate($startSchema);
|
2014-01-29 18:48:12 +04:00
|
|
|
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 1, 'name' => 'foo'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'bar'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 3, 'name' => 'qwerty'));
|
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
$migrator->checkMigrate($endSchema);
|
|
|
|
$migrator->migrate($endSchema);
|
2018-01-25 13:23:12 +03:00
|
|
|
$this->addToAssertionCount(1);
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
2014-03-27 17:44:19 +04:00
|
|
|
|
2015-05-18 17:57:59 +03:00
|
|
|
public function testUpgradeDifferentPrefix() {
|
|
|
|
$oldTablePrefix = $this->config->getSystemValue('dbtableprefix', 'oc_');
|
|
|
|
|
|
|
|
$this->config->setSystemValue('dbtableprefix', 'ownc_');
|
|
|
|
$this->tableName = strtolower($this->getUniqueID($this->config->getSystemValue('dbtableprefix') . 'test_'));
|
|
|
|
|
|
|
|
list($startSchema, $endSchema) = $this->getDuplicateKeySchemas();
|
|
|
|
$migrator = $this->manager->getMigrator();
|
|
|
|
$migrator->migrate($startSchema);
|
|
|
|
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 1, 'name' => 'foo'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'bar'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 3, 'name' => 'qwerty'));
|
|
|
|
|
|
|
|
$migrator->checkMigrate($endSchema);
|
|
|
|
$migrator->migrate($endSchema);
|
2018-01-25 13:23:12 +03:00
|
|
|
$this->addToAssertionCount(1);
|
2015-05-18 17:57:59 +03:00
|
|
|
|
|
|
|
$this->config->setSystemValue('dbtableprefix', $oldTablePrefix);
|
|
|
|
}
|
|
|
|
|
2014-03-27 17:44:19 +04:00
|
|
|
public function testInsertAfterUpgrade() {
|
|
|
|
list($startSchema, $endSchema) = $this->getDuplicateKeySchemas();
|
2014-07-01 14:54:35 +04:00
|
|
|
$migrator = $this->manager->getMigrator();
|
2014-03-27 17:44:19 +04:00
|
|
|
$migrator->migrate($startSchema);
|
|
|
|
|
|
|
|
$migrator->migrate($endSchema);
|
|
|
|
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 1, 'name' => 'foo'));
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'bar'));
|
|
|
|
try {
|
|
|
|
$this->connection->insert($this->tableName, array('id' => 2, 'name' => 'qwerty'));
|
|
|
|
$this->fail('Expected duplicate key insert to fail');
|
|
|
|
} catch (DBALException $e) {
|
2018-01-25 13:23:12 +03:00
|
|
|
$this->addToAssertionCount(1);
|
2014-03-27 17:44:19 +04:00
|
|
|
}
|
|
|
|
}
|
2014-06-17 17:07:36 +04:00
|
|
|
|
|
|
|
public function testAddingPrimaryKeyWithAutoIncrement() {
|
|
|
|
$startSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $startSchema->createTable($this->tableName);
|
|
|
|
$table->addColumn('id', 'integer');
|
|
|
|
$table->addColumn('name', 'string');
|
|
|
|
|
|
|
|
$endSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $endSchema->createTable($this->tableName);
|
|
|
|
$table->addColumn('id', 'integer', array('autoincrement' => true));
|
|
|
|
$table->addColumn('name', 'string');
|
|
|
|
$table->setPrimaryKey(array('id'));
|
|
|
|
|
2014-07-01 14:54:35 +04:00
|
|
|
$migrator = $this->manager->getMigrator();
|
2014-06-17 17:07:36 +04:00
|
|
|
$migrator->migrate($startSchema);
|
|
|
|
|
|
|
|
$migrator->checkMigrate($endSchema);
|
|
|
|
$migrator->migrate($endSchema);
|
|
|
|
|
2018-01-25 13:23:12 +03:00
|
|
|
$this->addToAssertionCount(1);
|
2014-06-17 17:07:36 +04:00
|
|
|
}
|
2014-07-01 14:55:36 +04:00
|
|
|
|
|
|
|
public function testReservedKeywords() {
|
|
|
|
$startSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $startSchema->createTable($this->tableName);
|
|
|
|
$table->addColumn('id', 'integer', array('autoincrement' => true));
|
|
|
|
$table->addColumn('user', 'string', array('length' => 255));
|
|
|
|
$table->setPrimaryKey(array('id'));
|
|
|
|
|
|
|
|
$endSchema = new Schema(array(), array(), $this->getSchemaConfig());
|
|
|
|
$table = $endSchema->createTable($this->tableName);
|
|
|
|
$table->addColumn('id', 'integer', array('autoincrement' => true));
|
|
|
|
$table->addColumn('user', 'string', array('length' => 64));
|
|
|
|
$table->setPrimaryKey(array('id'));
|
|
|
|
|
|
|
|
$migrator = $this->manager->getMigrator();
|
|
|
|
$migrator->migrate($startSchema);
|
|
|
|
|
|
|
|
$migrator->checkMigrate($endSchema);
|
|
|
|
$migrator->migrate($endSchema);
|
|
|
|
|
2018-01-25 13:23:12 +03:00
|
|
|
$this->addToAssertionCount(1);
|
2014-07-01 14:55:36 +04:00
|
|
|
}
|
2017-10-02 13:32:21 +03:00
|
|
|
|
|
|
|
public function testAddingForeignKey() {
|
|
|
|
$startSchema = new Schema([], [], $this->getSchemaConfig());
|
|
|
|
$table = $startSchema->createTable($this->tableName);
|
|
|
|
$table->addColumn('id', 'integer', ['autoincrement' => true]);
|
|
|
|
$table->addColumn('name', 'string');
|
|
|
|
$table->setPrimaryKey(['id']);
|
|
|
|
|
|
|
|
$fkName = "fkc";
|
|
|
|
$tableFk = $startSchema->createTable($this->tableNameTmp);
|
|
|
|
$tableFk->addColumn('fk_id', 'integer');
|
|
|
|
$tableFk->addColumn('name', 'string');
|
|
|
|
$tableFk->addForeignKeyConstraint($this->tableName, array('fk_id'), array('id'), array(), $fkName);
|
|
|
|
|
|
|
|
$migrator = $this->manager->getMigrator();
|
|
|
|
$migrator->migrate($startSchema);
|
|
|
|
|
|
|
|
|
|
|
|
$this->assertTrue($startSchema->getTable($this->tableNameTmp)->hasForeignKey($fkName));
|
|
|
|
}
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|