Subversion-Projekte lars-tiefland.prado

<?php

Prado::using('Application.App_Code.Dao.*');

class BaseTestCase extends UnitTestCase

{

	protected $sqlmap;

	function setup()

	{

		$app = Prado::getApplication();

		$this->sqlmap = $app->getModule('daos')->getClient();

	}

	function flushDatabase()

	{

		$conn = $this->sqlmap->getDbConnection();

		$find = 'sqlite:protected';

		if(is_int(strpos($conn->getConnectionString(),$find)))

			$conn->ConnectionString = str_replace($find, 'sqlite:../protected', $conn->ConnectionString);

		$conn->setActive(false);

		$conn->setActive(true);

		switch(strtolower($conn->getDriverName()))

		{

			case 'mysql':

			return $this->flushMySQLDatabase();

			case 'sqlite':

			return $this->flushSQLiteDatabase();

		}

	}

	function flushSQLiteDatabase()

	{

		$conn = $this->sqlmap->getDbConnection();

		$file = str_replace('sqlite:','',$conn->getConnectionString());

		$backup = $file.'.bak';

		copy($backup, $file);

	}

	function flushMySQLDatabase()

	{

		$conn = $this->sqlmap->getDbConnection();

		$file = Prado::getPathOfNamespace('Application.App_Data.MySQL4.mysql-reset','.sql');

		if(is_file($file))

			$this->runScript($conn, $file);

		else

			throw new Exception('unable to find script file '.$file);

	}

	protected function runScript($connection, $script)

	{

		$sql = file_get_contents($script);

		$lines = explode(';', $sql);

		foreach($lines as $line)

		{

			$line = trim($line);

			if(strlen($line) > 0)

				$connection->createCommand($line)->execute();

		}

	}

}

?>