NewParserTest.php

Go to the documentation of this file.

<?php

class NewParserTest extends MediaWikiTestCase {
        static protected $articles = array(); // Array of test articles defined by the tests
        /* The data provider is run on a different instance than the test, so it must be static
         * When running tests from several files, all tests will see all articles.
         */
        static protected $backendToUse;

        public $keepUploads = false;
        public $runDisabled = false;
        public $runParsoid = false;
        public $regex = '';
        public $showProgress = true;
        public $savedInitialGlobals = array();
        public $savedWeirdGlobals = array();
        public $savedGlobals = array();
        public $hooks = array();
        public $functionHooks = array();

        //Fuzz test
        public $maxFuzzTestLength = 300;
        public $fuzzSeed = 0;
        public $memoryLimit = 50;

        protected $file = false;

        protected function setUp() {
                global $wgNamespaceProtection, $wgNamespaceAliases;
                global $wgHooks, $IP;

                parent::setUp();

                //Setup CLI arguments
                if ( $this->getCliArg( 'regex=' ) ) {
                        $this->regex = $this->getCliArg( 'regex=' );
                } else {
                        # Matches anything
                        $this->regex = '';
                }

                $this->keepUploads = $this->getCliArg( 'keep-uploads' );

                $tmpGlobals = array();

                $tmpGlobals['wgLanguageCode'] = 'en';
                $tmpGlobals['wgContLang'] = Language::factory( 'en' );
                $tmpGlobals['wgScript'] = '/index.php';
                $tmpGlobals['wgScriptPath'] = '/';
                $tmpGlobals['wgArticlePath'] = '/wiki/$1';
                $tmpGlobals['wgStyleSheetPath'] = '/skins';
                $tmpGlobals['wgStylePath'] = '/skins';
                $tmpGlobals['wgThumbnailScriptPath'] = false;
                $tmpGlobals['wgLocalFileRepo'] = array(
                        'class' => 'LocalRepo',
                        'name' => 'local',
                        'url' => 'http://example.com/images',
                        'hashLevels' => 2,
                        'transformVia404' => false,
                        'backend' => 'local-backend'
                );
                $tmpGlobals['wgForeignFileRepos'] = array();
                $tmpGlobals['wgEnableParserCache'] = false;
                $tmpGlobals['wgHooks'] = $wgHooks;
                $tmpGlobals['wgDeferredUpdateList'] = array();
                $tmpGlobals['wgMemc'] = wfGetMainCache();
                $tmpGlobals['messageMemc'] = wfGetMessageCacheStorage();
                $tmpGlobals['parserMemc'] = wfGetParserCacheStorage();

                // $tmpGlobals['wgContLang'] = new StubContLang;
                $tmpGlobals['wgUser'] = new User;
                $context = new RequestContext();
                $tmpGlobals['wgLang'] = $context->getLanguage();
                $tmpGlobals['wgOut'] = $context->getOutput();
                $tmpGlobals['wgParser'] = new StubObject( 'wgParser', $GLOBALS['wgParserConf']['class'], array( $GLOBALS['wgParserConf'] ) );
                $tmpGlobals['wgRequest'] = $context->getRequest();

                if ( $GLOBALS['wgStyleDirectory'] === false ) {
                        $tmpGlobals['wgStyleDirectory'] = "$IP/skins";
                }


                foreach ( $tmpGlobals as $var => $val ) {
                        if ( array_key_exists( $var, $GLOBALS ) ) {
                                $this->savedInitialGlobals[$var] = $GLOBALS[$var];
                        }

                        $GLOBALS[$var] = $val;
                }

                $this->savedWeirdGlobals['mw_namespace_protection'] = $wgNamespaceProtection[NS_MEDIAWIKI];
                $this->savedWeirdGlobals['image_alias'] = $wgNamespaceAliases['Image'];
                $this->savedWeirdGlobals['image_talk_alias'] = $wgNamespaceAliases['Image_talk'];

                $wgNamespaceProtection[NS_MEDIAWIKI] = 'editinterface';
                $wgNamespaceAliases['Image'] = NS_FILE;
                $wgNamespaceAliases['Image_talk'] = NS_FILE_TALK;
        }

