142 lines
3.4 KiB
PHP
142 lines
3.4 KiB
PHP
<?php
|
|
|
|
class Mock_Database_DB {
|
|
|
|
/**
|
|
* @var array DB configuration
|
|
*/
|
|
private $config = array();
|
|
|
|
/**
|
|
* @var string DB driver name
|
|
*/
|
|
private static $dbdriver = '';
|
|
|
|
/**
|
|
* @var string DB sub-driver name
|
|
*/
|
|
private static $subdriver = '';
|
|
|
|
/**
|
|
* Prepare database configuration skeleton
|
|
*
|
|
* @param array DB configuration to set
|
|
* @return void
|
|
*/
|
|
public function __construct($config = array())
|
|
{
|
|
$this->config = $config;
|
|
}
|
|
|
|
/**
|
|
* Build DSN connection string for DB driver instantiate process
|
|
*
|
|
* @param string Group name
|
|
* @return string DSN Connection string
|
|
*/
|
|
public function set_dsn($group = 'default')
|
|
{
|
|
if ( ! isset($this->config[$group]))
|
|
{
|
|
throw new InvalidArgumentException('Group '.$group.' not exists');
|
|
}
|
|
|
|
self::$dbdriver = $this->config[$group]['dbdriver'];
|
|
if (isset($this->config[$group]['subdriver']))
|
|
{
|
|
self::$subdriver = $this->config[$group]['subdriver'];
|
|
}
|
|
|
|
$params = array(
|
|
'dbprefix' => '',
|
|
'pconnect' => FALSE,
|
|
'db_debug' => FALSE,
|
|
'cache_on' => FALSE,
|
|
'cachedir' => '',
|
|
'char_set' => 'utf8',
|
|
'dbcollat' => 'utf8_general_ci',
|
|
'swap_pre' => '',
|
|
'stricton' => FALSE
|
|
);
|
|
|
|
$config = array_merge($this->config[$group], $params);
|
|
$dsnstring = empty($config['dsn']) ? FALSE : $config['dsn'];
|
|
$subdriver = empty($config['subdriver']) ? FALSE: $config['subdriver'];
|
|
$failover = empty($config['failover']) ? FALSE : $config['failover'];
|
|
|
|
$dsn = $config['dbdriver'].'://'.$config['username'].':'.$config['password']
|
|
.'@'.$config['hostname'].'/'.$config['database'];
|
|
|
|
// Build the parameter
|
|
$other_params = array_slice($config, 6);
|
|
if ($dsnstring) $other_params['dsn'] = $dsnstring;
|
|
if ($subdriver) $other_params['subdriver'] = $subdriver;
|
|
if ($failover) $other_params['failover'] = $failover;
|
|
|
|
return $dsn.'?'.http_build_query($other_params);
|
|
}
|
|
|
|
/**
|
|
* Return a database config array
|
|
*
|
|
* @see ./config
|
|
* @param string Driver based configuration
|
|
* @return array
|
|
*/
|
|
public static function config($driver)
|
|
{
|
|
$dir = realpath(dirname(__FILE__)).DIRECTORY_SEPARATOR;
|
|
return include($dir.'config'.DIRECTORY_SEPARATOR.$driver.'.php');
|
|
}
|
|
|
|
/**
|
|
* Main DB method wrapper
|
|
*
|
|
* @param string Group or DSN string
|
|
* @param bool
|
|
* @return object
|
|
*/
|
|
public static function DB($group, $query_builder = FALSE)
|
|
{
|
|
// Create dummy driver and builder files to "load" - the mocks have
|
|
// already triggered autoloading of the real files
|
|
$case = CI_TestCase::instance();
|
|
$driver = self::$dbdriver;
|
|
$subdriver = self::$subdriver;
|
|
$case->ci_vfs_create(array(
|
|
'DB_driver.php' => '',
|
|
'DB_result.php' => '',
|
|
'DB_forge.php' => '',
|
|
'DB_query_builder.php' => ''
|
|
), '', $case->ci_base_root, 'database');
|
|
if (file_exists(SYSTEM_PATH.'database/drivers/'.$driver.'/'.$driver.'_driver.php'))
|
|
{
|
|
$case->ci_vfs_create(array(
|
|
$driver.'_driver.php' => '',
|
|
$driver.'_result.php' => '',
|
|
$driver.'_forge.php' => ''
|
|
), '', $case->ci_base_root, 'database/drivers/'.$driver);
|
|
}
|
|
if ($subdriver)
|
|
{
|
|
$case->ci_vfs_create(array(
|
|
$driver.'_'.$subdriver.'_driver.php' => '',
|
|
$driver.'_'.$subdriver.'_forge.php' => ''
|
|
), '', $case->ci_base_root, 'database/drivers/'.$driver.'/subdrivers');
|
|
}
|
|
|
|
include_once(SYSTEM_PATH.'database/DB.php');
|
|
|
|
try
|
|
{
|
|
$db = DB($group, $query_builder);
|
|
}
|
|
catch (Exception $e)
|
|
{
|
|
throw new RuntimeException($e->getMessage());
|
|
}
|
|
|
|
return $db;
|
|
}
|
|
|
|
} |