tests/parser/ParserTestRunner.php

   1 <?php
   2 /**
   3  * Generic backend for the MediaWiki parser test suite, used by both the
   4  * standalone parserTests.php and the PHPUnit "parsertests" suite.
   5  *
   6  * Copyright © 2004, 2010 Brion Vibber <brion@pobox.com>
   7  * https://www.mediawiki.org/
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License along
  20  * with this program; if not, write to the Free Software Foundation, Inc.,
  21  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  22  * http://www.gnu.org/copyleft/gpl.html
  23  *
  24  * @todo Make this more independent of the configuration (and if possible the database)
  25  * @file
  26  * @ingroup Testing
  27  */
  28 use Wikimedia\Rdbms\IDatabase;
  29 use MediaWiki\MediaWikiServices;
  30 use MediaWiki\Tidy\TidyDriverBase;
  31 use Wikimedia\ScopedCallback;
  32 use Wikimedia\TestingAccessWrapper;
  33
  34 /**
  35  * @ingroup Testing
  36  */
  37 class ParserTestRunner {
  38
  39         /**
  40          * MediaWiki core parser test files, paths
  41          * will be prefixed with __DIR__ . '/'
  42          *
  43          * @var array
  44          */
  45         private static $coreTestFiles = [
  46                 'parserTests.txt',
  47                 'extraParserTests.txt',
  48         ];
  49
  50         /**
  51          * @var bool $useTemporaryTables Use temporary tables for the temporary database
  52          */
  53         private $useTemporaryTables = true;
  54
  55         /**
  56          * @var array $setupDone The status of each setup function
  57          */
  58         private $setupDone = [
  59                 'staticSetup' => false,
  60                 'perTestSetup' => false,
  61                 'setupDatabase' => false,
  62                 'setDatabase' => false,
  63                 'setupUploads' => false,
  64         ];
  65
  66         /**
  67          * Our connection to the database
  68          * @var Database
  69          */
  70         private $db;
  71
  72         /**
  73          * Database clone helper
  74          * @var CloneDatabase
  75          */
  76         private $dbClone;
  77
  78         /**
  79          * @var TidyDriverBase
  80          */
  81         private $tidyDriver = null;
  82
  83         /**
  84          * @var TestRecorder
  85          */
  86         private $recorder;
  87
  88         /**
  89          * The upload directory, or null to not set up an upload directory
  90          *
  91          * @var string|null
  92          */
  93         private $uploadDir = null;
  94
  95         /**
  96          * The name of the file backend to use, or null to use MockFileBackend.
  97          * @var string|null
  98          */
  99         private $fileBackendName;
 100
 101         /**
 102          * A complete regex for filtering tests.
 103          * @var string
 104          */
 105         private $regex;
 106
 107         /**
 108          * A list of normalization functions to apply to the expected and actual
 109          * output.
 110          * @var array
 111          */
 112         private $normalizationFunctions = [];
 113
 114         /**
 115          * Run disabled parser tests
 116          * @var bool
 117          */
 118         private $runDisabled;
 119
 120         /**
 121          * Disable parse on article insertion
 122          * @var bool
 123          */
 124         private $disableSaveParse;
 125
 126         /**
 127          * Reuse upload directory
 128          * @var bool
 129          */
 130         private $keepUploads;
 131
 132         /**
 133          * @param TestRecorder $recorder
 134          * @param array $options
 135          */
 136         public function __construct( TestRecorder $recorder, $options = [] ) {
 137                 $this->recorder = $recorder;
 138
 139                 if ( isset( $options['norm'] ) ) {
 140                         foreach ( $options['norm'] as $func ) {
 141                                 if ( in_array( $func, [ 'removeTbody', 'trimWhitespace' ] ) ) {
 142                                         $this->normalizationFunctions[] = $func;
 143                                 } else {
 144                                         $this->recorder->warning(
 145                                                 "Warning: unknown normalization option \"$func\"\n" );
 146                                 }
 147                         }
 148                 }
 149
 150                 if ( isset( $options['regex'] ) && $options['regex'] !== false ) {
 151                         $this->regex = $options['regex'];
 152                 } else {
 153                         # Matches anything
 154                         $this->regex = '//';
 155                 }
 156
 157                 $this->keepUploads = !empty( $options['keep-uploads'] );
 158
 159                 $this->fileBackendName = $options['file-backend'] ?? false;
 160
 161                 $this->runDisabled = !empty( $options['run-disabled'] );
 162
 163                 $this->disableSaveParse = !empty( $options['disable-save-parse'] );
 164
 165                 if ( isset( $options['upload-dir'] ) ) {
 166                         $this->uploadDir = $options['upload-dir'];
 167                 }
 168         }
 169
 170         /**
 171          * Get list of filenames to extension and core parser tests
 172          *
 173          * @return array
 174          */
 175         public static function getParserTestFiles() {
 176                 global $wgParserTestFiles;
 177
 178                 // Add core test files
 179                 $files = array_map( function ( $item ) {
 180                         return __DIR__ . "/$item";
 181                 }, self::$coreTestFiles );
 182
 183                 // Plus legacy global files
 184                 $files = array_merge( $files, $wgParserTestFiles );
 185
 186                 // Auto-discover extension parser tests
 187                 $registry = ExtensionRegistry::getInstance();
 188                 foreach ( $registry->getAllThings() as $info ) {
 189                         $dir = dirname( $info['path'] ) . '/tests/parser';
 190                         if ( !file_exists( $dir ) ) {
 191                                 continue;
 192                         }
 193                         $counter = 1;
 194                         $dirIterator = new RecursiveIteratorIterator(
 195                                 new RecursiveDirectoryIterator( $dir )
 196                         );
 197                         foreach ( $dirIterator as $fileInfo ) {
 198                                 /** @var SplFileInfo $fileInfo */
 199                                 if ( substr( $fileInfo->getFilename(), -4 ) === '.txt' ) {
 200                                         $name = $info['name'] . $counter;
 201                                         while ( isset( $files[$name] ) ) {
 202                                                 $name = $info['name'] . '_' . $counter++;
 203                                         }
 204                                         $files[$name] = $fileInfo->getPathname();
 205                                 }
 206                         }
 207                 }
 208
 209                 return array_unique( $files );
 210         }
 211
 212         public function getRecorder() {
 213                 return $this->recorder;
 214         }
 215
 216         /**
 217          * Do any setup which can be done once for all tests, independent of test
 218          * options, except for database setup.
 219          *
 220          * Public setup functions in this class return a ScopedCallback object. When
 221          * this object is destroyed by going out of scope, teardown of the
 222          * corresponding test setup is performed.
 223          *
 224          * Teardown objects may be chained by passing a ScopedCallback from a
 225          * previous setup stage as the $nextTeardown parameter. This enforces the
 226          * convention that teardown actions are taken in reverse order to the
 227          * corresponding setup actions. When $nextTeardown is specified, a
 228          * ScopedCallback will be returned which first tears down the current
 229          * setup stage, and then tears down the previous setup stage which was
 230          * specified by $nextTeardown.
 231          *
 232          * @param ScopedCallback|null $nextTeardown
 233          * @return ScopedCallback
 234          */
 235         public function staticSetup( $nextTeardown = null ) {
 236                 // A note on coding style:
 237
 238                 // The general idea here is to keep setup code together with
 239                 // corresponding teardown code, in a fine-grained manner. We have two
 240                 // arrays: $setup and $teardown. The code snippets in the $setup array
 241                 // are executed at the end of the method, before it returns, and the
 242                 // code snippets in the $teardown array are executed in reverse order
 243                 // when the Wikimedia\ScopedCallback object is consumed.
 244
 245                 // Because it is a common operation to save, set and restore global
 246                 // variables, we have an additional convention: when the array key of
 247                 // $setup is a string, the string is taken to be the name of the global
 248                 // variable, and the element value is taken to be the desired new value.
 249
 250                 // It's acceptable to just do the setup immediately, instead of adding
 251                 // a closure to $setup, except when the setup action depends on global
 252                 // variable initialisation being done first. In this case, you have to
 253                 // append a closure to $setup after the global variable is appended.
 254
 255                 // When you add to setup functions in this class, please keep associated
 256                 // setup and teardown actions together in the source code, and please
 257                 // add comments explaining why the setup action is necessary.
 258
 259                 $setup = [];
 260                 $teardown = [];
 261
 262                 $teardown[] = $this->markSetupDone( 'staticSetup' );
 263
 264                 // Some settings which influence HTML output
 265                 $setup['wgSitename'] = 'MediaWiki';
 266                 $setup['wgServer'] = 'http://example.org';
 267                 $setup['wgServerName'] = 'example.org';
 268                 $setup['wgScriptPath'] = '';
 269                 $setup['wgScript'] = '/index.php';
 270                 $setup['wgResourceBasePath'] = '';
 271                 $setup['wgStylePath'] = '/skins';
 272                 $setup['wgExtensionAssetsPath'] = '/extensions';
 273                 $setup['wgArticlePath'] = '/wiki/$1';
 274                 $setup['wgActionPaths'] = [];
 275                 $setup['wgVariantArticlePath'] = false;
 276                 $setup['wgUploadNavigationUrl'] = false;
 277                 $setup['wgCapitalLinks'] = true;
 278                 $setup['wgNoFollowLinks'] = true;
 279                 $setup['wgNoFollowDomainExceptions'] = [ 'no-nofollow.org' ];
 280                 $setup['wgExternalLinkTarget'] = false;
 281                 $setup['wgLocaltimezone'] = 'UTC';
 282                 $setup['wgHtml5'] = true;
 283                 $setup['wgDisableLangConversion'] = false;
 284                 $setup['wgDisableTitleConversion'] = false;
 285
 286                 // "extra language links"
 287                 // see https://gerrit.wikimedia.org/r/111390
 288                 $setup['wgExtraInterlanguageLinkPrefixes'] = [ 'mul' ];
 289
 290                 // All FileRepo changes should be done here by injecting services,
 291                 // there should be no need to change global variables.
 292                 RepoGroup::setSingleton( $this->createRepoGroup() );
 293                 $teardown[] = function () {
 294                         RepoGroup::destroySingleton();
 295                 };
 296
 297                 // Set up null lock managers
 298                 $setup['wgLockManagers'] = [ [
 299                         'name' => 'fsLockManager',
 300                         'class' => NullLockManager::class,
 301                 ], [
 302                         'name' => 'nullLockManager',
 303                         'class' => NullLockManager::class,
 304                 ] ];
 305                 $reset = function () {
 306                         LockManagerGroup::destroySingletons();
 307                 };
 308                 $setup[] = $reset;
 309                 $teardown[] = $reset;
 310
 311                 // This allows article insertion into the prefixed DB
 312                 $setup['wgDefaultExternalStore'] = false;
 313
 314                 // This might slightly reduce memory usage
 315                 $setup['wgAdaptiveMessageCache'] = true;
 316
 317                 // This is essential and overrides disabling of database messages in TestSetup
 318                 $setup['wgUseDatabaseMessages'] = true;
 319                 $reset = function () {
 320                         MessageCache::destroyInstance();
 321                 };
 322                 $setup[] = $reset;
 323                 $teardown[] = $reset;
 324
 325                 // It's not necessary to actually convert any files
 326                 $setup['wgSVGConverter'] = 'null';
 327                 $setup['wgSVGConverters'] = [ 'null' => 'echo "1">$output' ];
 328
 329                 // Fake constant timestamp
 330                 Hooks::register( 'ParserGetVariableValueTs', function ( &$parser, &$ts ) {
 331                         $ts = $this->getFakeTimestamp();
 332                         return true;
 333                 } );
 334                 $teardown[] = function () {
 335                         Hooks::clear( 'ParserGetVariableValueTs' );
 336                 };
 337
 338                 $this->appendNamespaceSetup( $setup, $teardown );
 339
 340                 // Set up interwikis and append teardown function
 341                 $teardown[] = $this->setupInterwikis();
 342
 343                 // This affects title normalization in links. It invalidates
 344                 // MediaWikiTitleCodec objects.
 345                 $setup['wgLocalInterwikis'] = [ 'local', 'mi' ];
 346                 $reset = function () {
 347                         $this->resetTitleServices();
 348                 };
 349                 $setup[] = $reset;
 350                 $teardown[] = $reset;
 351
 352                 // Set up a mock MediaHandlerFactory
 353                 MediaWikiServices::getInstance()->disableService( 'MediaHandlerFactory' );
 354                 MediaWikiServices::getInstance()->redefineService(
 355                         'MediaHandlerFactory',
 356                         function ( MediaWikiServices $services ) {
 357                                 $handlers = $services->getMainConfig()->get( 'ParserTestMediaHandlers' );
 358                                 return new MediaHandlerFactory( $handlers );
 359                         }
 360                 );
 361                 $teardown[] = function () {
 362                         MediaWikiServices::getInstance()->resetServiceForTesting( 'MediaHandlerFactory' );
 363                 };
 364
 365                 // SqlBagOStuff broke when using temporary tables on r40209 (T17892).
 366                 // It seems to have been fixed since (r55079?), but regressed at some point before r85701.
 367                 // This works around it for now...
 368                 global $wgObjectCaches;
 369                 $setup['wgObjectCaches'] = [ CACHE_DB => $wgObjectCaches['hash'] ] + $wgObjectCaches;
 370                 if ( isset( ObjectCache::$instances[CACHE_DB] ) ) {
 371                         $savedCache = ObjectCache::$instances[CACHE_DB];
 372                         ObjectCache::$instances[CACHE_DB] = new HashBagOStuff;
 373                         $teardown[] = function () use ( $savedCache ) {
 374                                 ObjectCache::$instances[CACHE_DB] = $savedCache;
 375                         };
 376                 }
 377
 378                 $teardown[] = $this->executeSetupSnippets( $setup );
 379
 380                 // Schedule teardown snippets in reverse order
 381                 return $this->createTeardownObject( $teardown, $nextTeardown );
 382         }
 383
 384         private function appendNamespaceSetup( &$setup, &$teardown ) {
 385                 // Add a namespace shadowing a interwiki link, to test
 386                 // proper precedence when resolving links. (T53680)
 387                 $setup['wgExtraNamespaces'] = [
 388                         100 => 'MemoryAlpha',
 389                         101 => 'MemoryAlpha_talk'
 390                 ];
 391                 // Changing wgExtraNamespaces invalidates caches in MWNamespace and
 392                 // any live Language object, both on setup and teardown
 393                 $reset = function () {
 394                         MWNamespace::clearCaches();
 395                         MediaWikiServices::getInstance()->getContentLanguage()->resetNamespaces();
 396                 };
 397                 $setup[] = $reset;
 398                 $teardown[] = $reset;
 399         }
 400
 401         /**
 402          * Create a RepoGroup object appropriate for the current configuration
 403          * @return RepoGroup
 404          */
 405         protected function createRepoGroup() {
 406                 if ( $this->uploadDir ) {
 407                         if ( $this->fileBackendName ) {
 408                                 throw new MWException( 'You cannot specify both use-filebackend and upload-dir' );
 409                         }
 410                         $backend = new FSFileBackend( [
 411                                 'name' => 'local-backend',
 412                                 'wikiId' => wfWikiID(),
 413                                 'basePath' => $this->uploadDir,
 414                                 'tmpDirectory' => wfTempDir()
 415                         ] );
 416                 } elseif ( $this->fileBackendName ) {
 417                         global $wgFileBackends;
 418                         $name = $this->fileBackendName;
 419                         $useConfig = false;
 420                         foreach ( $wgFileBackends as $conf ) {
 421                                 if ( $conf['name'] === $name ) {
 422                                         $useConfig = $conf;
 423                                 }
 424                         }
 425                         if ( $useConfig === false ) {
 426                                 throw new MWException( "Unable to find file backend \"$name\"" );
 427                         }
 428                         $useConfig['name'] = 'local-backend'; // swap name
 429                         unset( $useConfig['lockManager'] );
 430                         unset( $useConfig['fileJournal'] );
 431                         $class = $useConfig['class'];
 432                         $backend = new $class( $useConfig );
 433                 } else {
 434                         # Replace with a mock. We do not care about generating real
 435                         # files on the filesystem, just need to expose the file
 436                         # informations.
 437                         $backend = new MockFileBackend( [
 438                                 'name' => 'local-backend',
 439                                 'wikiId' => wfWikiID()
 440                         ] );
 441                 }
 442
 443                 return new RepoGroup(
 444                         [
 445                                 'class' => MockLocalRepo::class,
 446                                 'name' => 'local',
 447                                 'url' => 'http://example.com/images',
 448                                 'hashLevels' => 2,
 449                                 'transformVia404' => false,
 450                                 'backend' => $backend
 451                         ],
 452                         []
 453                 );
 454         }
 455
 456         /**
 457          * Execute an array in which elements with integer keys are taken to be
 458          * callable objects, and other elements are taken to be global variable
 459          * set operations, with the key giving the variable name and the value
 460          * giving the new global variable value. A closure is returned which, when
 461          * executed, sets the global variables back to the values they had before
 462          * this function was called.
 463          *
 464          * @see staticSetup
 465          *
 466          * @param array $setup
 467          * @return closure
 468          */
 469         protected function executeSetupSnippets( $setup ) {
 470                 $saved = [];
 471                 foreach ( $setup as $name => $value ) {
 472                         if ( is_int( $name ) ) {
 473                                 $value();
 474                         } else {
 475                                 $saved[$name] = $GLOBALS[$name] ?? null;
 476                                 $GLOBALS[$name] = $value;
 477                         }
 478                 }
 479                 return function () use ( $saved ) {
 480                         $this->executeSetupSnippets( $saved );
 481                 };
 482         }
 483
 484         /**
 485          * Take a setup array in the same format as the one given to
 486          * executeSetupSnippets(), and return a ScopedCallback which, when consumed,
 487          * executes the snippets in the setup array in reverse order. This is used
 488          * to create "teardown objects" for the public API.
 489          *
 490          * @see staticSetup
 491          *
 492          * @param array $teardown The snippet array
 493          * @param ScopedCallback|null $nextTeardown A ScopedCallback to consume
 494          * @return ScopedCallback
 495          */
 496         protected function createTeardownObject( $teardown, $nextTeardown = null ) {
 497                 return new ScopedCallback( function () use ( $teardown, $nextTeardown ) {
 498                         // Schedule teardown snippets in reverse order
 499                         $teardown = array_reverse( $teardown );
 500
 501                         $this->executeSetupSnippets( $teardown );
 502                         if ( $nextTeardown ) {
 503                                 ScopedCallback::consume( $nextTeardown );
 504                         }
 505                 } );
 506         }
 507
 508         /**
 509          * Set a setupDone flag to indicate that setup has been done, and return
 510          * the teardown closure. If the flag was already set, throw an exception.
 511          *
 512          * @param string $funcName The setup function name
 513          * @return closure
 514          */
 515         protected function markSetupDone( $funcName ) {
 516                 if ( $this->setupDone[$funcName] ) {
 517                         throw new MWException( "$funcName is already done" );
 518                 }
 519                 $this->setupDone[$funcName] = true;
 520                 return function () use ( $funcName ) {
 521                         $this->setupDone[$funcName] = false;
 522                 };
 523         }
 524
 525         /**
 526          * Ensure a given setup stage has been done, throw an exception if it has
 527          * not.
 528          * @param string $funcName
 529          * @param string|null $funcName2
 530          */
 531         protected function checkSetupDone( $funcName, $funcName2 = null ) {
 532                 if ( !$this->setupDone[$funcName]
 533                         && ( $funcName === null || !$this->setupDone[$funcName2] )
 534                 ) {
 535                         throw new MWException( "$funcName must be called before calling " .
 536                                 wfGetCaller() );
 537                 }
 538         }
 539
 540         /**
 541          * Determine whether a particular setup function has been run
 542          *
 543          * @param string $funcName
 544          * @return bool
 545          */
 546         public function isSetupDone( $funcName ) {
 547                 return $this->setupDone[$funcName] ?? false;
 548         }
 549
 550         /**
 551          * Insert hardcoded interwiki in the lookup table.
 552          *
 553          * This function insert a set of well known interwikis that are used in
 554          * the parser tests. They can be considered has fixtures are injected in
 555          * the interwiki cache by using the 'InterwikiLoadPrefix' hook.
 556          * Since we are not interested in looking up interwikis in the database,
 557          * the hook completely replace the existing mechanism (hook returns false).
 558          *
 559          * @return closure for teardown
 560          */
 561         private function setupInterwikis() {
 562                 # Hack: insert a few Wikipedia in-project interwiki prefixes,
 563                 # for testing inter-language links
 564                 Hooks::register( 'InterwikiLoadPrefix', function ( $prefix, &$iwData ) {
 565                         static $testInterwikis = [
 566                                 'local' => [
 567                                         'iw_url' => 'http://doesnt.matter.org/$1',
 568                                         'iw_api' => '',
 569                                         'iw_wikiid' => '',
 570                                         'iw_local' => 0 ],
 571                                 'wikipedia' => [
 572                                         'iw_url' => 'http://en.wikipedia.org/wiki/$1',
 573                                         'iw_api' => '',
 574                                         'iw_wikiid' => '',
 575                                         'iw_local' => 0 ],
 576                                 'meatball' => [
 577                                         'iw_url' => 'http://www.usemod.com/cgi-bin/mb.pl?$1',
 578                                         'iw_api' => '',
 579                                         'iw_wikiid' => '',
 580                                         'iw_local' => 0 ],
 581                                 'memoryalpha' => [
 582                                         'iw_url' => 'http://www.memory-alpha.org/en/index.php/$1',
 583                                         'iw_api' => '',
 584                                         'iw_wikiid' => '',
 585                                         'iw_local' => 0 ],
 586                                 'zh' => [
 587                                         'iw_url' => 'http://zh.wikipedia.org/wiki/$1',
 588                                         'iw_api' => '',
 589                                         'iw_wikiid' => '',
 590                                         'iw_local' => 1 ],
 591                                 'es' => [
 592                                         'iw_url' => 'http://es.wikipedia.org/wiki/$1',
 593                                         'iw_api' => '',
 594                                         'iw_wikiid' => '',
 595                                         'iw_local' => 1 ],
 596                                 'fr' => [
 597                                         'iw_url' => 'http://fr.wikipedia.org/wiki/$1',
 598                                         'iw_api' => '',
 599                                         'iw_wikiid' => '',
 600                                         'iw_local' => 1 ],
 601                                 'ru' => [
 602                                         'iw_url' => 'http://ru.wikipedia.org/wiki/$1',
 603                                         'iw_api' => '',
 604                                         'iw_wikiid' => '',
 605                                         'iw_local' => 1 ],
 606                                 'mi' => [
 607                                         'iw_url' => 'http://mi.wikipedia.org/wiki/$1',
 608                                         'iw_api' => '',
 609                                         'iw_wikiid' => '',
 610                                         'iw_local' => 1 ],
 611                                 'mul' => [
 612                                         'iw_url' => 'http://wikisource.org/wiki/$1',
 613                                         'iw_api' => '',
 614                                         'iw_wikiid' => '',
 615                                         'iw_local' => 1 ],
 616                         ];
 617                         if ( array_key_exists( $prefix, $testInterwikis ) ) {
 618                                 $iwData = $testInterwikis[$prefix];
 619                         }
 620
 621                         // We only want to rely on the above fixtures
 622                         return false;
 623                 } );// hooks::register
 624
 625                 // Reset the service in case any other tests already cached some prefixes.
 626                 MediaWikiServices::getInstance()->resetServiceForTesting( 'InterwikiLookup' );
 627
 628                 return function () {
 629                         // Tear down
 630                         Hooks::clear( 'InterwikiLoadPrefix' );
 631                         MediaWikiServices::getInstance()->resetServiceForTesting( 'InterwikiLookup' );
 632                 };
 633         }
 634
 635         /**
 636          * Reset the Title-related services that need resetting
 637          * for each test
 638          */
 639         private function resetTitleServices() {
 640                 $services = MediaWikiServices::getInstance();
 641                 $services->resetServiceForTesting( 'TitleFormatter' );
 642                 $services->resetServiceForTesting( 'TitleParser' );
 643                 $services->resetServiceForTesting( '_MediaWikiTitleCodec' );
 644                 $services->resetServiceForTesting( 'LinkRenderer' );
 645                 $services->resetServiceForTesting( 'LinkRendererFactory' );
 646         }
 647
 648         /**
 649          * Remove last character if it is a newline
 650          * @param string $s
 651          * @return string
 652          */
 653         public static function chomp( $s ) {
 654                 if ( substr( $s, -1 ) === "\n" ) {
 655                         return substr( $s, 0, -1 );
 656                 } else {
 657                         return $s;
 658                 }
 659         }
 660
 661         /**
 662          * Run a series of tests listed in the given text files.
 663          * Each test consists of a brief description, wikitext input,
 664          * and the expected HTML output.
 665          *
 666          * Prints status updates on stdout and counts up the total
 667          * number and percentage of passed tests.
 668          *
 669          * Handles all setup and teardown.
 670          *
 671          * @param array $filenames Array of strings
 672          * @return bool True if passed all tests, false if any tests failed.
 673          */
 674         public function runTestsFromFiles( $filenames ) {
 675                 $ok = false;
 676
 677                 $teardownGuard = $this->staticSetup();
 678                 $teardownGuard = $this->setupDatabase( $teardownGuard );
 679                 $teardownGuard = $this->setupUploads( $teardownGuard );
 680
 681                 $this->recorder->start();
 682                 try {
 683                         $ok = true;
 684
 685                         foreach ( $filenames as $filename ) {
 686                                 $testFileInfo = TestFileReader::read( $filename, [
 687                                         'runDisabled' => $this->runDisabled,
 688                                         'regex' => $this->regex ] );
 689
 690                                 // Don't start the suite if there are no enabled tests in the file
 691                                 if ( !$testFileInfo['tests'] ) {
 692                                         continue;
 693                                 }
 694
 695                                 $this->recorder->startSuite( $filename );
 696                                 $ok = $this->runTests( $testFileInfo ) && $ok;
 697                                 $this->recorder->endSuite( $filename );
 698                         }
 699
 700                         $this->recorder->report();
 701                 } catch ( DBError $e ) {
 702                         $this->recorder->warning( $e->getMessage() );
 703                 }
 704                 $this->recorder->end();
 705
 706                 ScopedCallback::consume( $teardownGuard );
 707
 708                 return $ok;
 709         }
 710
 711         /**
 712          * Determine whether the current parser has the hooks registered in it
 713          * that are required by a file read by TestFileReader.
 714          * @param array $requirements
 715          * @return bool
 716          */
 717         public function meetsRequirements( $requirements ) {
 718                 foreach ( $requirements as $requirement ) {
 719                         switch ( $requirement['type'] ) {
 720                                 case 'hook':
 721                                         $ok = $this->requireHook( $requirement['name'] );
 722                                         break;
 723                                 case 'functionHook':
 724                                         $ok = $this->requireFunctionHook( $requirement['name'] );
 725                                         break;
 726                                 case 'transparentHook':
 727                                         $ok = $this->requireTransparentHook( $requirement['name'] );
 728                                         break;
 729                         }
 730                         if ( !$ok ) {
 731                                 return false;
 732                         }
 733                 }
 734                 return true;
 735         }
 736
 737         /**
 738          * Run the tests from a single file. staticSetup() and setupDatabase()
 739          * must have been called already.
 740          *
 741          * @param array $testFileInfo Parsed file info returned by TestFileReader
 742          * @return bool True if passed all tests, false if any tests failed.
 743          */
 744         public function runTests( $testFileInfo ) {
 745                 $ok = true;
 746
 747                 $this->checkSetupDone( 'staticSetup' );
 748
 749                 // Don't add articles from the file if there are no enabled tests from the file
 750                 if ( !$testFileInfo['tests'] ) {
 751                         return true;
 752                 }
 753
 754                 // If any requirements are not met, mark all tests from the file as skipped
 755                 if ( !$this->meetsRequirements( $testFileInfo['requirements'] ) ) {
 756                         foreach ( $testFileInfo['tests'] as $test ) {
 757                                 $this->recorder->startTest( $test );
 758                                 $this->recorder->skipped( $test, 'required extension not enabled' );
 759                         }
 760                         return true;
 761                 }
 762
 763                 // Add articles
 764                 $this->addArticles( $testFileInfo['articles'] );
 765
 766                 // Run tests
 767                 foreach ( $testFileInfo['tests'] as $test ) {
 768                         $this->recorder->startTest( $test );
 769                         $result =
 770                                 $this->runTest( $test );
 771                         if ( $result !== false ) {
 772                                 $ok = $ok && $result->isSuccess();
 773                                 $this->recorder->record( $test, $result );
 774                         }
 775                 }
 776
 777                 return $ok;
 778         }
 779
 780         /**
 781          * Get a Parser object
 782          *
 783          * @param string|null $preprocessor
 784          * @return Parser
 785          */
 786         function getParser( $preprocessor = null ) {
 787                 global $wgParserConf;
 788
 789                 $class = $wgParserConf['class'];
 790                 $parser = new $class( [ 'preprocessorClass' => $preprocessor ] + $wgParserConf );
 791                 ParserTestParserHook::setup( $parser );
 792
 793                 return $parser;
 794         }
 795
 796         /**
 797          * Run a given wikitext input through a freshly-constructed wiki parser,
 798          * and compare the output against the expected results.
 799          * Prints status and explanatory messages to stdout.
 800          *
 801          * staticSetup() and setupWikiData() must be called before this function
 802          * is entered.
 803          *
 804          * @param array $test The test parameters:
 805          *  - test: The test name
 806          *  - desc: The subtest description
 807          *  - input: Wikitext to try rendering
 808          *  - options: Array of test options
 809          *  - config: Overrides for global variables, one per line
 810          *
 811          * @return ParserTestResult|false false if skipped
 812          */
 813         public function runTest( $test ) {
 814                 wfDebug( __METHOD__ . ": running {$test['desc']}" );
 815                 $opts = $this->parseOptions( $test['options'] );
 816                 $teardownGuard = $this->perTestSetup( $test );
 817
 818                 $context = RequestContext::getMain();
 819                 $user = $context->getUser();
 820                 $options = ParserOptions::newFromContext( $context );
 821                 $options->setTimestamp( $this->getFakeTimestamp() );
 822
 823                 if ( isset( $opts['tidy'] ) ) {
 824                         $options->setTidy( true );
 825                 }
 826
 827                 if ( isset( $opts['title'] ) ) {
 828                         $titleText = $opts['title'];
 829                 } else {
 830                         $titleText = 'Parser test';
 831                 }
 832
 833                 if ( isset( $opts['maxincludesize'] ) ) {
 834                         $options->setMaxIncludeSize( $opts['maxincludesize'] );
 835                 }
 836                 if ( isset( $opts['maxtemplatedepth'] ) ) {
 837                         $options->setMaxTemplateDepth( $opts['maxtemplatedepth'] );
 838                 }
 839
 840                 $local = isset( $opts['local'] );
 841                 $preprocessor = $opts['preprocessor'] ?? null;
 842                 $parser = $this->getParser( $preprocessor );
 843                 $title = Title::newFromText( $titleText );
 844
 845                 if ( isset( $opts['styletag'] ) ) {
 846                         // For testing the behavior of <style> (including those deduplicated
 847                         // into <link> tags), add tag hooks to allow them to be generated.
 848                         $parser->setHook( 'style', function ( $content, $attributes, $parser ) {
 849                                 $marker = Parser::MARKER_PREFIX . '-style-' . md5( $content ) . Parser::MARKER_SUFFIX;
 850                                 $parser->mStripState->addNoWiki( $marker, $content );
 851                                 return Html::inlineStyle( $marker, 'all', $attributes );
 852                         } );
 853                         $parser->setHook( 'link', function ( $content, $attributes, $parser ) {
 854                                 return Html::element( 'link', $attributes );
 855                         } );
 856                 }
 857
 858                 if ( isset( $opts['pst'] ) ) {
 859                         $out = $parser->preSaveTransform( $test['input'], $title, $user, $options );
 860                         $output = $parser->getOutput();
 861                 } elseif ( isset( $opts['msg'] ) ) {
 862                         $out = $parser->transformMsg( $test['input'], $options, $title );
 863                 } elseif ( isset( $opts['section'] ) ) {
 864                         $section = $opts['section'];
 865                         $out = $parser->getSection( $test['input'], $section );
 866                 } elseif ( isset( $opts['replace'] ) ) {
 867                         $section = $opts['replace'][0];
 868                         $replace = $opts['replace'][1];
 869                         $out = $parser->replaceSection( $test['input'], $section, $replace );
 870                 } elseif ( isset( $opts['comment'] ) ) {
 871                         $out = Linker::formatComment( $test['input'], $title, $local );
 872                 } elseif ( isset( $opts['preload'] ) ) {
 873                         $out = $parser->getPreloadText( $test['input'], $title, $options );
 874                 } else {
 875                         $output = $parser->parse( $test['input'], $title, $options, true, true, 1337 );
 876                         $out = $output->getText( [
 877                                 'allowTOC' => !isset( $opts['notoc'] ),
 878                                 'unwrap' => !isset( $opts['wrap'] ),
 879                         ] );
 880                         if ( isset( $opts['tidy'] ) ) {
 881                                 $out = preg_replace( '/\s+$/', '', $out );
 882                         }
 883
 884                         if ( isset( $opts['showtitle'] ) ) {
 885                                 if ( $output->getTitleText() ) {
 886                                         $title = $output->getTitleText();
 887                                 }
 888
 889                                 $out = "$title\n$out";
 890                         }
 891
 892                         if ( isset( $opts['showindicators'] ) ) {
 893                                 $indicators = '';
 894                                 foreach ( $output->getIndicators() as $id => $content ) {
 895                                         $indicators .= "$id=$content\n";
 896                                 }
 897                                 $out = $indicators . $out;
 898                         }
 899
 900                         if ( isset( $opts['ill'] ) ) {
 901                                 $out = implode( ' ', $output->getLanguageLinks() );
 902                         } elseif ( isset( $opts['cat'] ) ) {
 903                                 $out = '';
 904                                 foreach ( $output->getCategories() as $name => $sortkey ) {
 905                                         if ( $out !== '' ) {
 906                                                 $out .= "\n";
 907                                         }
 908                                         $out .= "cat=$name sort=$sortkey";
 909                                 }
 910                         }
 911                 }
 912
 913                 if ( isset( $output ) && isset( $opts['showflags'] ) ) {
 914                         $actualFlags = array_keys( TestingAccessWrapper::newFromObject( $output )->mFlags );
 915                         sort( $actualFlags );
 916                         $out .= "\nflags=" . implode( ', ', $actualFlags );
 917                 }
 918
 919                 ScopedCallback::consume( $teardownGuard );
 920
 921                 $expected = $test['result'];
 922                 if ( count( $this->normalizationFunctions ) ) {
 923                         $expected = ParserTestResultNormalizer::normalize(
 924                                 $test['expected'], $this->normalizationFunctions );
 925                         $out = ParserTestResultNormalizer::normalize( $out, $this->normalizationFunctions );
 926                 }
 927
 928                 $testResult = new ParserTestResult( $test, $expected, $out );
 929                 return $testResult;
 930         }
 931
 932         /**
 933          * Use a regex to find out the value of an option
 934          * @param string $key Name of option val to retrieve
 935          * @param array $opts Options array to look in
 936          * @param mixed $default Default value returned if not found
 937          * @return mixed
 938          */
 939         private static function getOptionValue( $key, $opts, $default ) {
 940                 $key = strtolower( $key );
 941                 return $opts[$key] ?? $default;
 942         }
 943
 944         /**
 945          * Given the options string, return an associative array of options.
 946          * @todo Move this to TestFileReader
 947          *
 948          * @param string $instring
 949          * @return array
 950          */
 951         private function parseOptions( $instring ) {
 952                 $opts = [];
 953                 // foo
 954                 // foo=bar
 955                 // foo="bar baz"
 956                 // foo=[[bar baz]]
 957                 // foo=bar,"baz quux"
 958                 // foo={...json...}
 959                 $defs = '(?(DEFINE)
 960                         (?<qstr>                                        # Quoted string
 961                                 "
 962                                 (?:[^\\\\"] | \\\\.)*
 963                                 "
 964                         )
 965                         (?<json>
 966                                 \{              # Open bracket
 967                                 (?:
 968                                         [^"{}] |                                # Not a quoted string or object, or
 969                                         (?&qstr) |                              # A quoted string, or
 970                                         (?&json)                                # A json object (recursively)
 971                                 )*
 972                                 \}              # Close bracket
 973                         )
 974                         (?<value>
 975                                 (?:
 976                                         (?&qstr)                        # Quoted val
 977                                 |
 978                                         \[\[
 979                                                 [^]]*                   # Link target
 980                                         \]\]
 981                                 |
 982                                         [\w-]+                          # Plain word
 983                                 |
 984                                         (?&json)                        # JSON object
 985                                 )
 986                         )
 987                 )';
 988                 $regex = '/' . $defs . '\b
 989                         (?<k>[\w-]+)                            # Key
 990                         \b
 991                         (?:\s*
 992                                 =                                               # First sub-value
 993                                 \s*
 994                                 (?<v>
 995                                         (?&value)
 996                                         (?:\s*
 997                                                 ,                               # Sub-vals 1..N
 998                                                 \s*
 999                                                 (?&value)
1000                                         )*
1001                                 )
1002                         )?
1003                         /x';
1004                 $valueregex = '/' . $defs . '(?&value)/x';
1005
1006                 if ( preg_match_all( $regex, $instring, $matches, PREG_SET_ORDER ) ) {
1007                         foreach ( $matches as $bits ) {
1008                                 $key = strtolower( $bits['k'] );
1009                                 if ( !isset( $bits['v'] ) ) {
1010                                         $opts[$key] = true;
1011                                 } else {
1012                                         preg_match_all( $valueregex, $bits['v'], $vmatches );
1013                                         $opts[$key] = array_map( [ $this, 'cleanupOption' ], $vmatches[0] );
1014                                         if ( count( $opts[$key] ) == 1 ) {
1015                                                 $opts[$key] = $opts[$key][0];
1016                                         }
1017                                 }
1018                         }
1019                 }
1020                 return $opts;
1021         }
1022
1023         private function cleanupOption( $opt ) {
1024                 if ( substr( $opt, 0, 1 ) == '"' ) {
1025                         return stripcslashes( substr( $opt, 1, -1 ) );
1026                 }
1027
1028                 if ( substr( $opt, 0, 2 ) == '[[' ) {
1029                         return substr( $opt, 2, -2 );
1030                 }
1031
1032                 if ( substr( $opt, 0, 1 ) == '{' ) {
1033                         return FormatJson::decode( $opt, true );
1034                 }
1035                 return $opt;
1036         }
1037
1038         /**
1039          * Do any required setup which is dependent on test options.
1040          *
1041          * @see staticSetup() for more information about setup/teardown
1042          *
1043          * @param array $test Test info supplied by TestFileReader
1044          * @param callable|null $nextTeardown
1045          * @return ScopedCallback
1046          */
1047         public function perTestSetup( $test, $nextTeardown = null ) {
1048                 $teardown = [];
1049
1050                 $this->checkSetupDone( 'setupDatabase', 'setDatabase' );
1051                 $teardown[] = $this->markSetupDone( 'perTestSetup' );
1052
1053                 $opts = $this->parseOptions( $test['options'] );
1054                 $config = $test['config'];
1055
1056                 // Find out values for some special options.
1057                 $langCode =
1058                         self::getOptionValue( 'language', $opts, 'en' );
1059                 $variant =
1060                         self::getOptionValue( 'variant', $opts, false );
1061                 $maxtoclevel =
1062                         self::getOptionValue( 'wgMaxTocLevel', $opts, 999 );
1063                 $linkHolderBatchSize =
1064                         self::getOptionValue( 'wgLinkHolderBatchSize', $opts, 1000 );
1065
1066                 // Default to fallback skin, but allow it to be overridden
1067                 $skin = self::getOptionValue( 'skin', $opts, 'fallback' );
1068
1069                 $setup = [
1070                         'wgEnableUploads' => self::getOptionValue( 'wgEnableUploads', $opts, true ),
1071                         'wgLanguageCode' => $langCode,
1072                         'wgRawHtml' => self::getOptionValue( 'wgRawHtml', $opts, false ),
1073                         'wgNamespacesWithSubpages' => array_fill_keys(
1074                                 MWNamespace::getValidNamespaces(), isset( $opts['subpage'] )
1075                         ),
1076                         'wgMaxTocLevel' => $maxtoclevel,
1077                         'wgAllowExternalImages' => self::getOptionValue( 'wgAllowExternalImages', $opts, true ),
1078                         'wgThumbLimits' => [ self::getOptionValue( 'thumbsize', $opts, 180 ) ],
1079                         'wgDefaultLanguageVariant' => $variant,
1080                         'wgLinkHolderBatchSize' => $linkHolderBatchSize,
1081                         // Set as a JSON object like:
1082                         // wgEnableMagicLinks={"ISBN":false, "PMID":false, "RFC":false}
1083                         'wgEnableMagicLinks' => self::getOptionValue( 'wgEnableMagicLinks', $opts, [] )
1084                                 + [ 'ISBN' => true, 'PMID' => true, 'RFC' => true ],
1085                         // Test with legacy encoding by default until HTML5 is very stable and default
1086                         'wgFragmentMode' => [ 'legacy' ],
1087                 ];
1088
1089                 $nonIncludable = self::getOptionValue( 'wgNonincludableNamespaces', $opts, false );
1090                 if ( $nonIncludable !== false ) {
1091                         $setup['wgNonincludableNamespaces'] = [ $nonIncludable ];
1092                 }
1093
1094                 if ( $config ) {
1095                         $configLines = explode( "\n", $config );
1096
1097                         foreach ( $configLines as $line ) {
1098                                 list( $var, $value )  = explode( '=', $line, 2 );
1099                                 $setup[$var] = eval( "return $value;" );
1100                         }
1101                 }
1102
1103                 /** @since 1.20 */
1104                 Hooks::run( 'ParserTestGlobals', [ &$setup ] );
1105
1106                 // Create tidy driver
1107                 if ( isset( $opts['tidy'] ) ) {
1108                         // Cache a driver instance
1109                         if ( $this->tidyDriver === null ) {
1110                                 $this->tidyDriver = MWTidy::factory();
1111                         }
1112                         $tidy = $this->tidyDriver;
1113                 } else {
1114                         $tidy = false;
1115                 }
1116
1117                 # Suppress warnings about running tests without tidy
1118                 Wikimedia\suppressWarnings();
1119                 wfDeprecated( 'disabling tidy' );
1120                 wfDeprecated( 'MWTidy::setInstance' );
1121                 Wikimedia\restoreWarnings();
1122
1123                 MWTidy::setInstance( $tidy );
1124                 $teardown[] = function () {
1125                         MWTidy::destroySingleton();
1126                 };
1127
1128                 // Set content language. This invalidates the magic word cache and title services
1129                 $lang = Language::factory( $langCode );
1130                 $lang->resetNamespaces();
1131                 $setup['wgContLang'] = $lang;
1132                 $setup[] = function () use ( $lang ) {
1133                         MediaWikiServices::getInstance()->disableService( 'ContentLanguage' );
1134                         MediaWikiServices::getInstance()->redefineService(
1135                                 'ContentLanguage',
1136                                 function () use ( $lang ) {
1137                                         return $lang;
1138                                 }
1139                         );
1140                 };
1141                 $teardown[] = function () {
1142                         MediaWikiServices::getInstance()->resetServiceForTesting( 'ContentLanguage' );
1143                 };
1144                 $reset = function () {
1145                         MediaWikiServices::getInstance()->resetServiceForTesting( 'MagicWordFactory' );
1146                         $this->resetTitleServices();
1147                 };
1148                 $setup[] = $reset;
1149                 $teardown[] = $reset;
1150
1151                 // Make a user object with the same language
1152                 $user = new User;
1153                 $user->setOption( 'language', $langCode );
1154                 $setup['wgLang'] = $lang;
1155
1156                 // We (re)set $wgThumbLimits to a single-element array above.
1157                 $user->setOption( 'thumbsize', 0 );
1158
1159                 $setup['wgUser'] = $user;
1160
1161                 // And put both user and language into the context
1162                 $context = RequestContext::getMain();
1163                 $context->setUser( $user );
1164                 $context->setLanguage( $lang );
1165                 // And the skin!
1166                 $oldSkin = $context->getSkin();
1167                 $skinFactory = MediaWikiServices::getInstance()->getSkinFactory();
1168                 $context->setSkin( $skinFactory->makeSkin( $skin ) );
1169                 $context->setOutput( new OutputPage( $context ) );
1170                 $setup['wgOut'] = $context->getOutput();
1171                 $teardown[] = function () use ( $context, $oldSkin ) {
1172                         // Clear language conversion tables
1173                         $wrapper = TestingAccessWrapper::newFromObject(
1174                                 $context->getLanguage()->getConverter()
1175                         );
1176                         $wrapper->reloadTables();
1177                         // Reset context to the restored globals
1178                         $context->setUser( $GLOBALS['wgUser'] );
1179                         $context->setLanguage( $GLOBALS['wgContLang'] );
1180                         $context->setSkin( $oldSkin );
1181                         $context->setOutput( $GLOBALS['wgOut'] );
1182                 };
1183
1184                 $teardown[] = $this->executeSetupSnippets( $setup );
1185
1186                 return $this->createTeardownObject( $teardown, $nextTeardown );
1187         }
1188
1189         /**
1190          * List of temporary tables to create, without prefix.
1191          * Some of these probably aren't necessary.
1192          * @return array
1193          */
1194         private function listTables() {
1195                 global $wgActorTableSchemaMigrationStage;
1196
1197                 $tables = [ 'user', 'user_properties', 'user_former_groups', 'page', 'page_restrictions',
1198                         'protected_titles', 'revision', 'ip_changes', 'text', 'pagelinks', 'imagelinks',
1199                         'categorylinks', 'templatelinks', 'externallinks', 'langlinks', 'iwlinks',
1200                         'site_stats', 'ipblocks', 'image', 'oldimage',
1201                         'recentchanges', 'watchlist', 'interwiki', 'logging', 'log_search',
1202                         'querycache', 'objectcache', 'job', 'l10n_cache', 'redirect', 'querycachetwo',
1203                         'archive', 'user_groups', 'page_props', 'category',
1204                         'slots', 'content', 'slot_roles', 'content_models',
1205                         'comment', 'revision_comment_temp',
1206                 ];
1207
1208                 if ( $wgActorTableSchemaMigrationStage & SCHEMA_COMPAT_WRITE_NEW ) {
1209                         // The new tables for actors are in use
1210                         $tables[] = 'actor';
1211                         $tables[] = 'revision_actor_temp';
1212                 }
1213
1214                 if ( in_array( $this->db->getType(), [ 'mysql', 'sqlite', 'oracle' ] ) ) {
1215                         array_push( $tables, 'searchindex' );
1216                 }
1217
1218                 // Allow extensions to add to the list of tables to duplicate;
1219                 // may be necessary if they hook into page save or other code
1220                 // which will require them while running tests.
1221                 Hooks::run( 'ParserTestTables', [ &$tables ] );
1222
1223                 return $tables;
1224         }
1225
1226         public function setDatabase( IDatabase $db ) {
1227                 $this->db = $db;
1228                 $this->setupDone['setDatabase'] = true;
1229         }
1230
1231         /**
1232          * Set up temporary DB tables.
1233          *
1234          * For best performance, call this once only for all tests. However, it can
1235          * be called at the start of each test if more isolation is desired.
1236          *
1237          * @todo This is basically an unrefactored copy of
1238          * MediaWikiTestCase::setupAllTestDBs. They should be factored out somehow.
1239          *
1240          * Do not call this function from a MediaWikiTestCase subclass, since
1241          * MediaWikiTestCase does its own DB setup. Instead use setDatabase().
1242          *
1243          * @see staticSetup() for more information about setup/teardown
1244          *
1245          * @param ScopedCallback|null $nextTeardown The next teardown object
1246          * @return ScopedCallback The teardown object
1247          */
1248         public function setupDatabase( $nextTeardown = null ) {
1249                 global $wgDBprefix;
1250
1251                 $this->db = wfGetDB( DB_MASTER );
1252                 $dbType = $this->db->getType();
1253
1254                 if ( $dbType == 'oracle' ) {
1255                         $suspiciousPrefixes = [ 'pt_', MediaWikiTestCase::ORA_DB_PREFIX ];
1256                 } else {
1257                         $suspiciousPrefixes = [ 'parsertest_', MediaWikiTestCase::DB_PREFIX ];
1258                 }
1259                 if ( in_array( $wgDBprefix, $suspiciousPrefixes ) ) {
1260                         throw new MWException( "\$wgDBprefix=$wgDBprefix suggests DB setup is already done" );
1261                 }
1262
1263                 $teardown = [];
1264
1265                 $teardown[] = $this->markSetupDone( 'setupDatabase' );
1266
1267                 # CREATE TEMPORARY TABLE breaks if there is more than one server
1268                 if ( MediaWikiServices::getInstance()->getDBLoadBalancer()->getServerCount() != 1 ) {
1269                         $this->useTemporaryTables = false;
1270                 }
1271
1272                 $temporary = $this->useTemporaryTables || $dbType == 'postgres';
1273                 $prefix = $dbType != 'oracle' ? 'parsertest_' : 'pt_';
1274
1275                 $this->dbClone = new CloneDatabase( $this->db, $this->listTables(), $prefix );
1276                 $this->dbClone->useTemporaryTables( $temporary );
1277                 $this->dbClone->cloneTableStructure();
1278                 CloneDatabase::changePrefix( $prefix );
1279
1280                 if ( $dbType == 'oracle' ) {
1281                         $this->db->query( 'BEGIN FILL_WIKI_INFO; END;' );
1282                         # Insert 0 user to prevent FK violations
1283
1284                         # Anonymous user
1285                         $this->db->insert( 'user', [
1286                                 'user_id' => 0,
1287                                 'user_name' => 'Anonymous' ] );
1288                 }
1289
1290                 $teardown[] = function () {
1291                         $this->teardownDatabase();
1292                 };
1293
1294                 // Wipe some DB query result caches on setup and teardown
1295                 $reset = function () {
1296                         MediaWikiServices::getInstance()->getLinkCache()->clear();
1297
1298                         // Clear the message cache
1299                         MessageCache::singleton()->clear();
1300                 };
1301                 $reset();
1302                 $teardown[] = $reset;
1303                 return $this->createTeardownObject( $teardown, $nextTeardown );
1304         }
1305
1306         /**
1307          * Add data about uploads to the new test DB, and set up the upload
1308          * directory. This should be called after either setDatabase() or
1309          * setupDatabase().
1310          *
1311          * @param ScopedCallback|null $nextTeardown The next teardown object
1312          * @return ScopedCallback The teardown object
1313          */
1314         public function setupUploads( $nextTeardown = null ) {
1315                 $teardown = [];
1316
1317                 $this->checkSetupDone( 'setupDatabase', 'setDatabase' );
1318                 $teardown[] = $this->markSetupDone( 'setupUploads' );
1319
1320                 // Create the files in the upload directory (or pretend to create them
1321                 // in a MockFileBackend). Append teardown callback.
1322                 $teardown[] = $this->setupUploadBackend();
1323
1324                 // Create a user
1325                 $user = User::createNew( 'WikiSysop' );
1326
1327                 // Register the uploads in the database
1328
1329                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Foobar.jpg' ) );
1330                 # note that the size/width/height/bits/etc of the file
1331                 # are actually set by inspecting the file itself; the arguments
1332                 # to recordUpload2 have no effect.  That said, we try to make things
1333                 # match up so it is less confusing to readers of the code & tests.
1334                 $image->recordUpload2( '', 'Upload of some lame file', 'Some lame file', [
1335                         'size' => 7881,
1336                         'width' => 1941,
1337                         'height' => 220,
1338                         'bits' => 8,
1339                         'media_type' => MEDIATYPE_BITMAP,
1340                         'mime' => 'image/jpeg',
1341                         'metadata' => serialize( [] ),
1342                         'sha1' => Wikimedia\base_convert( '1', 16, 36, 31 ),
1343                         'fileExists' => true
1344                 ], $this->db->timestamp( '20010115123500' ), $user );
1345
1346                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Thumb.png' ) );
1347                 # again, note that size/width/height below are ignored; see above.
1348                 $image->recordUpload2( '', 'Upload of some lame thumbnail', 'Some lame thumbnail', [
1349                         'size' => 22589,
1350                         'width' => 135,
1351                         'height' => 135,
1352                         'bits' => 8,
1353                         'media_type' => MEDIATYPE_BITMAP,
1354                         'mime' => 'image/png',
1355                         'metadata' => serialize( [] ),
1356                         'sha1' => Wikimedia\base_convert( '2', 16, 36, 31 ),
1357                         'fileExists' => true
1358                 ], $this->db->timestamp( '20130225203040' ), $user );
1359
1360                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Foobar.svg' ) );
1361                 $image->recordUpload2( '', 'Upload of some lame SVG', 'Some lame SVG', [
1362                                 'size'        => 12345,
1363                                 'width'       => 240,
1364                                 'height'      => 180,
1365                                 'bits'        => 0,
1366                                 'media_type'  => MEDIATYPE_DRAWING,
1367                                 'mime'        => 'image/svg+xml',
1368                                 'metadata'    => serialize( [
1369                                         'version'        => SvgHandler::SVG_METADATA_VERSION,
1370                                         'width'          => 240,
1371                                         'height'         => 180,
1372                                         'originalWidth'  => '100%',
1373                                         'originalHeight' => '100%',
1374                                         'translations'   => [
1375                                                 'en' => SVGReader::LANG_FULL_MATCH,
1376                                                 'ru' => SVGReader::LANG_FULL_MATCH,
1377                                         ],
1378                                 ] ),
1379                                 'sha1'        => Wikimedia\base_convert( '', 16, 36, 31 ),
1380                                 'fileExists'  => true
1381                 ], $this->db->timestamp( '20010115123500' ), $user );
1382
1383                 # This image will be blacklisted in [[MediaWiki:Bad image list]]
1384                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Bad.jpg' ) );
1385                 $image->recordUpload2( '', 'zomgnotcensored', 'Borderline image', [
1386                         'size' => 12345,
1387                         'width' => 320,
1388                         'height' => 240,
1389                         'bits' => 24,
1390                         'media_type' => MEDIATYPE_BITMAP,
1391                         'mime' => 'image/jpeg',
1392                         'metadata' => serialize( [] ),
1393                         'sha1' => Wikimedia\base_convert( '3', 16, 36, 31 ),
1394                         'fileExists' => true
1395                 ], $this->db->timestamp( '20010115123500' ), $user );
1396
1397                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Video.ogv' ) );
1398                 $image->recordUpload2( '', 'A pretty movie', 'Will it play', [
1399                         'size' => 12345,
1400                         'width' => 320,
1401                         'height' => 240,
1402                         'bits' => 0,
1403                         'media_type' => MEDIATYPE_VIDEO,
1404                         'mime' => 'application/ogg',
1405                         'metadata' => serialize( [] ),
1406                         'sha1' => Wikimedia\base_convert( '', 16, 36, 31 ),
1407                         'fileExists' => true
1408                 ], $this->db->timestamp( '20010115123500' ), $user );
1409
1410                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'Audio.oga' ) );
1411                 $image->recordUpload2( '', 'An awesome hitsong', 'Will it play', [
1412                         'size' => 12345,
1413                         'width' => 0,
1414                         'height' => 0,
1415                         'bits' => 0,
1416                         'media_type' => MEDIATYPE_AUDIO,
1417                         'mime' => 'application/ogg',
1418                         'metadata' => serialize( [] ),
1419                         'sha1' => Wikimedia\base_convert( '', 16, 36, 31 ),
1420                         'fileExists' => true
1421                 ], $this->db->timestamp( '20010115123500' ), $user );
1422
1423                 # A DjVu file
1424                 $image = wfLocalFile( Title::makeTitle( NS_FILE, 'LoremIpsum.djvu' ) );
1425                 $image->recordUpload2( '', 'Upload a DjVu', 'A DjVu', [
1426                         'size' => 3249,
1427                         'width' => 2480,
1428                         'height' => 3508,
1429                         'bits' => 0,
1430                         'media_type' => MEDIATYPE_BITMAP,
1431                         'mime' => 'image/vnd.djvu',
1432                         'metadata' => '<?xml version="1.0" ?>
1433 <!DOCTYPE DjVuXML PUBLIC "-//W3C//DTD DjVuXML 1.1//EN" "pubtext/DjVuXML-s.dtd">
1434 <DjVuXML>
1435 <HEAD></HEAD>
1436 <BODY><OBJECT height="3508" width="2480">
1437 <PARAM name="DPI" value="300" />
1438 <PARAM name="GAMMA" value="2.2" />
1439 </OBJECT>
1440 <OBJECT height="3508" width="2480">
1441 <PARAM name="DPI" value="300" />
1442 <PARAM name="GAMMA" value="2.2" />
1443 </OBJECT>
1444 <OBJECT height="3508" width="2480">
1445 <PARAM name="DPI" value="300" />
1446 <PARAM name="GAMMA" value="2.2" />
1447 </OBJECT>
1448 <OBJECT height="3508" width="2480">
1449 <PARAM name="DPI" value="300" />
1450 <PARAM name="GAMMA" value="2.2" />
1451 </OBJECT>
1452 <OBJECT height="3508" width="2480">
1453 <PARAM name="DPI" value="300" />
1454 <PARAM name="GAMMA" value="2.2" />
1455 </OBJECT>
1456 </BODY>
1457 </DjVuXML>',
1458                         'sha1' => Wikimedia\base_convert( '', 16, 36, 31 ),
1459                         'fileExists' => true
1460                 ], $this->db->timestamp( '20010115123600' ), $user );
1461
1462                 return $this->createTeardownObject( $teardown, $nextTeardown );
1463         }
1464
1465         /**
1466          * Helper for database teardown, called from the teardown closure. Destroy
1467          * the database clone and fix up some things that CloneDatabase doesn't fix.
1468          *
1469          * @todo Move most things here to CloneDatabase
1470          */
1471         private function teardownDatabase() {
1472                 $this->checkSetupDone( 'setupDatabase' );
1473
1474                 $this->dbClone->destroy();
1475
1476                 if ( $this->useTemporaryTables ) {
1477                         if ( $this->db->getType() == 'sqlite' ) {
1478                                 # Under SQLite the searchindex table is virtual and need
1479                                 # to be explicitly destroyed. See T31912
1480                                 # See also MediaWikiTestCase::destroyDB()
1481                                 wfDebug( __METHOD__ . " explicitly destroying sqlite virtual table parsertest_searchindex\n" );
1482                                 $this->db->query( "DROP TABLE `parsertest_searchindex`" );
1483                         }
1484                         # Don't need to do anything
1485                         return;
1486                 }
1487
1488                 $tables = $this->listTables();
1489
1490                 foreach ( $tables as $table ) {
1491                         if ( $this->db->getType() == 'oracle' ) {
1492                                 $this->db->query( "DROP TABLE pt_$table DROP CONSTRAINTS" );
1493                         } else {
1494                                 $this->db->query( "DROP TABLE `parsertest_$table`" );
1495                         }
1496                 }
1497
1498                 if ( $this->db->getType() == 'oracle' ) {
1499                         $this->db->query( 'BEGIN FILL_WIKI_INFO; END;' );
1500                 }
1501         }
1502
1503         /**
1504          * Upload test files to the backend created by createRepoGroup().
1505          *
1506          * @return callable The teardown callback
1507          */
1508         private function setupUploadBackend() {
1509                 global $IP;
1510
1511                 $repo = RepoGroup::singleton()->getLocalRepo();
1512                 $base = $repo->getZonePath( 'public' );
1513                 $backend = $repo->getBackend();
1514                 $backend->prepare( [ 'dir' => "$base/3/3a" ] );
1515                 $backend->store( [
1516                         'src' => "$IP/tests/phpunit/data/parser/headbg.jpg",
1517                         'dst' => "$base/3/3a/Foobar.jpg"
1518                 ] );
1519                 $backend->prepare( [ 'dir' => "$base/e/ea" ] );
1520                 $backend->store( [
1521                         'src' => "$IP/tests/phpunit/data/parser/wiki.png",
1522                         'dst' => "$base/e/ea/Thumb.png"
1523                 ] );
1524                 $backend->prepare( [ 'dir' => "$base/0/09" ] );
1525                 $backend->store( [
1526                         'src' => "$IP/tests/phpunit/data/parser/headbg.jpg",
1527                         'dst' => "$base/0/09/Bad.jpg"
1528                 ] );
1529                 $backend->prepare( [ 'dir' => "$base/5/5f" ] );
1530                 $backend->store( [
1531                         'src' => "$IP/tests/phpunit/data/parser/LoremIpsum.djvu",
1532                         'dst' => "$base/5/5f/LoremIpsum.djvu"
1533                 ] );
1534
1535                 // No helpful SVG file to copy, so make one ourselves
1536                 $data = '<?xml version="1.0" encoding="utf-8"?>' .
1537                         '<svg xmlns="http://www.w3.org/2000/svg"' .
1538                         ' version="1.1" width="240" height="180"/>';
1539
1540                 $backend->prepare( [ 'dir' => "$base/f/ff" ] );
1541                 $backend->quickCreate( [
1542                         'content' => $data, 'dst' => "$base/f/ff/Foobar.svg"
1543                 ] );
1544
1545                 return function () use ( $backend ) {
1546                         if ( $backend instanceof MockFileBackend ) {
1547                                 // In memory backend, so dont bother cleaning them up.
1548                                 return;
1549                         }
1550                         $this->teardownUploadBackend();
1551                 };
1552         }
1553
1554         /**
1555          * Remove the dummy uploads directory
1556          */
1557         private function teardownUploadBackend() {
1558                 if ( $this->keepUploads ) {
1559                         return;
1560                 }
1561
1562                 $repo = RepoGroup::singleton()->getLocalRepo();
1563                 $public = $repo->getZonePath( 'public' );
1564
1565                 $this->deleteFiles(
1566                         [
1567                                 "$public/3/3a/Foobar.jpg",
1568                                 "$public/e/ea/Thumb.png",
1569                                 "$public/0/09/Bad.jpg",
1570                                 "$public/5/5f/LoremIpsum.djvu",
1571                                 "$public/f/ff/Foobar.svg",
1572                                 "$public/0/00/Video.ogv",
1573                                 "$public/4/41/Audio.oga",
1574                         ]
1575                 );
1576         }
1577
1578         /**
1579          * Delete the specified files and their parent directories
1580          * @param array $files File backend URIs mwstore://...
1581          */
1582         private function deleteFiles( $files ) {
1583                 // Delete the files
1584                 $backend = RepoGroup::singleton()->getLocalRepo()->getBackend();
1585                 foreach ( $files as $file ) {
1586                         $backend->delete( [ 'src' => $file ], [ 'force' => 1 ] );
1587                 }
1588
1589                 // Delete the parent directories
1590                 foreach ( $files as $file ) {
1591                         $tmp = FileBackend::parentStoragePath( $file );
1592                         while ( $tmp ) {
1593                                 if ( !$backend->clean( [ 'dir' => $tmp ] )->isOK() ) {
1594                                         break;
1595                                 }
1596                                 $tmp = FileBackend::parentStoragePath( $tmp );
1597                         }
1598                 }
1599         }
1600
1601         /**
1602          * Add articles to the test DB.
1603          *
1604          * @param array $articles Article info array from TestFileReader
1605          */
1606         public function addArticles( $articles ) {
1607                 $setup = [];
1608                 $teardown = [];
1609
1610                 // Be sure ParserTestRunner::addArticle has correct language set,
1611                 // so that system messages get into the right language cache
1612                 if ( MediaWikiServices::getInstance()->getContentLanguage()->getCode() !== 'en' ) {
1613                         $setup['wgLanguageCode'] = 'en';
1614                         $lang = Language::factory( 'en' );
1615                         $setup['wgContLang'] = $lang;
1616                         $setup[] = function () use ( $lang ) {
1617                                 $services = MediaWikiServices::getInstance();
1618                                 $services->disableService( 'ContentLanguage' );
1619                                 $services->redefineService( 'ContentLanguage', function () use ( $lang ) {
1620                                         return $lang;
1621                                 } );
1622                         };
1623                         $teardown[] = function () {
1624                                 MediaWikiServices::getInstance()->resetServiceForTesting( 'ContentLanguage' );
1625                         };
1626                 }
1627
1628                 // Add special namespaces, in case that hasn't been done by staticSetup() yet
1629                 $this->appendNamespaceSetup( $setup, $teardown );
1630
1631                 // wgCapitalLinks obviously needs initialisation
1632                 $setup['wgCapitalLinks'] = true;
1633
1634                 $teardown[] = $this->executeSetupSnippets( $setup );
1635
1636                 foreach ( $articles as $info ) {
1637                         $this->addArticle( $info['name'], $info['text'], $info['file'], $info['line'] );
1638                 }
1639
1640                 // Wipe WANObjectCache process cache, which is invalidated by article insertion
1641                 // due to T144706
1642                 ObjectCache::getMainWANInstance()->clearProcessCache();
1643
1644                 $this->executeSetupSnippets( $teardown );
1645         }
1646
1647         /**
1648          * Insert a temporary test article
1649          * @param string $name The title, including any prefix
1650          * @param string $text The article text
1651          * @param string $file The input file name
1652          * @param int|string $line The input line number, for reporting errors
1653          * @throws Exception
1654          * @throws MWException
1655          */
1656         private function addArticle( $name, $text, $file, $line ) {
1657                 $text = self::chomp( $text );
1658                 $name = self::chomp( $name );
1659
1660                 $title = Title::newFromText( $name );
1661                 wfDebug( __METHOD__ . ": adding $name" );
1662
1663                 if ( is_null( $title ) ) {
1664                         throw new MWException( "invalid title '$name' at $file:$line\n" );
1665                 }
1666
1667                 $newContent = ContentHandler::makeContent( $text, $title );
1668
1669                 $page = WikiPage::factory( $title );
1670                 $page->loadPageData( 'fromdbmaster' );
1671
1672                 if ( $page->exists() ) {
1673                         $content = $page->getContent( Revision::RAW );
1674                         // Only reject the title, if the content/content model is different.
1675                         // This makes it easier to create Template:(( or Template:)) in different extensions
1676                         if ( $newContent->equals( $content ) ) {
1677                                 return;
1678                         }
1679                         throw new MWException(
1680                                 "duplicate article '$name' with different content at $file:$line\n"
1681                         );
1682                 }
1683
1684                 // Optionally use mock parser, to make debugging of actual parser tests simpler.
1685                 // But initialise the MessageCache clone first, don't let MessageCache
1686                 // get a reference to the mock object.
1687                 if ( $this->disableSaveParse ) {
1688                         MessageCache::singleton()->getParser();
1689                         $restore = $this->executeSetupSnippets( [ 'wgParser' => new ParserTestMockParser ] );
1690                 } else {
1691                         $restore = false;
1692                 }
1693                 try {
1694                         $status = $page->doEditContent(
1695                                 $newContent,
1696                                 '',
1697                                 EDIT_NEW | EDIT_INTERNAL
1698                         );
1699                 } finally {
1700                         if ( $restore ) {
1701                                 $restore();
1702                         }
1703                 }
1704
1705                 if ( !$status->isOK() ) {
1706                         throw new MWException( $status->getWikiText( false, false, 'en' ) );
1707                 }
1708
1709                 // The RepoGroup cache is invalidated by the creation of file redirects
1710                 if ( $title->inNamespace( NS_FILE ) ) {
1711                         RepoGroup::singleton()->clearCache( $title );
1712                 }
1713         }
1714
1715         /**
1716          * Check if a hook is installed
1717          *
1718          * @param string $name
1719          * @return bool True if tag hook is present
1720          */
1721         public function requireHook( $name ) {
1722                 global $wgParser;
1723
1724                 $wgParser->firstCallInit(); // make sure hooks are loaded.
1725                 if ( isset( $wgParser->mTagHooks[$name] ) ) {
1726                         return true;
1727                 } else {
1728                         $this->recorder->warning( "   This test suite requires the '$name' hook " .
1729                                 "extension, skipping." );
1730                         return false;
1731                 }
1732         }
1733
1734         /**
1735          * Check if a function hook is installed
1736          *
1737          * @param string $name
1738          * @return bool True if function hook is present
1739          */
1740         public function requireFunctionHook( $name ) {
1741                 global $wgParser;
1742
1743                 $wgParser->firstCallInit(); // make sure hooks are loaded.
1744
1745                 if ( isset( $wgParser->mFunctionHooks[$name] ) ) {
1746                         return true;
1747                 } else {
1748                         $this->recorder->warning( "   This test suite requires the '$name' function " .
1749                                 "hook extension, skipping." );
1750                         return false;
1751                 }
1752         }
1753
1754         /**
1755          * Check if a transparent tag hook is installed
1756          *
1757          * @param string $name
1758          * @return bool True if function hook is present
1759          */
1760         public function requireTransparentHook( $name ) {
1761                 global $wgParser;
1762
1763                 $wgParser->firstCallInit(); // make sure hooks are loaded.
1764
1765                 if ( isset( $wgParser->mTransparentTagHooks[$name] ) ) {
1766                         return true;
1767                 } else {
1768                         $this->recorder->warning( "   This test suite requires the '$name' transparent " .
1769                                 "hook extension, skipping.\n" );
1770                         return false;
1771                 }
1772         }
1773
1774         /**
1775          * Fake constant timestamp to make sure time-related parser
1776          * functions give a persistent value.
1777          *
1778          * - Parser::getVariableValue (via ParserGetVariableValueTs hook)
1779          * - Parser::preSaveTransform (via ParserOptions)
1780          */
1781         private function getFakeTimestamp() {
1782                 // parsed as '1970-01-01T00:02:03Z'
1783                 return 123;
1784         }
1785 }