        protected function tearDown() {
                foreach ( $this->savedInitialGlobals as $var => $val ) {
                        $GLOBALS[$var] = $val;
                }

                global $wgNamespaceProtection, $wgNamespaceAliases;

                $wgNamespaceProtection[NS_MEDIAWIKI] = $this->savedWeirdGlobals['mw_namespace_protection'];
                $wgNamespaceAliases['Image'] = $this->savedWeirdGlobals['image_alias'];
                $wgNamespaceAliases['Image_talk'] = $this->savedWeirdGlobals['image_talk_alias'];

                // Restore backends
                RepoGroup::destroySingleton();
                FileBackendGroup::destroySingleton();

                parent::tearDown();
        }

        function addDBData() {
                $this->tablesUsed[] = 'site_stats';
                $this->tablesUsed[] = 'interwiki';
                # disabled for performance
                #$this->tablesUsed[] = 'image';

                # Hack: insert a few Wikipedia in-project interwiki prefixes,
                # for testing inter-language links
                $this->db->insert( 'interwiki', array(
                                array( 'iw_prefix' => 'wikipedia',
                                        'iw_url' => 'http://en.wikipedia.org/wiki/$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 0 ),
                                array( 'iw_prefix' => 'meatball',
                                        'iw_url' => 'http://www.usemod.com/cgi-bin/mb.pl?$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 0 ),
                                array( 'iw_prefix' => 'zh',
                                        'iw_url' => 'http://zh.wikipedia.org/wiki/$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 1 ),
                                array( 'iw_prefix' => 'es',
                                        'iw_url' => 'http://es.wikipedia.org/wiki/$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 1 ),
                                array( 'iw_prefix' => 'fr',
                                        'iw_url' => 'http://fr.wikipedia.org/wiki/$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 1 ),
                                array( 'iw_prefix' => 'ru',
                                        'iw_url' => 'http://ru.wikipedia.org/wiki/$1',
                                        'iw_api' => '',
                                        'iw_wikiid' => '',
                                        'iw_local' => 1 ),
                        ), __METHOD__, array( 'IGNORE' )
                );

                # Update certain things in site_stats
                $this->db->insert( 'site_stats',
                        array( 'ss_row_id' => 1, 'ss_images' => 2, 'ss_good_articles' => 1 ),
                        __METHOD__
                );

                # Reinitialise the LocalisationCache to match the database state
                Language::getLocalisationCache()->unloadAll();

                # Clear the message cache
                MessageCache::singleton()->clear();

                $user = User::newFromId( 0 );
                LinkCache::singleton()->clear(); # Avoids the odd failure at creating the nullRevision

                # Upload DB table entries for files.
                # We will upload the actual files later. Note that if anything causes LocalFile::load()
                # to be triggered before then, it will break via maybeUpgrade() setting the fileExists
                # member to false and storing it in cache.
                $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Foobar.jpg' ) );
                if ( !$this->db->selectField( 'image', '1', array( 'img_name' => $image->getName() ) ) ) {
                        $image->recordUpload2(
                                '', // archive name
                                'Upload of some lame file',
                                'Some lame file',
                                array(
                                        'size' => 12345,
                                        'width' => 1941,
                                        'height' => 220,
                                        'bits' => 24,
                                        'media_type' => MEDIATYPE_BITMAP,
                                        'mime' => 'image/jpeg',
                                        'metadata' => serialize( array() ),
                                        'sha1' => wfBaseConvert( '', 16, 36, 31 ),
                                        'fileExists' => true ),
                                $this->db->timestamp( '20010115123500' ), $user
                        );
                }

                # This image will be blacklisted in [[MediaWiki:Bad image list]]
                $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Bad.jpg' ) );
                if ( !$this->db->selectField( 'image', '1', array( 'img_name' => $image->getName() ) ) ) {
                        $image->recordUpload2(
                                '', // archive name
                                'zomgnotcensored',
                                'Borderline image',
                                array(
                                        'size' => 12345,
                                        'width' => 320,
                                        'height' => 240,
                                        'bits' => 24,
                                        'media_type' => MEDIATYPE_BITMAP,
                                        'mime' => 'image/jpeg',
                                        'metadata' => serialize( array() ),
                                        'sha1' => wfBaseConvert( '', 16, 36, 31 ),
                                        'fileExists' => true ),
                                $this->db->timestamp( '20010115123500' ), $user
                        );
                }
        }

        //ParserTest setup/teardown functions

        protected function setupGlobals( $opts = array(), $config = '' ) {
                global $wgFileBackends;
                # Find out values for some special options.
                $lang =
                        self::getOptionValue( 'language', $opts, 'en' );
                $variant =
                        self::getOptionValue( 'variant', $opts, false );
                $maxtoclevel =
                        self::getOptionValue( 'wgMaxTocLevel', $opts, 999 );
                $linkHolderBatchSize =
                        self::getOptionValue( 'wgLinkHolderBatchSize', $opts, 1000 );

                $uploadDir = $this->getUploadDir();
                if ( $this->getCliArg( 'use-filebackend=' ) ) {
                        if ( self::$backendToUse ) {
                                $backend = self::$backendToUse;
                        } else {
                                $name = $this->getCliArg( 'use-filebackend=' );
                                $useConfig = array();
                                foreach ( $wgFileBackends as $conf ) {
                                        if ( $conf['name'] == $name ) {
                                                $useConfig = $conf;
                                        }
                                }
                                $useConfig['name'] = 'local-backend'; // swap name
                                $class = $conf['class'];
                                self::$backendToUse = new $class( $useConfig );
                                $backend = self::$backendToUse;
                        }
                } else {
                        $backend = new FSFileBackend( array(
                                'name' => 'local-backend',
                                'lockManager' => 'nullLockManager',
                                'containerPaths' => array(
                                        'local-public' => "$uploadDir",
                                        'local-thumb' => "$uploadDir/thumb",
                                )
                        ) );
                }

                $settings = array(
                        'wgServer' => 'http://example.org',
                        'wgScript' => '/index.php',
                        'wgScriptPath' => '/',
                        'wgArticlePath' => '/wiki/$1',
                        'wgExtensionAssetsPath' => '/extensions',
                        'wgActionPaths' => array(),
                        'wgLocalFileRepo' => array(
                                'class' => 'LocalRepo',
                                'name' => 'local',
                                'url' => 'http://example.com/images',
                                'hashLevels' => 2,
                                'transformVia404' => false,
                                'backend' => $backend
                        ),
                        'wgEnableUploads' => self::getOptionValue( 'wgEnableUploads', $opts, true ),
                        'wgStylePath' => '/skins',
                        'wgStyleSheetPath' => '/skins',
                        'wgSitename' => 'MediaWiki',
                        'wgLanguageCode' => $lang,
                        'wgDBprefix' => $this->db->getType() != 'oracle' ? 'unittest_' : 'ut_',
                        'wgRawHtml' => isset( $opts['rawhtml'] ),
                        'wgLang' => null,
                        'wgContLang' => null,
                        'wgNamespacesWithSubpages' => array( NS_MAIN => isset( $opts['subpage'] ) ),
                        'wgMaxTocLevel' => $maxtoclevel,
                        'wgCapitalLinks' => true,
                        'wgNoFollowLinks' => true,
                        'wgNoFollowDomainExceptions' => array(),
                        'wgThumbnailScriptPath' => false,
                        'wgUseImageResize' => true,
                        'wgUseTeX' => isset( $opts['math'] ),
                        'wgMathDirectory' => $uploadDir . '/math',
                        'wgLocaltimezone' => 'UTC',
                        'wgAllowExternalImages' => true,
                        'wgUseTidy' => false,
                        'wgDefaultLanguageVariant' => $variant,
                        'wgVariantArticlePath' => false,
                        'wgGroupPermissions' => array( '*' => array(
                                'createaccount' => true,
                                'read' => true,
                                'edit' => true,
                                'createpage' => true,
                                'createtalk' => true,
                        ) ),
                        'wgNamespaceProtection' => array( NS_MEDIAWIKI => 'editinterface' ),
                        'wgDefaultExternalStore' => array(),
                        'wgForeignFileRepos' => array(),
                        'wgLinkHolderBatchSize' => $linkHolderBatchSize,
                        'wgExperimentalHtmlIds' => false,
                        'wgExternalLinkTarget' => false,
                        'wgAlwaysUseTidy' => false,
                        'wgHtml5' => true,
                        'wgWellFormedXml' => true,
                        'wgAllowMicrodataAttributes' => true,
                        'wgAdaptiveMessageCache' => true,
                        'wgUseDatabaseMessages' => true,
                );

                if ( $config ) {
                        $configLines = explode( "\n", $config );

                        foreach ( $configLines as $line ) {
                                list( $var, $value ) = explode( '=', $line, 2 );

                                $settings[$var] = eval( "return $value;" ); //???
                        }
                }

                $this->savedGlobals = array();

                wfRunHooks( 'ParserTestGlobals', array( &$settings ) );

                foreach ( $settings as $var => $val ) {
                        if ( array_key_exists( $var, $GLOBALS ) ) {
                                $this->savedGlobals[$var] = $GLOBALS[$var];
                        }

                        $GLOBALS[$var] = $val;
                }

                $langObj = Language::factory( $lang );
                $GLOBALS['wgContLang'] = $langObj;
                $context = new RequestContext();
                $GLOBALS['wgLang'] = $context->getLanguage();

                $GLOBALS['wgMemc'] = new EmptyBagOStuff;
                $GLOBALS['wgOut'] = $context->getOutput();
                $GLOBALS['wgUser'] = $context->getUser();

                global $wgHooks;

                $wgHooks['ParserTestParser'][] = 'ParserTestParserHook::setup';
                $wgHooks['ParserGetVariableValueTs'][] = 'ParserTest::getFakeTimestamp';

                MagicWord::clearCache();
                RepoGroup::destroySingleton();
                FileBackendGroup::destroySingleton();

                # Create dummy files in storage
                $this->setupUploads();

                # Publish the articles after we have the final language set
                $this->publishTestArticles();

                # The entries saved into RepoGroup cache with previous globals will be wrong.
                RepoGroup::destroySingleton();
                FileBackendGroup::destroySingleton();
                MessageCache::destroyInstance();

                return $context;
        }

        protected function getUploadDir() {
                if ( $this->keepUploads ) {
                        $dir = wfTempDir() . '/mwParser-images';

                        if ( is_dir( $dir ) ) {
                                return $dir;
                        }
                } else {
                        $dir = wfTempDir() . "/mwParser-" . mt_rand() . "-images";
                }

                // wfDebug( "Creating upload directory $dir\n" );
                if ( file_exists( $dir ) ) {
                        wfDebug( "Already exists!\n" );
                        return $dir;
                }

                return $dir;
        }

        protected function setupUploads() {
                global $IP;

                $base = $this->getBaseDir();
                $backend = RepoGroup::singleton()->getLocalRepo()->getBackend();
                $backend->prepare( array( 'dir' => "$base/local-public/3/3a" ) );
                $backend->store( array(
                        'src' => "$IP/skins/monobook/headbg.jpg", 'dst' => "$base/local-public/3/3a/Foobar.jpg"
                ) );
                $backend->prepare( array( 'dir' => "$base/local-public/0/09" ) );
                $backend->store( array(
                        'src' => "$IP/skins/monobook/headbg.jpg", 'dst' => "$base/local-public/0/09/Bad.jpg"
                ) );
        }

        protected function teardownGlobals() {
                $this->teardownUploads();

                foreach ( $this->savedGlobals as $var => $val ) {
                        $GLOBALS[$var] = $val;
                }

                RepoGroup::destroySingleton();
                LinkCache::singleton()->clear();
        }

        private function teardownUploads() {
                if ( $this->keepUploads ) {
                        return;
                }

                $base = $this->getBaseDir();
                // delete the files first, then the dirs.
                self::deleteFiles(
                        array(
                                "$base/local-public/3/3a/Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/180px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/200px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/640px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/120px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/1280px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/20px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/270px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/300px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/30px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/360px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/400px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/40px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/70px-Foobar.jpg",
                                "$base/local-thumb/3/3a/Foobar.jpg/960px-Foobar.jpg",

                                "$base/local-public/0/09/Bad.jpg",
                                "$base/local-thumb/0/09/Bad.jpg",

                                "$base/local-public/math/f/a/5/fa50b8b616463173474302ca3e63586b.png",
                        )
                );
        }

        private static function deleteFiles( $files ) {
                $backend = RepoGroup::singleton()->getLocalRepo()->getBackend();
                foreach ( $files as $file ) {
                        $backend->delete( array( 'src' => $file ), array( 'force' => 1 ) );
                }
                foreach ( $files as $file ) {
                        $tmp = $file;
                        while ( $tmp = FileBackend::parentStoragePath( $tmp ) ) {
                                if ( !$backend->clean( array( 'dir' => $tmp ) )->isOK() ) {
                                        break;
                                }
                        }
                }
        }

        protected function getBaseDir() {
                return 'mwstore://local-backend';
        }

        public function parserTestProvider() {
                if ( $this->file === false ) {
                        global $wgParserTestFiles;
                        $this->file = $wgParserTestFiles[0];
                }
                return new TestFileIterator( $this->file, $this );
        }

        public function setParserTestFile( $filename ) {
                $this->file = $filename;
        }

        public function testParserTest( $desc, $input, $result, $opts, $config ) {
                if ( $this->regex != '' && !preg_match( '/' . $this->regex . '/', $desc ) ) {
                        $this->assertTrue( true ); // XXX: don't flood output with "test made no assertions"
                        //$this->markTestSkipped( 'Filtered out by the user' );
                        return;
                }

                if ( !$this->isWikitextNS( NS_MAIN ) ) {
                        // parser tests frequently assume that the main namespace contains wikitext.
                        // @todo: When setting up pages, force the content model. Only skip if
                        //        $wgtContentModelUseDB is false.
                        $this->markTestSkipped( "Main namespace does not support wikitext,"
                                . "skipping parser test: $desc" );
                }

                wfDebug( "Running parser test: $desc\n" );

                $opts = $this->parseOptions( $opts );
                $context = $this->setupGlobals( $opts, $config );

                $user = $context->getUser();
                $options = ParserOptions::newFromContext( $context );

                if ( isset( $opts['title'] ) ) {
                        $titleText = $opts['title'];
                } else {
                        $titleText = 'Parser test';
                }

                $local = isset( $opts['local'] );
                $preprocessor = isset( $opts['preprocessor'] ) ? $opts['preprocessor'] : null;
                $parser = $this->getParser( $preprocessor );

                $title = Title::newFromText( $titleText );

                if ( isset( $opts['pst'] ) ) {
                        $out = $parser->preSaveTransform( $input, $title, $user, $options );
                } elseif ( isset( $opts['msg'] ) ) {
                        $out = $parser->transformMsg( $input, $options, $title );
                } elseif ( isset( $opts['section'] ) ) {
                        $section = $opts['section'];
                        $out = $parser->getSection( $input, $section );
                } elseif ( isset( $opts['replace'] ) ) {
                        $section = $opts['replace'][0];
                        $replace = $opts['replace'][1];
                        $out = $parser->replaceSection( $input, $section, $replace );
                } elseif ( isset( $opts['comment'] ) ) {
                        $out = Linker::formatComment( $input, $title, $local );
                } elseif ( isset( $opts['preload'] ) ) {
                        $out = $parser->getpreloadText( $input, $title, $options );
                } else {
                        $output = $parser->parse( $input, $title, $options, true, true, 1337 );
                        $out = $output->getText();

                        if ( isset( $opts['showtitle'] ) ) {
                                if ( $output->getTitleText() ) {
                                        $title = $output->getTitleText();
                                }

                                $out = "$title\n$out";
                        }

                        if ( isset( $opts['ill'] ) ) {
                                $out = $this->tidy( implode( ' ', $output->getLanguageLinks() ) );
                        } elseif ( isset( $opts['cat'] ) ) {
                                $outputPage = $context->getOutput();
                                $outputPage->addCategoryLinks( $output->getCategories() );
                                $cats = $outputPage->getCategoryLinks();

                                if ( isset( $cats['normal'] ) ) {
                                        $out = $this->tidy( implode( ' ', $cats['normal'] ) );
                                } else {
                                        $out = '';
                                }
                        }
                        $parser->mPreprocessor = null;

                        $result = $this->tidy( $result );
                }

                $this->teardownGlobals();

                $this->assertEquals( $result, $out, $desc );
        }

        function testFuzzTests() {
                global $wgParserTestFiles;

                $files = $wgParserTestFiles;

                if ( $this->getCliArg( 'file=' ) ) {
                        $files = array( $this->getCliArg( 'file=' ) );
                }

                $dict = $this->getFuzzInput( $files );
                $dictSize = strlen( $dict );
                $logMaxLength = log( $this->maxFuzzTestLength );

                ini_set( 'memory_limit', $this->memoryLimit * 1048576 );

                $user = new User;
                $opts = ParserOptions::newFromUser( $user );
                $title = Title::makeTitle( NS_MAIN, 'Parser_test' );

                $id = 1;

                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 );
                                $this->assertTrue( true, "Test $id, fuzz seed {$this->fuzzSeed}" );
                        } catch ( Exception $exception ) {
                                $input_dump = sprintf( "string(%d) \"%s\"\n", strlen( $input ), $input );

                                $this->assertTrue( false, "Test $id, fuzz seed {$this->fuzzSeed}. \n\nInput: $input_dump\n\nError: {$exception->getMessage()}\n\nBacktrace: {$exception->getTraceAsString()}" );
                        }

                        $this->teardownGlobals();
                        $parser->__destruct();

                        if ( $id % 100 == 0 ) {
                                $usage = intval( memory_get_usage( true ) / $this->memoryLimit / 1048576 * 100 );
                                //echo "{$this->fuzzSeed}: $numSuccess/$numTotal (mem: $usage%)\n";
                                if ( $usage > 90 ) {
                                        $ret = "Out of memory:\n";
                                        $memStats = $this->getMemoryBreakdown();

                                        foreach ( $memStats as $name => $usage ) {
                                                $ret .= "$name: $usage\n";
                                        }

                                        throw new MWException( $ret );
                                }
                        }

                        $id++;

                }
        }

        //Various getter functions

        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;
        }

        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 getParser( $preprocessor = null ) {
                global $wgParserConf;

                $class = $wgParserConf['class'];
                $parser = new $class( array( 'preprocessorClass' => $preprocessor ) + $wgParserConf );

                wfRunHooks( 'ParserTestParser', array( &$parser ) );

                return $parser;
        }

        //Various action functions

        public function addArticle( $name, $text, $line ) {
                self::$articles[$name] = array( $text, $line );
        }

        public function publishTestArticles() {
                if ( empty( self::$articles ) ) {
                        return;
                }

                foreach ( self::$articles as $name => $info ) {
                        list( $text, $line ) = $info;
                        ParserTest::addArticle( $name, $text, $line, 'ignoreduplicate' );
                }
        }

        public function requireHook( $name ) {
                global $wgParser;
                $wgParser->firstCallInit(); // make sure hooks are loaded.
                return isset( $wgParser->mTagHooks[$name] );
        }

        public function requireFunctionHook( $name ) {
                global $wgParser;
                $wgParser->firstCallInit(); // make sure hooks are loaded.
                return isset( $wgParser->mFunctionHooks[$name] );
        }

        //Various "cleanup" functions

        protected function tidy( $text ) {
                global $wgUseTidy;

                if ( $wgUseTidy ) {
                        $text = MWTidy::tidy( $text );
                }

                return $text;
        }

        public function removeEndingNewline( $s ) {
                if ( substr( $s, -1 ) === "\n" ) {
                        return substr( $s, 0, -1 );
                } else {
                        return $s;
                }
        }

        //Test options parser functions

        protected function parseOptions( $instring ) {
                $opts = array();
                // foo
                // foo=bar
                // foo="bar baz"
                // foo=[[bar baz]]
                // foo=bar,"baz quux"
                $regex = '/\b
                        ([\w-]+)                                                # Key
                        \b
                        (?:\s*
                                =                                               # First sub-value
                                \s*
                                (
                                        "
                                                [^"]*                   # Quoted val
                                        "
                                |
                                        \[\[
                                                [^]]*                   # Link target
                                        \]\]
                                |
                                        [\w-]+                          # Plain word
                                )
                                (?:\s*
                                        ,                                       # Sub-vals 1..N
                                        \s*
                                        (
                                                "[^"]*"                 # Quoted val
                                        |
                                                \[\[[^]]*\]\]   # Link target
                                        |
                                                [\w-]+                  # Plain word
                                        )
                                )*
                        )?
                        /x';

                if ( preg_match_all( $regex, $instring, $matches, PREG_SET_ORDER ) ) {
                        foreach ( $matches as $bits ) {
                                array_shift( $bits );
                                $key = strtolower( array_shift( $bits ) );
                                if ( count( $bits ) == 0 ) {
                                        $opts[$key] = true;
                                } elseif ( count( $bits ) == 1 ) {
                                        $opts[$key] = $this->cleanupOption( array_shift( $bits ) );
                                } else {
                                        // Array!
                                        $opts[$key] = array_map( array( $this, 'cleanupOption' ), $bits );
                                }
                        }
                }
                return $opts;
        }

        protected function cleanupOption( $opt ) {
                if ( substr( $opt, 0, 1 ) == '"' ) {
                        return substr( $opt, 1, -1 );
                }

                if ( substr( $opt, 0, 2 ) == '[[' ) {
                        return substr( $opt, 2, -2 );
                }
                return $opt;
        }

        protected static function getOptionValue( $key, $opts, $default ) {
                $key = strtolower( $key );

                if ( isset( $opts[$key] ) ) {
                        return $opts[$key];
                } else {
                        return $default;
                }
        }
}