}
$this->query( $query, $fname );
}
+
+ /**
+ * List all tables on the database
+ *
+ * @param $prefix Only show tables with this prefix, e.g. mw_
+ * @param $fname String: calling function name
+ */
+ function listTables( $prefix = null, $fname = 'DatabaseMysql::listTables' ) {
+ $result = $this->query( "SHOW TABLES", $fname);
+
+ $endArray = array();
+
+ foreach( $result as $table ) {
+ $vars = get_object_vars($table);
+ $table = array_pop( $vars );
+
+ if( strpos( $table, $prefix ) === 0 || is_null( $prefix ) ) {
+ $endArray[] = $table;
+ }
+ }
+
+ return $endArray;
+ }
public function dropTable( $tableName, $fName = 'DatabaseMysql::dropTable' ) {
if( !$this->tableExists( $tableName ) ) {
$sql = preg_replace( '/\b' . preg_quote( $oldName ) . '\b/', $newName, $sql, 1 );
return $this->query( $sql, $fname );
}
+
+
+ /**
+ * List all tables on the database
+ *
+ * @param $prefix Only show tables with this prefix, e.g. mw_
+ * @param $fname String: calling function name
+ */
+ function listTables( $prefix = null, $fname = 'DatabaseSqlite::listTables' ) {
+ $result = $this->select(
+ 'sqlite_master',
+ 'name',
+ "type='TABLE'"
+ );
+
+ $endArray = array();
+
+ foreach( $result as $table ) {
+ $vars = get_object_vars($table);
+ $table = array_pop( $vars );
+
+ if( strpos( $table, $prefix ) === 0 || is_null( $prefix ) ) {
+ $endArray[] = $table;
+ }
+ }
+
+ return $endArray;
+ }
} // end DatabaseSqlite class
protected $suite;
public $regex = '';
public $runDisabled = false;
+
+ protected static $databaseSetupDone = false;
+ protected $db;
+ protected $dbClone;
+ protected $oldTablePrefix;
+ protected $useTemporaryTables = true;
function __construct( PHPUnit_Framework_TestSuite $suite = null ) {
if ( null !== $suite ) {
$this->suite = $suite;
}
+ parent::__construct();
+
+ if( $this->needsDB() && !is_object( $this->dbClone ) ) {
+ $this->initDB();
+ $this->addCoreDBData();
+ $this->addDBData();
+ }
+ }
+
+ function __destruct() {
+ if( $this->needsDB() && is_object( $this->dbClone ) && $this->dbClone instanceof CloneDatabase ) {
+ $this->destroyDB();
+ }
+ }
+
+ function needsDB() { return false; }
+
+ function addDBData() {}
+
+ private function addCoreDBData() {
+
+ //Make sysop user
+ $user = User::newFromName( 'UTSysop' );
+ if ( $user->idForName() == 0 ) {
+ $user->addToDatabase();
+ $user->setPassword( 'UTSysopPassword' );
+
+ $user->addGroup( 'sysop' );
+ $user->addGroup( 'bureaucrat' );
+ $user->saveSettings();
+ }
+
+
+ //Make 1 page with 1 revision
+ $article = new Article( Title::newFromText( 'UTPage' ) );
+ $article->doEdit( 'UTContent',
+ 'UTPageSummary',
+ EDIT_NEW,
+ false,
+ User::newFromName( 'UTSysop' ) );
+ }
+
+ private function initDB() {
+ global $wgDBprefix;
+
+ if ( self::$databaseSetupDone ) {
+ return;
+ }
+
+ $this->db = wfGetDB( DB_MASTER );
+ $dbType = $this->db->getType();
+
+ if ( $wgDBprefix === 'unittest_' || ( $dbType == 'oracle' && $wgDBprefix === 'ut_' ) ) {
+ throw new MWException( 'Cannot run unit tests, the database prefix is already "unittest_"' );
+ }
+
+ self::$databaseSetupDone = true;
+ $this->oldTablePrefix = $wgDBprefix;
+
+ # SqlBagOStuff broke when using temporary tables on r40209 (bug 15892).
+ # It seems to have been fixed since (r55079?).
+ # If it fails, $wgCaches[CACHE_DB] = new HashBagOStuff(); should work around it.
+
+ # CREATE TEMPORARY TABLE breaks if there is more than one server
+ if ( wfGetLB()->getServerCount() != 1 ) {
+ $this->useTemporaryTables = false;
+ }
+
+ $temporary = $this->useTemporaryTables || $dbType == 'postgres';
+
+ $tables = $this->listTables();
+
+ $prefix = $dbType != 'oracle' ? 'unittest_' : 'ut_';
+
+ $this->dbClone = new CloneDatabase( $this->db, $tables, $prefix );
+ $this->dbClone->useTemporaryTables( $temporary );
+ $this->dbClone->cloneTableStructure();
+
+ if ( $dbType == 'oracle' )
+ $this->db->query( 'BEGIN FILL_WIKI_INFO; END;' );
+
+ if ( $dbType == 'oracle' ) {
+ # Insert 0 user to prevent FK violations
+
+ # Anonymous user
+ $this->db->insert( 'user', array(
+ 'user_id' => 0,
+ 'user_name' => 'Anonymous' ) );
+ }
+
+ }
+
+ private function destroyDB() {
+ if ( !self::$databaseSetupDone ) {
+ return;
+ }
+
+ $this->dbClone->destroy();
+ self::$databaseSetupDone = false;
+
+ if ( $this->useTemporaryTables ) {
+ # Don't need to do anything
+ //return;
+ //Temporary tables seem to be broken ATM, delete anyway
+ }
+
+ if( $this->db->getType() == 'oracle' ) {
+ $tables = $this->db->listTables( 'ut_', __METHOD__ );
+ }
+ else {
+ $tables = $this->db->listTables( 'unittest_', __METHOD__ );
+ }
+
+ foreach ( $tables as $table ) {
+ $sql = $this->db->getType() == 'oracle' ? "DROP TABLE $table DROP CONSTRAINTS" : "DROP TABLE `$table`";
+ $this->db->query( $sql );
+ }
+
+ if ( $this->db->getType() == 'oracle' )
+ $this->db->query( 'BEGIN FILL_WIKI_INFO; END;' );
+
+
}
function __call( $func, $args ) {
. get_class( $this ) );
}
}
+
+ protected function listTables() {
+ global $wgDBprefix;
+
+ $tables = $this->db->listTables( $wgDBprefix, __METHOD__ );
+ $tables = array_map( create_function( '$table', 'global $wgDBprefix; return substr( $table, strlen( $wgDBprefix ) );' ), $tables );
+ return $tables;
+
+ }
}