Reset the cache used in User::idFromName(), otherwise tests will try to add the user...

[lhc/web/wiklou.git] / tests / phpunit / MediaWikiTestCase.php

diff --git a/tests/phpunit/MediaWikiTestCase.php b/tests/phpunit/MediaWikiTestCase.php
index 25767be..48b6212 100644 (file)
--- a/tests/phpunit/MediaWikiTestCase.php
+++ b/tests/phpunit/MediaWikiTestCase.php
@@ -10,6 +10,18 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
        protected $oldTablePrefix;
        protected $useTemporaryTables = true;
 
+       /**
+        * Table name prefixes. Oracle likes it shorter.
+        */
+       const DB_PREFIX = 'unittest_';
+       const ORA_DB_PREFIX = 'ut_';
+       
+       protected $supportedDBs = array(
+               'mysql',
+               'sqlite',
+               'oracle'
+       );
+
        function  __construct( $name = null, array $data = array(), $dataName = '' ) {
                if ($name !== null) {
                        $this->setName($name);
@@ -17,37 +29,61 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
 
                $this->data = $data;
                $this->dataName = $dataName;
+               
+               $this->backupGlobals = false;
+        $this->backupStaticAttributes = false;
        }
        
        function run( PHPUnit_Framework_TestResult $result = NULL ) {
                
                if( $this->needsDB() ) {
+               
+                       global $wgDBprefix;
                        
                        $this->db = wfGetDB( DB_MASTER );
                        
+                       $this->checkDbIsSupported();
+                       
+                       $this->oldTablePrefix = $wgDBprefix;
+                       
                        $this->destroyDB();
                        
                        $this->initDB();
                        $this->addCoreDBData();
                        $this->addDBData();
+                       
+                       parent::run( $result );
+               
+                       $this->destroyDB();
+               }
+               else {
+                       parent::run( $result );
+               
                }
                
-               parent::run( $result );
        }
        
        function __destruct() {
-               $this->destroyDB();
+               if( $this->needsDB() ) {
+                       $this->destroyDB();
+               }
        }
        
        function needsDB() {
                $rc = new ReflectionClass( $this );
                return strpos( $rc->getDocComment(), '@group Database' ) !== false;
        }
-       
+
+       /**
+        * Stub. If a test needs to add additional data to the database, it should
+        * implement this method and do so
+        */
        function addDBData() {}
        
        private function addCoreDBData() {
-               
+
+               User::resetIdByNameCache();
+
                //Make sysop user
                $user = User::newFromName( 'UTSysop' );
                
@@ -75,19 +111,18 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
 
                $dbType = $this->db->getType();
                
-               if ( $wgDBprefix === 'unittest_' || ( $dbType == 'oracle' && $wgDBprefix === 'ut_' ) ) {
+               if ( $wgDBprefix === self::DB_PREFIX || ( $dbType == 'oracle' && $wgDBprefix === self::ORA_DB_PREFIX ) ) {
                        throw new MWException( 'Cannot run unit tests, the database prefix is already "unittest_"' );
                }
 
-               $this->oldTablePrefix = $wgDBprefix;
-
                $tables = $this->listTables();
                
-               $prefix = $dbType != 'oracle' ? 'unittest_' : 'ut_';
-
+               $prefix = $dbType != 'oracle' ? self::DB_PREFIX : self::ORA_DB_PREFIX;
+               
                $this->dbClone = new CloneDatabase( $this->db, $tables, $prefix );
+               $this->dbClone->useTemporaryTables( false ); //reported problems with temp tables, disabling until fixed
                $this->dbClone->cloneTableStructure();
-
+               
                if ( $dbType == 'oracle' )
                        $this->db->query( 'BEGIN FILL_WIKI_INFO; END;' );
 
@@ -103,25 +138,29 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
        }
        
        protected function destroyDB() {
-               global $wgDBprefix;
                
                if ( $this->useTemporaryTables ) {
                        # Don't need to do anything
                        //return;
                        //Temporary tables seem to be broken ATM, delete anyway
                }
-
+               
+               if( is_null( $this->db ) ) {
+                       return;
+               }
+               
                if( $this->db->getType() == 'oracle' ) {
-                       $tables = $this->db->listTables( 'ut_', __METHOD__ );
+                       $tables = $this->db->listTables( self::ORA_DB_PREFIX, __METHOD__ );
                }
                else {
-                       $tables = $this->db->listTables( 'unittest_', __METHOD__ );
+                       $tables = $this->db->listTables( self::DB_PREFIX, __METHOD__ );
                }
                
                foreach ( $tables as $table ) {
-                       if( $this->db->tableExists( "`$table`" ) ) {
-                               $sql = $this->db->getType() == 'oracle' ? "DROP TABLE $table DROP CONSTRAINTS" : "DROP TABLE `$table`";
+                       try {
+                               $sql = $this->db->getType() == 'oracle' ? "DROP TABLE $table CASCADE CONSTRAINTS PURGE" : "DROP TABLE `$table`";
                                $this->db->query( $sql, __METHOD__ );
+                       } catch( Exception $e ) {
                        }
                }
                
@@ -161,5 +200,25 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
                return $tables;
                
        }
+       
+       protected function checkDbIsSupported() {
+               if( !in_array( $this->db->getType(), $this->supportedDBs ) ) {
+                       throw new MWException( $this->db->getType() . " is not currently supported for unit testing." );
+               }
+       }
+       
+       public function getCliArg( $offset ) {
+       
+               if( isset( MediaWikiPHPUnitCommand::$additionalOptions[$offset] ) ) {
+                       return MediaWikiPHPUnitCommand::$additionalOptions[$offset];
+               }
+               
+       }
+       
+       public function setCliArg( $offset, $value ) {
+       
+               MediaWikiPHPUnitCommand::$additionalOptions[$offset] = $value;
+               
+       }
 }