Fix this broken crap some more
[lhc/web/wiklou.git] / maintenance / parserTests.inc
index 00b1fe1..5c7ed47 100644 (file)
 /**
  * @todo Make this more independent of the configuration (and if possible the database)
  * @todo document
- * @addtogroup Maintenance
+ * @file
+ * @ingroup Maintenance
  */
 
 /** */
 $options = array( 'quick', 'color', 'quiet', 'help', 'show-output', 'record' );
-$optionsWithArgs = array( 'regex' );
+$optionsWithArgs = array( 'regex', 'seed' );
 
 require_once( 'commandLine.inc' );
 require_once( "$IP/maintenance/parserTestsParserHook.php" );
@@ -33,7 +34,7 @@ require_once( "$IP/maintenance/parserTestsStaticParserHook.php" );
 require_once( "$IP/maintenance/parserTestsParserTime.php" );
 
 /**
- * @addtogroup Maintenance
+ * @ingroup Maintenance
  */
 class ParserTest {
        /**
@@ -46,6 +47,25 @@ class ParserTest {
         */
        private $showOutput;
 
+       /**
+        * boolean $useTemporaryTables Use temporary tables for the temporary database
+        */
+       private $useTemporaryTables = true;
+
+       /**
+        * boolean $databaseSetupDone True if the database has been set up
+        */
+       private $databaseSetupDone = false;
+
+       /**
+        * string $oldTablePrefix Original table prefix
+        */
+       private $oldTablePrefix;
+
+       private $maxFuzzTestLength = 300;
+       private $fuzzSeed = 0;
+       private $memoryLimit = 50;
+
        /**
         * Sets terminal colorization and diff/quick modes depending on OS and
         * command-line options (--color and --quick).
@@ -82,6 +102,10 @@ class ParserTest {
 
 
                if (isset($options['regex'])) {
+                       if ( isset( $options['record'] ) ) {
+                               echo "Warning: --record cannot be used with --regex, disabling --record\n";
+                               unset( $options['record'] );
+                       }
                        $this->regex = $options['regex'];
                } else {
                        # Matches anything
@@ -89,14 +113,18 @@ class ParserTest {
                }
 
                if( isset( $options['record'] ) ) {
-                       $this->recorder = new DbTestRecorder( $this->term );
+                       $this->recorder = new DbTestRecorder( $this );
                } elseif( isset( $options['compare'] ) ) {
-                       $this->recorder = new DbTestPreviewer( $this->term );
+                       $this->recorder = new DbTestPreviewer( $this );
                } else {
-                       $this->recorder = new TestRecorder( $this->term );
+                       $this->recorder = new TestRecorder( $this );
                }
                $this->keepUploads = isset( $options['keep-uploads'] );
 
+               if ( isset( $options['seed'] ) ) {
+                       $this->fuzzSeed = intval( $options['seed'] ) - 1;
+               }
+
                $this->hooks = array();
                $this->functionHooks = array();
        }
@@ -113,6 +141,119 @@ class ParserTest {
                }
        }
 
+       /**
+        * Run a fuzz test series
+        * Draw input from a set of test files
+        */
+       function fuzzTest( $filenames ) {
+               $dict = $this->getFuzzInput( $filenames );
+               $dictSize = strlen( $dict );
+               $logMaxLength = log( $this->maxFuzzTestLength );
+               $this->setupDatabase();
+               ini_set( 'memory_limit', $this->memoryLimit * 1048576 );
+
+               $numTotal = 0;
+               $numSuccess = 0;
+               $user = new User;
+               $opts = ParserOptions::newFromUser( $user );
+               $title = Title::makeTitle( NS_MAIN, 'Parser_test' );
+
+               while ( true ) {
+                       // Generate test input
+                       mt_srand( ++$this->fuzzSeed );
+                       $totalLength = mt_rand( 1, $this->maxFuzzTestLength );
+                       $input = '';
+                       while ( strlen( $input ) < $totalLength ) {
+                               $logHairLength = mt_rand( 0, 1000000 ) / 1000000 * $logMaxLength;
+                               $hairLength = min( intval( exp( $logHairLength ) ), $dictSize );
+                               $offset = mt_rand( 0, $dictSize - $hairLength );
+                               $input .= substr( $dict, $offset, $hairLength );
+                       }
+
+                       $this->setupGlobals();
+                       $parser = $this->getParser();
+                       // Run the test
+                       try {
+                               $parser->parse( $input, $title, $opts );
+                               $fail = false;
+                       } catch ( Exception $exception ) {
+                               $fail = true;
+                       }
+
+                       if ( $fail ) {
+                               echo "Test failed with seed {$this->fuzzSeed}\n";
+                               echo "Input:\n";
+                               var_dump( $input );
+                               echo "\n\n";
+                               echo "$exception\n";
+                       } else {
+                               $numSuccess++;
+                       }
+                       $numTotal++;
+                       $this->teardownGlobals();
+                       $parser->__destruct();
+
+                       if ( $numTotal % 100 == 0 ) {
+                               $usage = intval( memory_get_usage( true ) / $this->memoryLimit / 1048576 * 100 );
+                               echo "{$this->fuzzSeed}: $numSuccess/$numTotal (mem: $usage%)\n";
+                               if ( $usage > 90 ) {
+                                       echo "Out of memory:\n";
+                                       $memStats = $this->getMemoryBreakdown();
+                                       foreach ( $memStats as $name => $usage ) {
+                                               echo "$name: $usage\n";
+                                       }
+                                       $this->abort();
+                               }
+                       }
+               }
+       }
+
+       /**
+        * Get an input dictionary from a set of parser test files
+        */
+       function getFuzzInput( $filenames ) {
+               $dict = '';
+               foreach( $filenames as $filename ) {
+                       $contents = file_get_contents( $filename );
+                       preg_match_all( '/!!\s*input\n(.*?)\n!!\s*result/s', $contents, $matches );
+                       foreach ( $matches[1] as $match ) {
+                               $dict .= $match . "\n";
+                       }
+               }
+               return $dict;
+       }
+
+       /**
+        * Get a memory usage breakdown
+        */
+       function getMemoryBreakdown() {
+               $memStats = array();
+               foreach ( $GLOBALS as $name => $value ) {
+                       $memStats['$'.$name] = strlen( serialize( $value ) );
+               }
+               $classes = get_declared_classes();
+               foreach ( $classes as $class ) {
+                       $rc = new ReflectionClass( $class );
+                       $props = $rc->getStaticProperties();
+                       $memStats[$class] = strlen( serialize( $props ) );
+                       $methods = $rc->getMethods();
+                       foreach ( $methods as $method ) {
+                               $memStats[$class] += strlen( serialize( $method->getStaticVariables() ) );
+                       }
+               }
+               $functions = get_defined_functions();
+               foreach ( $functions['user'] as $function ) {
+                       $rf = new ReflectionFunction( $function );
+                       $memStats["$function()"] = strlen( serialize( $rf->getStaticVariables() ) );
+               }
+               asort( $memStats );
+               return $memStats;
+       }
+
+       function abort() {
+               $this->abort();
+       }
+
        /**
         * Run a series of tests listed in the given text files.
         * Each test consists of a brief description, wikitext input,
@@ -126,10 +267,12 @@ class ParserTest {
         */
        public function runTestsFromFiles( $filenames ) {
                $this->recorder->start();
+               $this->setupDatabase();
                $ok = true;
                foreach( $filenames as $filename ) {
                        $ok = $this->runFile( $filename ) && $ok;
                }
+               $this->teardownDatabase();
                $this->recorder->report();
                $this->recorder->end();
                return $ok;
@@ -142,10 +285,7 @@ class ParserTest {
                } else {
                        global $IP;
                        $relative = wfRelativePath( $filename, $IP );
-                       print $this->term->color( 1 ) .
-                               "Reading tests from \"$relative\"..." .
-                               $this->term->reset() .
-                               "\n";
+                       $this->showRunFile( $relative );
                }
 
                $data = array();
@@ -247,6 +387,24 @@ class ParserTest {
                return $ok;
        }
 
+       /**
+        * Get a Parser object
+        */
+       function getParser() {
+               global $wgParserConf;
+               $class = $wgParserConf['class'];
+               $parser = new $class( $wgParserConf );
+               foreach( $this->hooks as $tag => $callback ) {
+                       $parser->setHook( $tag, $callback );
+               }
+               foreach( $this->functionHooks as $tag => $bits ) {
+                       list( $callback, $flags ) = $bits;
+                       $parser->setFunctionHook( $tag, $callback, $flags );
+               }
+               wfRunHooks( 'ParserTestParser', array( &$parser ) );
+               return $parser;
+       }
+
        /**
         * Run a given wikitext input through a freshly-constructed wiki parser,
         * and compare the output against the expected results.
@@ -257,7 +415,6 @@ class ParserTest {
         * @return bool
         */
        private function runTest( $desc, $input, $result, $opts ) {
-               global $wgParserConf;
                if( $this->showProgress ) {
                        $this->showTesting( $desc );
                }
@@ -281,18 +438,7 @@ class ParserTest {
                }
 
                $noxml = (bool)preg_match( '~\\b noxml \\b~x', $opts );
-
-               $class = $wgParserConf['class'];
-               $parser = new $class( $wgParserConf );
-               foreach( $this->hooks as $tag => $callback ) {
-                       $parser->setHook( $tag, $callback );
-               }
-               foreach( $this->functionHooks as $tag => $bits ) {
-                       list( $callback, $flags ) = $bits;
-                       $parser->setFunctionHook( $tag, $callback, $flags );
-               }
-               wfRunHooks( 'ParserTestParser', array( &$parser ) );
-
+               $parser = $this->getParser();
                $title =& Title::makeTitle( NS_MAIN, $titleText );
 
                $matches = array();
@@ -316,7 +462,12 @@ class ParserTest {
                        } else if (preg_match('/\\bcat\\b/i', $opts)) {
                                global $wgOut;
                                $wgOut->addCategoryLinks($output->getCategories());
-                               $out = $this->tidy ( implode( ' ', $wgOut->getCategoryLinks() ) );
+                               $cats = $wgOut->getCategoryLinks();
+                               if ( isset( $cats['normal'] ) ) {
+                                       $out = $this->tidy( implode( ' ', $cats['normal'] ) );
+                               } else {
+                                       $out = '';
+                               }
                        }
 
                        $result = $this->tidy($result);
@@ -352,12 +503,6 @@ class ParserTest {
         * Ideally this should replace the global configuration entirely.
         */
        private function setupGlobals($opts = '') {
-               # Save the prefixed / quoted table names for later use when we make the temporaries.
-               $db = wfGetDB( DB_SLAVE );
-               $this->oldTableNames = array();
-               foreach( $this->listTables() as $table ) {
-                       $this->oldTableNames[$table] = $db->tableName( $table );
-               }
                if( !isset( $this->uploadDir ) ) {
                        $this->uploadDir = $this->setupUploadDir();
                }
@@ -369,6 +514,8 @@ class ParserTest {
                        self::getOptionValue( '/variant=([a-z]+(?:-[a-z]+)?)/', $opts, false );
                $maxtoclevel =
                        self::getOptionValue( '/wgMaxTocLevel=(\d+)/', $opts, 999 );
+               $linkHolderBatchSize = 
+                       self::getOptionValue( '/wgLinkHolderBatchSize=(\d+)/', $opts, 1000 );
 
                $settings = array(
                        'wgServer' => 'http://localhost',
@@ -405,6 +552,16 @@ class ParserTest {
                        'wgUseTidy' => false,
                        'wgDefaultLanguageVariant' => $variant,
                        'wgVariantArticlePath' => false,
+                       'wgGroupPermissions' => array( '*' => array(
+                               'createaccount' => true,
+                               'read'          => true,
+                               'edit'          => true,
+                               'createpage'    => true,
+                               'createtalk'    => true,
+                       ) ),
+                       'wgDefaultExternalStore' => array(),
+                       'wgForeignFileRepos' => array(),
+                       'wgLinkHolderBatchSize' => $linkHolderBatchSize,
                        );
                $this->savedGlobals = array();
                foreach( $settings as $var => $val ) {
@@ -414,10 +571,9 @@ class ParserTest {
                $langObj = Language::factory( $lang );
                $GLOBALS['wgLang'] = $langObj;
                $GLOBALS['wgContLang'] = $langObj;
+               $GLOBALS['wgMemc'] = new FakeMemCachedClient;
 
-               $GLOBALS['wgLoadBalancer']->loadMasterPos();
                //$GLOBALS['wgMessageCache'] = new MessageCache( new BagOStuff(), false, 0, $GLOBALS['wgDBname'] );
-               $this->setupDatabase();
 
                global $wgUser;
                $wgUser = new User();
@@ -435,7 +591,7 @@ class ParserTest {
                        'site_stats', 'hitcounter',     'ipblocks', 'image', 'oldimage',
                        'recentchanges', 'watchlist', 'math', 'interwiki',
                        'querycache', 'objectcache', 'job', 'redirect', 'querycachetwo',
-                       'archive', 'user_groups', 'page_props'
+                       'archive', 'user_groups', 'page_props', 'category'
                );
 
                if ($wgDBtype === 'mysql') 
@@ -455,97 +611,157 @@ class ParserTest {
         * the db will be visible to later tests in the run.
         */
        private function setupDatabase() {
-               static $setupDB = false;
                global $wgDBprefix;
+               if ( $this->databaseSetupDone ) {
+                       return;
+               }
+               if ( $wgDBprefix === 'parsertest_' ) {
+                       throw new MWException( 'setupDatabase should be called before setupGlobals' );
+               }
+               $this->databaseSetupDone = true;
+               $this->oldTablePrefix = $wgDBprefix;
+
+               # CREATE TEMPORARY TABLE breaks if there is more than one server
+               # FIXME: r40209 makes temporary tables break even with just one server
+               # FIXME: (bug 15892); disabling the feature entirely as a temporary fix
+               if ( true || wfGetLB()->getServerCount() != 1 ) {
+                       $this->useTemporaryTables = false;
+               }
 
-               # Make sure we don't mess with the live DB
-               if (!$setupDB && $wgDBprefix === 'parsertest_') {
-                       # oh teh horror
-                       $GLOBALS['wgLoadBalancer'] = LoadBalancer::newFromParams( $GLOBALS['wgDBservers'] );
-                       $db = wfGetDB( DB_MASTER );
+               $temporary = $this->useTemporaryTables ? 'TEMPORARY' : '';
 
-                       $tables = $this->listTables();
+               $db = wfGetDB( DB_MASTER );
+               $tables = $this->listTables();
 
-                       if (!(strcmp($db->getServerVersion(), '4.1') < 0 and stristr($db->getSoftwareLink(), 'MySQL'))) {
-                               # Database that supports CREATE TABLE ... LIKE
-                               global $wgDBtype;
-                               if( $wgDBtype == 'postgres' ) {
-                                       $def = 'INCLUDING DEFAULTS';
-                               } else {
-                                       $def = '';
-                               }
-                               foreach ($tables as $tbl) {
-                                       $newTableName = $db->tableName( $tbl );
-                                       $tableName = $this->oldTableNames[$tbl];
-                                       $db->query("CREATE TEMPORARY TABLE $newTableName (LIKE $tableName $def)");
-                               }
+               if (!(strcmp($db->getServerVersion(), '4.1') < 0 and stristr($db->getSoftwareLink(), 'MySQL'))) {
+                       # Database that supports CREATE TABLE ... LIKE
+                       global $wgDBtype;
+                       if( $wgDBtype == 'postgres' ) {
+                               $def = 'INCLUDING DEFAULTS';
                        } else {
-                               # Hack for MySQL versions < 4.1, which don't support
-                               # "CREATE TABLE ... LIKE". Note that
-                               # "CREATE TEMPORARY TABLE ... SELECT * FROM ... LIMIT 0"
-                               # would not create the indexes we need....
-                               foreach ($tables as $tbl) {
-                                       $res = $db->query("SHOW CREATE TABLE {$this->oldTableNames[$tbl]}");
-                                       $row = $db->fetchRow($res);
-                                       $create = $row[1];
-                                       $create_tmp = preg_replace('/CREATE TABLE `(.*?)`/', 'CREATE TEMPORARY TABLE `'
-                                               . $wgDBprefix . $tbl .'`', $create);
-                                       if ($create === $create_tmp) {
-                                               # Couldn't do replacement
-                                               wfDie("could not create temporary table $tbl");
-                                       }
-                                       $db->query($create_tmp);
+                               $def = '';
+                       }
+                       foreach ($tables as $tbl) {
+                               # Clean up from previous aborted run.  So that table escaping
+                               # works correctly across DB engines, we need to change the pre-
+                               # fix back and forth so tableName() works right.
+                               $this->changePrefix( $this->oldTablePrefix );
+                               $oldTableName = $db->tableName( $tbl );
+                               $this->changePrefix( 'parsertest_' );
+                               $newTableName = $db->tableName( $tbl );
+
+                               if ( $db->tableExists( $tbl ) ) {
+                                       $db->query("DROP TABLE $newTableName");
                                }
-
+                               # Create new table
+                               $db->query("CREATE $temporary TABLE $newTableName (LIKE $oldTableName $def)");
+                       }
+               } else {
+                       # Hack for MySQL versions < 4.1, which don't support
+                       # "CREATE TABLE ... LIKE". Note that
+                       # "CREATE TEMPORARY TABLE ... SELECT * FROM ... LIMIT 0"
+                       # would not create the indexes we need....
+                       #
+                       # Note that we don't bother changing around the prefixes here be-
+                       # cause we know we're using MySQL anyway.
+                       foreach ($tables as $tbl) {
+                               $oldTableName = $db->tableName( $tbl );
+                               $res = $db->query("SHOW CREATE TABLE $oldTableName");
+                               $row = $db->fetchRow($res);
+                               $create = $row[1];
+                               $create_tmp = preg_replace('/CREATE TABLE `(.*?)`/', 
+                                       "CREATE $temporary TABLE `parsertest_$tbl`", $create);
+                               if ($create === $create_tmp) {
+                                       # Couldn't do replacement
+                                       wfDie("could not create temporary table $tbl");
+                               }
+                               $db->query($create_tmp);
                        }
-
-                       # Hack: insert a few Wikipedia in-project interwiki prefixes,
-                       # for testing inter-language links
-                       $db->insert( 'interwiki', array(
-                               array( 'iw_prefix' => 'Wikipedia',
-                                      'iw_url'    => 'http://en.wikipedia.org/wiki/$1',
-                                      'iw_local'  => 0 ),
-                               array( 'iw_prefix' => 'MeatBall',
-                                      'iw_url'    => 'http://www.usemod.com/cgi-bin/mb.pl?$1',
-                                      'iw_local'  => 0 ),
-                               array( 'iw_prefix' => 'zh',
-                                      'iw_url'    => 'http://zh.wikipedia.org/wiki/$1',
-                                      'iw_local'  => 1 ),
-                               array( 'iw_prefix' => 'es',
-                                      'iw_url'    => 'http://es.wikipedia.org/wiki/$1',
-                                      'iw_local'  => 1 ),
-                               array( 'iw_prefix' => 'fr',
-                                      'iw_url'    => 'http://fr.wikipedia.org/wiki/$1',
-                                      'iw_local'  => 1 ),
-                               array( 'iw_prefix' => 'ru',
-                                      'iw_url'    => 'http://ru.wikipedia.org/wiki/$1',
-                                      'iw_local'  => 1 ),
-                               ) );
-
-                       # Hack: Insert an image to work with
-                       $db->insert( 'image', array(
-                               'img_name'        => 'Foobar.jpg',
-                               'img_size'        => 12345,
-                               'img_description' => 'Some lame file',
-                               'img_user'        => 1,
-                               'img_user_text'   => 'WikiSysop',
-                               'img_timestamp'   => $db->timestamp( '20010115123500' ),
-                               'img_width'       => 1941,
-                               'img_height'      => 220,
-                               'img_bits'        => 24,
-                               'img_media_type'  => MEDIATYPE_BITMAP,
-                               'img_major_mime'  => "image",
-                               'img_minor_mime'  => "jpeg",
-                               'img_metadata'    => serialize( array() ),
-                               ) );
-
-                       # Update certain things in site_stats
-                       $db->insert( 'site_stats', array( 'ss_row_id' => 1, 'ss_images' => 1, 'ss_good_articles' => 1 ) );
-
-                       $setupDB = true;
                }
+
+               $this->changePrefix( 'parsertest_' );
+
+               # Hack: insert a few Wikipedia in-project interwiki prefixes,
+               # for testing inter-language links
+               $db->insert( 'interwiki', array(
+                       array( 'iw_prefix' => 'wikipedia',
+                                  'iw_url'    => 'http://en.wikipedia.org/wiki/$1',
+                                  'iw_local'  => 0 ),
+                       array( 'iw_prefix' => 'meatball',
+                                  'iw_url'    => 'http://www.usemod.com/cgi-bin/mb.pl?$1',
+                                  'iw_local'  => 0 ),
+                       array( 'iw_prefix' => 'zh',
+                                  'iw_url'    => 'http://zh.wikipedia.org/wiki/$1',
+                                  'iw_local'  => 1 ),
+                       array( 'iw_prefix' => 'es',
+                                  'iw_url'    => 'http://es.wikipedia.org/wiki/$1',
+                                  'iw_local'  => 1 ),
+                       array( 'iw_prefix' => 'fr',
+                                  'iw_url'    => 'http://fr.wikipedia.org/wiki/$1',
+                                  'iw_local'  => 1 ),
+                       array( 'iw_prefix' => 'ru',
+                                  'iw_url'    => 'http://ru.wikipedia.org/wiki/$1',
+                                  'iw_local'  => 1 ),
+                       ) );
+
+               # Hack: Insert an image to work with
+               $db->insert( 'image', array(
+                       'img_name'        => 'Foobar.jpg',
+                       'img_size'        => 12345,
+                       'img_description' => 'Some lame file',
+                       'img_user'        => 1,
+                       'img_user_text'   => 'WikiSysop',
+                       'img_timestamp'   => $db->timestamp( '20010115123500' ),
+                       'img_width'       => 1941,
+                       'img_height'      => 220,
+                       'img_bits'        => 24,
+                       'img_media_type'  => MEDIATYPE_BITMAP,
+                       'img_major_mime'  => "image",
+                       'img_minor_mime'  => "jpeg",
+                       'img_metadata'    => serialize( array() ),
+                       ) );
+
+               # Update certain things in site_stats
+               $db->insert( 'site_stats', array( 'ss_row_id' => 1, 'ss_images' => 1, 'ss_good_articles' => 1 ) );
        }
 
+       /**
+        * Change the table prefix on all open DB connections/
+        */
+       protected function changePrefix( $prefix ) {
+               global $wgDBprefix;
+               wfGetLBFactory()->forEachLB( array( $this, 'changeLBPrefix' ), array( $prefix ) );
+               $wgDBprefix = $prefix;
+       }
+
+       public function changeLBPrefix( $lb, $prefix ) {
+               $lb->forEachOpenConnection( array( $this, 'changeDBPrefix' ), array( $prefix ) );
+       }
+
+       public function changeDBPrefix( $db, $prefix ) {
+               $db->tablePrefix( $prefix );
+       }
+
+       private function teardownDatabase() {
+               global $wgDBprefix;
+               if ( !$this->databaseSetupDone ) {
+                       return;
+               }
+               $this->changePrefix( $this->oldTablePrefix );
+               $this->databaseSetupDone = false;
+               if ( $this->useTemporaryTables ) {
+                       # Don't need to do anything
+                       return;
+               }
+
+               /*
+               $tables = $this->listTables();
+               $db = wfGetDB( DB_MASTER );
+               foreach ( $tables as $table ) {
+                       $db->query( "DROP TABLE `parsertest_$table`" );
+               }*/
+       }
+       
        /**
         * Create a dummy uploads directory which will contain a couple
         * of files in order to pass existence tests.
@@ -563,9 +779,11 @@ class ParserTest {
                }
 
                wfDebug( "Creating upload directory $dir\n" );
-               mkdir( $dir );
-               mkdir( $dir . '/3' );
-               mkdir( $dir . '/3/3a' );
+               if ( file_exists( $dir ) ) {
+                       wfDebug( "Already exists!\n" );
+                       return $dir;
+               }
+               wfMkdirParents( $dir . '/3/3a' );
                copy( "$IP/skins/monobook/headbg.jpg", "$dir/3/3a/Foobar.jpg" );
                return $dir;
        }
@@ -576,6 +794,7 @@ class ParserTest {
         */
        private function teardownGlobals() {
                RepoGroup::destroySingleton();
+               LinkCache::singleton()->clear();
                foreach( $this->savedGlobals as $var => $val ) {
                        $GLOBALS[$var] = $val;
                }
@@ -646,7 +865,7 @@ class ParserTest {
        /**
         * "Running test $desc..."
         */
-       private function showTesting( $desc ) {
+       protected function showTesting( $desc ) {
                print "Running test $desc... ";
        }
 
@@ -656,7 +875,7 @@ class ParserTest {
         * @param string $desc The test name
         * @return bool
         */
-       private function showSuccess( $desc ) {
+       protected function showSuccess( $desc ) {
                if( $this->showProgress ) {
                        print $this->term->color( '1;32' ) . 'PASSED' . $this->term->reset() . "\n";
                }
@@ -672,7 +891,7 @@ class ParserTest {
         * @param string $html Actual HTML output
         * @return bool
         */
-       private function showFailure( $desc, $result, $html ) {
+       protected function showFailure( $desc, $result, $html ) {
                if( $this->showFailure ) {
                        if( !$this->showProgress ) {
                                # In quiet mode we didn't show the 'Testing' message before the
@@ -703,7 +922,7 @@ class ParserTest {
         * @param string $outFileTail Tailing for the output file name
         * @return string
         */
-       private function quickDiff( $input, $output, $inFileTail='expected', $outFileTail='actual' ) {
+       protected function quickDiff( $input, $output, $inFileTail='expected', $outFileTail='actual' ) {
                $prefix = wfTempDir() . "/mwParser-" . mt_rand();
 
                $infile = "$prefix-$inFileTail";
@@ -738,7 +957,7 @@ class ParserTest {
         * @param string $text
         * @return string
         */
-       private function colorDiff( $text ) {
+       protected function colorDiff( $text ) {
                return preg_replace(
                        array( '/^(-.*)$/m', '/^(\+.*)$/m' ),
                        array( $this->term->color( 34 ) . '$1' . $this->term->reset(),
@@ -746,6 +965,18 @@ class ParserTest {
                        $text );
        }
 
+       /**
+        * Show "Reading tests from ..."
+        *
+        * @param String $path
+        */
+       protected function showRunFile( $path ){
+               print $this->term->color( 1 ) .
+                       "Reading tests from \"$path\"..." .
+                       $this->term->reset() .
+                       "\n";
+       }
+
        /**
         * Insert a temporary test article
         * @param string $name the title, including any prefix
@@ -902,8 +1133,12 @@ class DummyTermColorer {
 }
 
 class TestRecorder {
-       function __construct( $term ) {
-               $this->term = $term;
+       var $parent;
+       var $term;
+
+       function __construct( $parent ) {
+               $this->parent = $parent;
+               $this->term = $parent->term;
        }
 
        function start() {
@@ -942,14 +1177,21 @@ class TestRecorder {
        }
 }
 
-class DbTestRecorder extends TestRecorder  {
+class DbTestPreviewer extends TestRecorder  {
+       protected $lb;      ///< Database load balancer
        protected $db;      ///< Database connection to the main DB
        protected $curRun;  ///< run ID number for the current run
        protected $prevRun; ///< run ID number for the previous run, if any
+       protected $results; ///< Result array
 
-       function __construct( $term ) {
-               parent::__construct( $term );
-               $this->db = wfGetDB( DB_MASTER );
+       /**
+        * This should be called before the table prefix is changed
+        */
+       function __construct( $parent ) {
+               parent::__construct( $parent );
+               $this->lb = wfGetLBFactory()->newMainLB();
+               // This connection will have the wiki's table prefix, not parsertest_
+               $this->db = $this->lb->getConnection( DB_MASTER );
        }
 
        /**
@@ -957,81 +1199,84 @@ class DbTestRecorder extends TestRecorder  {
         * and all that fun stuff
         */
        function start() {
-               global $wgDBtype;
+               global $wgDBtype, $wgDBprefix;
                parent::start();
 
-               $this->db->begin();
-
-               if( ! $this->db->tableExists( 'testrun' ) or ! $this->db->tableExists( 'testitem') ) {
-                       print "WARNING> `testrun` table not found in database. Trying to create table.\n";
-            if ($wgDBtype === 'postgres')
-                               dbsource( dirname(__FILE__) . '/testRunner.postgres.sql',  $this->db );
-                       else
-                               dbsource( dirname(__FILE__) . '/testRunner.sql',  $this->db );
-                       echo "OK, resuming.\n";
+               if( ! $this->db->tableExists( 'testrun' ) 
+                       or ! $this->db->tableExists( 'testitem' ) ) 
+               {
+                       print "WARNING> `testrun` table not found in database.\n";
+                       $this->prevRun = false;
+               } else {
+                       // We'll make comparisons against the previous run later...
+                       $this->prevRun = $this->db->selectField( 'testrun', 'MAX(tr_id)' );
                }
-
-               // We'll make comparisons against the previous run later...
-               $this->prevRun = $this->db->selectField( 'testrun', 'MAX(tr_id)' );
-
-               $this->db->insert( 'testrun',
-                       array(
-                               'tr_date'        => $this->db->timestamp(),
-                               'tr_mw_version'  => SpecialVersion::getVersion(),
-                               'tr_php_version' => phpversion(),
-                               'tr_db_version'  => $this->db->getServerVersion(),
-                               'tr_uname'       => php_uname()
-                       ),
-                       __METHOD__ );
-                       if ($wgDBtype === 'postgres')
-                               $this->curRun = $this->db->currentSequenceValue('testrun_id_seq');
-                       else
-                               $this->curRun = $this->db->insertId();
+               $this->results = array();
        }
 
-       /**
-        * Record an individual test item's success or failure to the db
-        * @param string $test
-        * @param bool $result
-        */
        function record( $test, $result ) {
                parent::record( $test, $result );
-               $this->db->insert( 'testitem',
-                       array(
-                               'ti_run'     => $this->curRun,
-                               'ti_name'    => $test,
-                               'ti_success' => $result ? 1 : 0,
-                       ),
-                       __METHOD__ );
-       }
-
-       /**
-        * Commit transaction and clean up for result recording
-        */
-       function end() {
-               $this->db->commit();
-               parent::end();
+               $this->results[$test] = $result;
        }
 
        function report() {
                if( $this->prevRun ) {
+                       // f = fail, p = pass, n = nonexistent
+                       // codes show before then after
                        $table = array(
-                               array( 'previously failing test(s) now PASSING! :)', 0, 1 ),
-                               array( 'previously PASSING test(s) removed o_O', 1, null ),
-                               array( 'new PASSING test(s) :)', null, 1 ),
-
-                               array( 'previously passing test(s) now FAILING! :(', 1, 0 ),
-                               array( 'previously FAILING test(s) removed O_o', 0, null ),
-                               array( 'new FAILING test(s) :(', null, 0 ),
-                               array( 'still FAILING test(s) :(', 0, 0 ),
+                               'fp' => 'previously failing test(s) now PASSING! :)',
+                               'pn' => 'previously PASSING test(s) removed o_O',
+                               'np' => 'new PASSING test(s) :)',
+
+                               'pf' => 'previously passing test(s) now FAILING! :(',
+                               'fn' => 'previously FAILING test(s) removed O_o',
+                               'nf' => 'new FAILING test(s) :(',
+                               'ff' => 'still FAILING test(s) :(',
                        );
-                       foreach( $table as $criteria ) {
-                               list( $label, $before, $after ) = $criteria;
-                               $differences = $this->compareResult( $before, $after );
-                               if( $differences ) {
-                                       $count = count($differences);
+
+                       $prevResults = array();
+
+                       $res = $this->db->select( 'testitem', array( 'ti_name', 'ti_success' ),
+                               array( 'ti_run' => $this->prevRun ), __METHOD__ );
+                       foreach ( $res as $row ) {
+                               if ( !$this->parent->regex 
+                                       || preg_match( "/{$this->parent->regex}/i", $row->ti_name ) )
+                               {
+                                       $prevResults[$row->ti_name] = $row->ti_success;
+                               }
+                       }
+
+                       $combined = array_keys( $this->results + $prevResults );
+
+                       # Determine breakdown by change type
+                       $breakdown = array();
+                       foreach ( $combined as $test ) {
+                               if ( !isset( $prevResults[$test] ) ) {
+                                       $before = 'n';
+                               } elseif ( $prevResults[$test] == 1 ) {
+                                       $before = 'p';
+                               } else /* if ( $prevResults[$test] == 0 )*/ {
+                                       $before = 'f';
+                               }
+                               if ( !isset( $this->results[$test] ) ) {
+                                       $after = 'n';
+                               } elseif ( $this->results[$test] == 1 ) {
+                                       $after = 'p';
+                               } else /*if ( $this->results[$test] == 0 ) */ {
+                                       $after = 'f';
+                               }
+                               $code = $before . $after;
+                               if ( isset( $table[$code] ) ) {
+                                       $breakdown[$code][$test] = $this->getTestStatusInfo( $test, $after );
+                               }
+                       }
+
+                       # Write out results
+                       foreach ( $table as $code => $label ) {
+                               if( !empty( $breakdown[$code] ) ) {
+                                       $count = count($breakdown[$code]);
                                        printf( "\n%4d %s\n", $count, $label );
-                                       foreach ($differences as $differing_test_name => $statusInfo) {
+                                       foreach ($breakdown[$code] as $differing_test_name => $statusInfo) {
                                                print "      * $differing_test_name  [$statusInfo]\n";
                                        }
                                }
@@ -1043,54 +1288,15 @@ class DbTestRecorder extends TestRecorder  {
                parent::report();
        }
 
-       /**
-        ** Returns an array of the test names with changed results, based on the specified
-        ** before/after criteria.
-        */
-       private function compareResult( $before, $after ) {
-               $testitem = $this->db->tableName( 'testitem' );
-               $prevRun = intval( $this->prevRun );
-               $curRun = intval( $this->curRun );
-               $prevStatus = $this->condition( $before );
-               $curStatus = $this->condition( $after );
-
-               // note: requires mysql >= ver 4.1 for subselects
-               if( is_null( $after ) ) {
-                       $sql = "
-                               select prev.ti_name as t from $testitem as prev
-                                       where prev.ti_run=$prevRun and
-                                               prev.ti_success $prevStatus and
-                                               (select current.ti_success from $testitem as current
-                                                       where current.ti_run=$curRun
-                                                               and prev.ti_name=current.ti_name) $curStatus";
-               } else {
-                       $sql = "
-                               select current.ti_name as t from $testitem as current 
-                                       where current.ti_run=$curRun and
-                                               current.ti_success $curStatus and
-                                               (select prev.ti_success from $testitem as prev
-                                                       where prev.ti_run=$prevRun
-                                                               and prev.ti_name=current.ti_name) $prevStatus";
-               }
-               $result = $this->db->query( $sql, __METHOD__ );
-               $retval = array();
-               while ($row = $this->db->fetchObject( $result )) {
-                       $testname = $row->t;
-                       $retval[$testname] = $this->getTestStatusInfo( $testname, $after, $curRun );
-               }
-               $this->db->freeResult( $result );
-               return $retval;
-       }
-
        /**
         ** Returns a string giving information about when a test last had a status change.
         ** Could help to track down when regressions were introduced, as distinct from tests
         ** which have never passed (which are more change requests than regressions).
         */
-       private function getTestStatusInfo($testname, $after, $curRun) {
+       private function getTestStatusInfo($testname, $after) {
 
                // If we're looking at a test that has just been removed, then say when it first appeared.
-               if ( is_null( $after ) ) {
+               if ( $after == 'n' ) {
                        $changedRun = $this->db->selectField ( 'testitem',
                                                                                                   'MIN(ti_run)',
                                                                                                   array( 'ti_name' => $testname ),
@@ -1106,18 +1312,18 @@ class DbTestRecorder extends TestRecorder  {
 
                // Otherwise, this test has previous recorded results.
                // See when this test last had a different result to what we're seeing now.
-               $changedRun = $this->db->selectField ( 'testitem',
-                                                                                          'MAX(ti_run)',
-                                                                                          array( 
-                                                                                              'ti_name'    => $testname,
-                                                                                              'ti_success' => ($after ? "0" : "1"),
-                                                                                              "ti_run != " . $this->db->addQuotes ( $curRun )
-                                                                                               ), 
-                                                                                               __METHOD__ );
+               $conds = array( 
+                       'ti_name'    => $testname,
+                       'ti_success' => ($after == 'f' ? "1" : "0") );
+               if ( $this->curRun ) {
+                       $conds[] = "ti_run != " . $this->db->addQuotes ( $this->curRun );
+               }
+
+               $changedRun = $this->db->selectField ( 'testitem', 'MAX(ti_run)', $conds, __METHOD__ );
 
                // If no record of ever having had a different result.
                if ( is_null ( $changedRun ) ) {
-                       if ($after == "0") {
+                       if ($after == "f") {
                                return "Has never passed";
                        } else {
                                return "Has never failed";
@@ -1138,33 +1344,78 @@ class DbTestRecorder extends TestRecorder  {
                                                                                array( "LIMIT" => 1, "ORDER BY" => 'tr_id' )
                                                                         );
 
-               return ( $after == "0" ? "Introduced" : "Fixed" ) . " between "
+               if ( $post ) {
+                       $postDate = date( "d-M-Y H:i:s",  strtotime ( $post->tr_date  ) ) . ", {$post->tr_mw_version}";
+               } else {
+                       $postDate = 'now';
+               }
+               return ( $after == "f" ? "Introduced" : "Fixed" ) . " between "
                                . date( "d-M-Y H:i:s",  strtotime ( $pre->tr_date ) ) .  ", " . $pre->tr_mw_version
-                               . " and "
-                               . date( "d-M-Y H:i:s",  strtotime ( $post->tr_date  ) ) .  ", " . $post->tr_mw_version ;
+                               . " and $postDate";
+
        }
 
        /**
-        ** Helper function for compareResult() database querying.
+        * Commit transaction and clean up for result recording
         */
-       private function condition( $value ) {
-               if( is_null( $value ) ) {
-                       return 'IS NULL';
-               } else {
-                       return '=' . intval( $value );
-               }
+       function end() {
+               $this->lb->commitMasterChanges();
+               $this->lb->closeAll();
+               parent::end();
        }
 
 }
 
-class DbTestPreviewer extends DbTestRecorder  {
+class DbTestRecorder extends DbTestPreviewer  {
        /**
-        * Commit transaction and clean up for result recording
+        * Set up result recording; insert a record for the run with the date
+        * and all that fun stuff
         */
-       function end() {
-               $this->db->rollback();
-               TestRecorder::end();
+       function start() {
+               global $wgDBtype, $wgDBprefix;
+               $this->db->begin();
+
+               if( ! $this->db->tableExists( 'testrun' ) 
+                       or ! $this->db->tableExists( 'testitem' ) ) 
+               {
+                       print "WARNING> `testrun` table not found in database. Trying to create table.\n";
+                       if ($wgDBtype === 'postgres')
+                               $this->db->sourceFile( dirname(__FILE__) . '/testRunner.postgres.sql' );
+                       else
+                               $this->db->sourceFile( dirname(__FILE__) . '/testRunner.sql' );
+                       echo "OK, resuming.\n";
+               }
+               
+               parent::start();
+
+               $this->db->insert( 'testrun',
+                       array(
+                               'tr_date'        => $this->db->timestamp(),
+                               'tr_mw_version'  => SpecialVersion::getVersion(),
+                               'tr_php_version' => phpversion(),
+                               'tr_db_version'  => $this->db->getServerVersion(),
+                               'tr_uname'       => php_uname()
+                       ),
+                       __METHOD__ );
+                       if ($wgDBtype === 'postgres')
+                               $this->curRun = $this->db->currentSequenceValue('testrun_id_seq');
+                       else
+                               $this->curRun = $this->db->insertId();
        }
-}
 
-?>
+       /**
+        * Record an individual test item's success or failure to the db
+        * @param string $test
+        * @param bool $result
+        */
+       function record( $test, $result ) {
+               parent::record( $test, $result );
+               $this->db->insert( 'testitem',
+                       array(
+                               'ti_run'     => $this->curRun,
+                               'ti_name'    => $test,
+                               'ti_success' => $result ? 1 : 0,
+                       ),
+                       __METHOD__ );
+       }
+}