Mirrors/yii
2
0
Fork 0
mirror of https://github.com/yiisoft/yii.git synced 2026-03-04 15:24:07 +01:00
Code Issues Packages Projects Releases Wiki Activity
Files
2a6eee6bd47dbfe5bf31bf8a2895c56fc1663edc
yii/tests/framework/db/schema/CPostgresTest.php
resurtm 351ca14c28 Unit tests for CDbColumnSchema::$comment.
2012-08-03 20:56:28 +06:00

274 lines
12 KiB
PHP
Raw Blame History

<?php
Yii::import('system.db.CDbConnection');
class CPostgresTest extends CTestCase
{
private $db;
public function setUp()
{
if(!extension_loaded('pdo') || !extension_loaded('pdo_pgsql'))
$this->markTestSkipped('PDO and PostgreSQL extensions are required.');
$this->db=new CDbConnection('pgsql:host=127.0.0.1;dbname=yii','test','test');
$this->db->charset='UTF8';
try
{
$this->db->active=true;
}
catch(Exception $e)
{
$schemaFile=realpath(dirname(__FILE__).'/../data/postgres.sql');
$this->markTestSkipped("Please read $schemaFile for details on setting up the test environment for PostgreSQL test case.");
}
try { $this->db->createCommand('DROP SCHEMA test CASCADE')->execute(); } catch(Exception $e) { }
try { $this->db->createCommand('DROP TABLE yii_types CASCADE')->execute(); } catch(Exception $e) { }
$sqls=file_get_contents(dirname(__FILE__).'/../data/postgres.sql');
foreach(explode(';',$sqls) as $sql)
{
if(trim($sql)!=='')
$this->db->createCommand($sql)->execute();
}
}
public function tearDown()
{
$this->db->active=false;
}
public function testSchema()
{
$schema=$this->db->schema;
$this->assertTrue($schema instanceof CDbSchema);
$this->assertEquals($schema->dbConnection,$this->db);
$this->assertTrue($schema->commandBuilder instanceof CDbCommandBuilder);
$this->assertEquals('"posts"',$schema->quoteTableName('posts'));
$this->assertEquals('"id"',$schema->quoteColumnName('id'));
$this->assertTrue($schema->getTable('test.posts') instanceof CDbTableSchema);
$this->assertTrue($schema->getTable('foo')===null);
}
public function testTable()
{
$table=$this->db->schema->getTable('test.posts');
$this->assertTrue($table instanceof CDbTableSchema);
$this->assertEquals('posts',$table->name);
$this->assertEquals('"test"."posts"',$table->rawName);
$this->assertEquals('id',$table->primaryKey);
$this->assertEquals(array('author_id'=>array('users','id')),$table->foreignKeys);
$this->assertEquals('test.posts_id_seq',$table->sequenceName);
$this->assertEquals(5,count($table->columns));
$this->assertTrue($table->getColumn('id') instanceof CDbColumnSchema);
$this->assertTrue($table->getColumn('foo')===null);
$this->assertEquals(array('id','title','create_time','author_id','content'),$table->columnNames);
$table=$this->db->schema->getTable('test.orders');
$this->assertEquals(array('key1','key2'),$table->primaryKey);
$table=$this->db->schema->getTable('test.items');
$this->assertEquals('id',$table->primaryKey);
$this->assertEquals(array('col1'=>array('orders','key1'),'col2'=>array('orders','key2')),$table->foreignKeys);
$table=$this->db->schema->getTable('yii_types');
$this->assertTrue($table instanceof CDbTableSchema);
$this->assertEquals('yii_types',$table->name);
$this->assertEquals('"yii_types"',$table->rawName);
$this->assertTrue($table->primaryKey===null);
$this->assertTrue($table->foreignKeys===array());
$this->assertTrue($table->sequenceName===null);
$table=$this->db->schema->getTable('invalid');
$this->assertNull($table);
}
public function testColumn()
{
$values=array
(
'name'=>array('id', 'title', 'create_time', 'author_id', 'content'),
'rawName'=>array('"id"', '"title"', '"create_time"', '"author_id"', '"content"'),
'defaultValue'=>array(null, null, null, null, null),
'size'=>array(null, 128, null, null, null),
'precision'=>array(null, 128, null, null, null),
'scale'=>array(null, null, null, null, null),
'type'=>array('integer','string','string','integer','string'),
'isPrimaryKey'=>array(true,false,false,false,false),
'isForeignKey'=>array(false,false,false,true,false),
);
$this->checkColumns('test.posts',$values);
$values=array
(
'name'=>array('int_col', 'int_col2', 'char_col', 'char_col2', 'char_col3', 'numeric_col', 'real_col', 'blob_col', 'time', 'bool_col', 'bool_col2'),
'rawName'=>array('"int_col"', '"int_col2"', '"char_col"', '"char_col2"', '"char_col3"', '"numeric_col"', '"real_col"', '"blob_col"', '"time"', '"bool_col"', '"bool_col2"'),
'defaultValue'=>array(null, 1, null, 'something', null, null, '1.23', null, null, null, true),
'size'=>array(null, null, 100, 100, null, 4, null, null, null, null, null),
'precision'=>array(null, null, 100, 100, null, 4, null, null, null, null, null),
'scale'=>array(null, null, null, null, null, 3, null, null, null, null, null),
'type'=>array('integer','integer','string','string','string','string','double','string','string','boolean','boolean'),
'isPrimaryKey'=>array(false,false,false,false,false,false,false,false,false,false,false),
'isForeignKey'=>array(false,false,false,false,false,false,false,false,false,false,false),
);
$this->checkColumns('yii_types',$values);
}
protected function checkColumns($tableName,$values)
{
$table=$this->db->schema->getTable($tableName);
foreach($values as $name=>$value)
{
foreach(array_values($table->columns) as $i=>$column)
{
$type1=gettype($column->$name);
$type2=gettype($value[$i]);
$this->assertTrue($column->$name===$value[$i], "$tableName.{$column->name}.$name is {$column->$name} ($type1), different from the expected {$value[$i]} ($type2).");
}
}
}
public function testCommandBuilder()
{
$schema=$this->db->schema;
$builder=$schema->commandBuilder;
$this->assertTrue($builder instanceof CDbCommandBuilder);
$table=$schema->getTable('test.posts');
$c=$builder->createInsertCommand($table,array('title'=>'test post','create_time'=>'2004-10-19 10:23:54','author_id'=>1,'content'=>'test content'));
$this->assertEquals('INSERT INTO "test"."posts" ("title", "create_time", "author_id", "content") VALUES (:yp0, :yp1, :yp2, :yp3)',$c->text);
$c->execute();
$this->assertEquals(6,$builder->getLastInsertId($table));
$c=$builder->createCountCommand($table,new CDbCriteria);
$this->assertEquals('SELECT COUNT(*) FROM "test"."posts" "t"',$c->text);
$this->assertEquals(6,$c->queryScalar());
$c=$builder->createDeleteCommand($table,new CDbCriteria(array(
'condition'=>'id=:id',
'params'=>array('id'=>6))));
$this->assertEquals('DELETE FROM "test"."posts" WHERE id=:id',$c->text);
$c->execute();
$c=$builder->createCountCommand($table,new CDbCriteria);
$this->assertEquals(5,$c->queryScalar());
$c=$builder->createFindCommand($table,new CDbCriteria(array(
'select'=>'id, title',
'condition'=>'id=:id',
'params'=>array('id'=>5),
'order'=>'title',
'limit'=>2,
'offset'=>0)));
$this->assertEquals('SELECT id, title FROM "test"."posts" "t" WHERE id=:id ORDER BY title LIMIT 2',$c->text);
$rows=$c->query()->readAll();
$this->assertEquals(1,count($rows));
$this->assertEquals('post 5',$rows[0]['title']);
$c=$builder->createUpdateCommand($table,array('title'=>'new post 5'),new CDbCriteria(array(
'condition'=>'id=:id',
'params'=>array('id'=>5))));
$c->execute();
$c=$builder->createFindCommand($table,new CDbCriteria(array(
'select'=>'title',
'condition'=>'id=:id',
'params'=>array('id'=>5))));
$this->assertEquals('new post 5',$c->queryScalar());
$c=$builder->createSqlCommand('SELECT title FROM test.posts WHERE id=:id',array(':id'=>3));
$this->assertEquals('post 3',$c->queryScalar());
$c=$builder->createUpdateCounterCommand($table,array('author_id'=>-2),new CDbCriteria(array('condition'=>'id=5')));
$this->assertEquals('UPDATE "test"."posts" SET "author_id"="author_id"-2 WHERE id=5',$c->text);
$c->execute();
$c=$builder->createSqlCommand('SELECT author_id FROM posts WHERE id=5');
$this->assertEquals(1,$c->queryScalar());
// test bind by position
$c=$builder->createFindCommand($table,new CDbCriteria(array(
'select'=>'title',
'condition'=>'id=?',
'params'=>array(4))));
$this->assertEquals('SELECT title FROM "test"."posts" "t" WHERE id=?',$c->text);
$this->assertEquals('post 4',$c->queryScalar());
// another bind by position
$c=$builder->createUpdateCommand($table,array('title'=>'new post 4'),new CDbCriteria(array(
'condition'=>'id=?',
'params'=>array(4))));
$c->execute();
$c=$builder->createSqlCommand('SELECT title FROM test.posts WHERE id=4');
$this->assertEquals('new post 4',$c->queryScalar());
// testCreateCriteria
$c=$builder->createCriteria('column=:value',array(':value'=>'value'));
$this->assertEquals('column=:value',$c->condition);
$this->assertEquals(array(':value'=>'value'),$c->params);
$c=$builder->createCriteria(array('condition'=>'column=:value','params'=>array(':value'=>'value')));
$this->assertEquals('column=:value',$c->condition);
$this->assertEquals(array(':value'=>'value'),$c->params);
$c2=$builder->createCriteria($c);
$this->assertTrue($c2!==$c);
$this->assertEquals('column=:value',$c2->condition);
$this->assertEquals(array(':value'=>'value'),$c2->params);
// testCreatePkCriteria
$c=$builder->createPkCriteria($table,1,'author_id>1');
$this->assertEquals('"test"."posts"."id"=1 AND (author_id>1)',$c->condition);
$c=$builder->createPkCriteria($table,array(1,2));
$this->assertEquals('"test"."posts"."id" IN (1, 2)',$c->condition);
$table2=$schema->getTable('test.orders');
$c=$builder->createPkCriteria($table2,array('key1'=>1,'key2'=>2),'name=\'\'');
$this->assertEquals('"test"."orders"."key1"=1 AND "test"."orders"."key2"=2 AND (name=\'\')',$c->condition);
$c=$builder->createPkCriteria($table2,array(array('key1'=>1,'key2'=>2),array('key1'=>3,'key2'=>4)));
$this->assertEquals('("test"."orders"."key1", "test"."orders"."key2") IN ((1, 2), (3, 4))',$c->condition);
// createColumnCriteria
$c=$builder->createColumnCriteria($table,array('id'=>1,'author_id'=>2),'title=\'\'');
$this->assertEquals('"test"."posts"."id"=:yp0 AND "test"."posts"."author_id"=:yp1 AND (title=\'\')',$c->condition);
}
public function testResetSequence()
{
$max=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->db->createCommand("DELETE FROM test.users")->execute();
$this->db->createCommand("INSERT INTO test.users (username, password, email) VALUES ('user4','pass4','email4')")->execute();
$max2=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->assertEquals($max+1,$max2);
$userTable=$this->db->schema->getTable('test.users');
$this->db->createCommand("DELETE FROM test.users")->execute();
$this->db->schema->resetSequence($userTable);
$this->db->createCommand("INSERT INTO test.users (username, password, email) VALUES ('user4','pass4','email4')")->execute();
$max=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->assertEquals(1,$max);
$this->db->createCommand("INSERT INTO test.users (username, password, email) VALUES ('user4','pass4','email4')")->execute();
$max=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->assertEquals(2,$max);
$this->db->createCommand("DELETE FROM test.users")->execute();
$this->db->schema->resetSequence($userTable,10);
$this->db->createCommand("INSERT INTO test.users (username, password, email) VALUES ('user4','pass4','email4')")->execute();
$max=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->assertEquals(10,$max);
$this->db->createCommand("INSERT INTO test.users (username, password, email) VALUES ('user4','pass4','email4')")->execute();
$max=$this->db->createCommand("SELECT MAX(id) FROM test.users")->queryScalar();
$this->assertEquals(11,$max);
}
public function testColumnComments()
{
$usersColumns=$this->db->schema->getTable('test.users')->columns;
$this->assertEquals('',$usersColumns['id']->comment);
$this->assertEquals('Name of the user',$usersColumns['username']->comment);
$this->assertEquals('Hashed password',$usersColumns['password']->comment);
$this->assertEquals('',$usersColumns['email']->comment);
}
}
Reference in New Issue View Git Blame Copy Permalink