yii/tests/framework/db/CDbConnectionTest.php

<?php
Yii::import('system.db.CDbConnection');

class CDbConnectionTest extends CTestCase
{
	private $_connection;

	public function setUp()
	{
		if(!extension_loaded('pdo') || !extension_loaded('pdo_sqlite'))
			$this->markTestSkipped('PDO and SQLite extensions are required.');

		$this->_connection=new CDbConnection('sqlite::memory:');
	}

	public function tearDown()
	{
		$this->_connection->active=false;
	}

	public function testAutoConnect()
	{
		$db=new CDbConnection;
		$db->connectionString='sqlite::memory:';
		$this->assertFalse($db->active);
		$this->assertTrue($db->autoConnect);
		$db->init();
		$this->assertTrue($db->active);

		$db2=new CDbConnection;
		$db2->connectionString='sqlite::memory:';
		$db2->autoConnect=false;
		$this->assertFalse($db2->autoConnect);
		$this->assertFalse($db2->active);
		$db->init();
		$this->assertFalse($db2->active);
	}

	public function testInitialized()
	{
		$db=new CDbConnection;
		$db->autoConnect=false;
		$this->assertFalse($db->isInitialized);
		$db->init();
		$this->assertTrue($db->isInitialized);
	}

	public function testActive()
	{
	    $this->assertFalse($this->_connection->active);
	    $this->_connection->active=true;
	    $this->assertTrue($this->_connection->active);
	    $pdo=$this->_connection->pdoInstance;
	    $this->assertTrue($pdo instanceof PDO);

	    $this->_connection->active=true;
	    $this->assertEquals($pdo,$this->_connection->pdoInstance);

		$this->_connection->active=false;
	    $this->assertFalse($this->_connection->active);
	    $this->assertEquals($this->_connection->pdoInstance,null);

		$connection=new CDbConnection('unknown::memory:');
		$this->setExpectedException('CException');
		$connection->active=true;
	}

	public function testCreateCommand()
	{
		$sql='SELECT * FROM posts';
		$this->_connection->active=true;
		$this->_connection->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$command=$this->_connection->createCommand($sql);
		$this->assertTrue($command instanceof CDbCommand);
	}

	public function testLastInsertID()
	{
	    $this->_connection->active=true;
	    $this->_connection->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$sql='INSERT INTO posts(title,create_time,author_id) VALUES(\'test post\',11000,1)';
		$this->_connection->createCommand($sql)->execute();
		$this->assertEquals($this->_connection->lastInsertID,6);
	}

	public function testQuoteValue()
	{
	    $this->_connection->active=true;
	    $this->_connection->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$str="this is 'my' name";
		$expectedStr="'this is ''my'' name'";
		$this->assertEquals($expectedStr,$this->_connection->quoteValue($str));
	}

	public function testColumnNameCase()
	{
	    $this->_connection->active=true;
	    $this->_connection->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$this->assertEquals(PDO::CASE_NATURAL,$this->_connection->ColumnCase);
		$this->_connection->columnCase=PDO::CASE_LOWER;
		$this->assertEquals(PDO::CASE_LOWER,$this->_connection->ColumnCase);
	}

	public function testNullConversion()
	{
	    $this->_connection->active=true;
	    $this->_connection->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$this->assertEquals(PDO::NULL_NATURAL,$this->_connection->NullConversion);
		$this->_connection->nullConversion=PDO::NULL_EMPTY_STRING;
		$this->assertEquals(PDO::NULL_EMPTY_STRING,$this->_connection->NullConversion);
	}

	public function testOdbc()
	{
		if(!extension_loaded('pdo_odbc'))
			$this->markTestSkipped('pdo_odbc extension is required.');

		$db=new CDbConnection();
		$db->connectionString='odbc:Driver=SQLite3;Database=:memory:';
		$db->driverName='sqlite';

		try
		{
			$this->assertEquals('odbc',$db->getAttribute(PDO::ATTR_DRIVER_NAME));
			$this->assertEquals('sqlite',$db->getDriverName());
		}
		catch(Exception $e)
		{
			$this->markTestSkipped('SQLite3 ODBC driver package must be installed in your operating system.');
		}

		$db->active=true;
		$db->pdoInstance->exec(file_get_contents(dirname(__FILE__).'/data/sqlite.sql'));
		$command=$db->createCommand('SELECT * FROM posts');
		$this->assertTrue($command instanceof CDbCommand);
	}
}