maintenance/tests/testHelpers.inc

   1 <?php
   2
   3 /**
   4  * @ingroup Maintenance
   5  *
   6  * Set of classes to help with test output and such. Right now pretty specific
   7  * to the parser tests but could be more useful one day :)
   8  *
   9  * @todo Fixme: Make this more generic
  10  */
  11
  12 class AnsiTermColorer {
  13         function __construct() {
  14         }
  15
  16         /**
  17          * Return ANSI terminal escape code for changing text attribs/color
  18          *
  19          * @param $color String: semicolon-separated list of attribute/color codes
  20          * @return String
  21          */
  22         public function color( $color ) {
  23                 global $wgCommandLineDarkBg;
  24
  25                 $light = $wgCommandLineDarkBg ? "1;" : "0;";
  26
  27                 return "\x1b[{$light}{$color}m";
  28         }
  29
  30         /**
  31          * Return ANSI terminal escape code for restoring default text attributes
  32          *
  33          * @return String
  34          */
  35         public function reset() {
  36                 return $this->color( 0 );
  37         }
  38 }
  39
  40 /* A colour-less terminal */
  41 class DummyTermColorer {
  42         public function color( $color ) {
  43                 return '';
  44         }
  45
  46         public function reset() {
  47                 return '';
  48         }
  49 }
  50
  51 class TestRecorder {
  52         var $parent;
  53         var $term;
  54
  55         function __construct( $parent ) {
  56                 $this->parent = $parent;
  57                 $this->term = $parent->term;
  58         }
  59
  60         function start() {
  61                 $this->total = 0;
  62                 $this->success = 0;
  63         }
  64
  65         function record( $test, $result ) {
  66                 $this->total++;
  67                 $this->success += ( $result ? 1 : 0 );
  68         }
  69
  70         function end() {
  71                 // dummy
  72         }
  73
  74         function report() {
  75                 if ( $this->total > 0 ) {
  76                         $this->reportPercentage( $this->success, $this->total );
  77                 } else {
  78                         wfDie( "No tests found.\n" );
  79                 }
  80         }
  81
  82         function reportPercentage( $success, $total ) {
  83                 $ratio = wfPercent( 100 * $success / $total );
  84                 print $this->term->color( 1 ) . "Passed $success of $total tests ($ratio)... ";
  85
  86                 if ( $success == $total ) {
  87                         print $this->term->color( 32 ) . "ALL TESTS PASSED!";
  88                 } else {
  89                         $failed = $total - $success ;
  90                         print $this->term->color( 31 ) . "$failed tests failed!";
  91                 }
  92
  93                 print $this->term->reset() . "\n";
  94
  95                 return ( $success == $total );
  96         }
  97 }
  98
  99 class DbTestPreviewer extends TestRecorder  {
 100         protected $lb;      // /< Database load balancer
 101         protected $db;      // /< Database connection to the main DB
 102         protected $curRun;  // /< run ID number for the current run
 103         protected $prevRun; // /< run ID number for the previous run, if any
 104         protected $results; // /< Result array
 105
 106         /**
 107          * This should be called before the table prefix is changed
 108          */
 109         function __construct( $parent ) {
 110                 parent::__construct( $parent );
 111
 112                 $this->lb = wfGetLBFactory()->newMainLB();
 113                 // This connection will have the wiki's table prefix, not parsertest_
 114                 $this->db = $this->lb->getConnection( DB_MASTER );
 115         }
 116
 117         /**
 118          * Set up result recording; insert a record for the run with the date
 119          * and all that fun stuff
 120          */
 121         function start() {
 122                 parent::start();
 123
 124                 if ( ! $this->db->tableExists( 'testrun' )
 125                         or ! $this->db->tableExists( 'testitem' ) )
 126                 {
 127                         print "WARNING> `testrun` table not found in database.\n";
 128                         $this->prevRun = false;
 129                 } else {
 130                         // We'll make comparisons against the previous run later...
 131                         $this->prevRun = $this->db->selectField( 'testrun', 'MAX(tr_id)' );
 132                 }
 133
 134                 $this->results = array();
 135         }
 136
 137         function record( $test, $result ) {
 138                 parent::record( $test, $result );
 139                 $this->results[$test] = $result;
 140         }
 141
 142         function report() {
 143                 if ( $this->prevRun ) {
 144                         // f = fail, p = pass, n = nonexistent
 145                         // codes show before then after
 146                         $table = array(
 147                                 'fp' => 'previously failing test(s) now PASSING! :)',
 148                                 'pn' => 'previously PASSING test(s) removed o_O',
 149                                 'np' => 'new PASSING test(s) :)',
 150
 151                                 'pf' => 'previously passing test(s) now FAILING! :(',
 152                                 'fn' => 'previously FAILING test(s) removed O_o',
 153                                 'nf' => 'new FAILING test(s) :(',
 154                                 'ff' => 'still FAILING test(s) :(',
 155                         );
 156
 157                         $prevResults = array();
 158
 159                         $res = $this->db->select( 'testitem', array( 'ti_name', 'ti_success' ),
 160                                 array( 'ti_run' => $this->prevRun ), __METHOD__ );
 161
 162                         foreach ( $res as $row ) {
 163                                 if ( !$this->parent->regex
 164                                         || preg_match( "/{$this->parent->regex}/i", $row->ti_name ) )
 165                                 {
 166                                         $prevResults[$row->ti_name] = $row->ti_success;
 167                                 }
 168                         }
 169
 170                         $combined = array_keys( $this->results + $prevResults );
 171
 172                         # Determine breakdown by change type
 173                         $breakdown = array();
 174                         foreach ( $combined as $test ) {
 175                                 if ( !isset( $prevResults[$test] ) ) {
 176                                         $before = 'n';
 177                                 } elseif ( $prevResults[$test] == 1 ) {
 178                                         $before = 'p';
 179                                 } else /* if ( $prevResults[$test] == 0 )*/ {
 180                                         $before = 'f';
 181                                 }
 182
 183                                 if ( !isset( $this->results[$test] ) ) {
 184                                         $after = 'n';
 185                                 } elseif ( $this->results[$test] == 1 ) {
 186                                         $after = 'p';
 187                                 } else /*if ( $this->results[$test] == 0 ) */ {
 188                                         $after = 'f';
 189                                 }
 190
 191                                 $code = $before . $after;
 192
 193                                 if ( isset( $table[$code] ) ) {
 194                                         $breakdown[$code][$test] = $this->getTestStatusInfo( $test, $after );
 195                                 }
 196                         }
 197
 198                         # Write out results
 199                         foreach ( $table as $code => $label ) {
 200                                 if ( !empty( $breakdown[$code] ) ) {
 201                                         $count = count( $breakdown[$code] );
 202                                         printf( "\n%4d %s\n", $count, $label );
 203
 204                                         foreach ( $breakdown[$code] as $differing_test_name => $statusInfo ) {
 205                                                 print "      * $differing_test_name  [$statusInfo]\n";
 206                                         }
 207                                 }
 208                         }
 209                 } else {
 210                         print "No previous test runs to compare against.\n";
 211                 }
 212
 213                 print "\n";
 214                 parent::report();
 215         }
 216
 217         /**
 218          * Returns a string giving information about when a test last had a status change.
 219          * Could help to track down when regressions were introduced, as distinct from tests
 220          * which have never passed (which are more change requests than regressions).
 221          */
 222         private function getTestStatusInfo( $testname, $after ) {
 223                 // If we're looking at a test that has just been removed, then say when it first appeared.
 224                 if ( $after == 'n' ) {
 225                         $changedRun = $this->db->selectField ( 'testitem',
 226                                 'MIN(ti_run)',
 227                                 array( 'ti_name' => $testname ),
 228                                 __METHOD__ );
 229                         $appear = $this->db->selectRow ( 'testrun',
 230                                 array( 'tr_date', 'tr_mw_version' ),
 231                                 array( 'tr_id' => $changedRun ),
 232                                 __METHOD__ );
 233
 234                         return "First recorded appearance: "
 235                                    . date( "d-M-Y H:i:s",  strtotime ( $appear->tr_date ) )
 236                                    .  ", " . $appear->tr_mw_version;
 237                 }
 238
 239                 // Otherwise, this test has previous recorded results.
 240                 // See when this test last had a different result to what we're seeing now.
 241                 $conds = array(
 242                         'ti_name'    => $testname,
 243                         'ti_success' => ( $after == 'f' ? "1" : "0" ) );
 244
 245                 if ( $this->curRun ) {
 246                         $conds[] = "ti_run != " . $this->db->addQuotes ( $this->curRun );
 247                 }
 248
 249                 $changedRun = $this->db->selectField ( 'testitem', 'MAX(ti_run)', $conds, __METHOD__ );
 250
 251                 // If no record of ever having had a different result.
 252                 if ( is_null ( $changedRun ) ) {
 253                         if ( $after == "f" ) {
 254                                 return "Has never passed";
 255                         } else {
 256                                 return "Has never failed";
 257                         }
 258                 }
 259
 260                 // Otherwise, we're looking at a test whose status has changed.
 261                 // (i.e. it used to work, but now doesn't; or used to fail, but is now fixed.)
 262                 // In this situation, give as much info as we can as to when it changed status.
 263                 $pre  = $this->db->selectRow ( 'testrun',
 264                         array( 'tr_date', 'tr_mw_version' ),
 265                         array( 'tr_id' => $changedRun ),
 266                         __METHOD__ );
 267                 $post = $this->db->selectRow ( 'testrun',
 268                         array( 'tr_date', 'tr_mw_version' ),
 269                         array( "tr_id > " . $this->db->addQuotes ( $changedRun ) ),
 270                         __METHOD__,
 271                         array( "LIMIT" => 1, "ORDER BY" => 'tr_id' )
 272                 );
 273
 274                 if ( $post ) {
 275                         $postDate = date( "d-M-Y H:i:s",  strtotime ( $post->tr_date  ) ) . ", {$post->tr_mw_version}";
 276                 } else {
 277                         $postDate = 'now';
 278                 }
 279
 280                 return ( $after == "f" ? "Introduced" : "Fixed" ) . " between "
 281                                 . date( "d-M-Y H:i:s",  strtotime ( $pre->tr_date ) ) .  ", " . $pre->tr_mw_version
 282                                 . " and $postDate";
 283
 284         }
 285
 286         /**
 287          * Commit transaction and clean up for result recording
 288          */
 289         function end() {
 290                 $this->lb->commitMasterChanges();
 291                 $this->lb->closeAll();
 292                 parent::end();
 293         }
 294
 295 }
 296
 297 class DbTestRecorder extends DbTestPreviewer  {
 298         var $version;
 299
 300         /**
 301          * Set up result recording; insert a record for the run with the date
 302          * and all that fun stuff
 303          */
 304         function start() {
 305                 global $wgDBtype;
 306                 $this->db->begin();
 307
 308                 if ( ! $this->db->tableExists( 'testrun' )
 309                         or ! $this->db->tableExists( 'testitem' ) )
 310                 {
 311                         print "WARNING> `testrun` table not found in database. Trying to create table.\n";
 312                         $this->db->sourceFile( $this->db->patchPath( 'patch-testrun.sql' ) );
 313                         echo "OK, resuming.\n";
 314                 }
 315
 316                 parent::start();
 317
 318                 $this->db->insert( 'testrun',
 319                         array(
 320                                 'tr_date'        => $this->db->timestamp(),
 321                                 'tr_mw_version'  => $this->version,
 322                                 'tr_php_version' => phpversion(),
 323                                 'tr_db_version'  => $this->db->getServerVersion(),
 324                                 'tr_uname'       => php_uname()
 325                         ),
 326                         __METHOD__ );
 327                         if ( $wgDBtype === 'postgres' ) {
 328                                 $this->curRun = $this->db->currentSequenceValue( 'testrun_id_seq' );
 329                         } else {
 330                                 $this->curRun = $this->db->insertId();
 331                         }
 332         }
 333
 334         /**
 335          * Record an individual test item's success or failure to the db
 336          *
 337          * @param $test String
 338          * @param $result Boolean
 339          */
 340         function record( $test, $result ) {
 341                 parent::record( $test, $result );
 342
 343                 $this->db->insert( 'testitem',
 344                         array(
 345                                 'ti_run'     => $this->curRun,
 346                                 'ti_name'    => $test,
 347                                 'ti_success' => $result ? 1 : 0,
 348                         ),
 349                         __METHOD__ );
 350         }
 351 }
 352
 353 class RemoteTestRecorder extends TestRecorder {
 354         function start() {
 355                 parent::start();
 356
 357                 $this->results = array();
 358                 $this->ping( 'running' );
 359         }
 360
 361         function record( $test, $result ) {
 362                 parent::record( $test, $result );
 363                 $this->results[$test] = (bool)$result;
 364         }
 365
 366         function end() {
 367                 $this->ping( 'complete', $this->results );
 368                 parent::end();
 369         }
 370
 371         /**
 372          * Inform a CodeReview instance that we've started or completed a test run...
 373          *
 374          * @param $status string: "running" - tell it we've started
 375          *                        "complete" - provide test results array
 376          *                        "abort" - something went horribly awry
 377          * @param $results array of test name => true/false
 378          */
 379         function ping( $status, $results = false ) {
 380                 global $wgParserTestRemote, $IP;
 381
 382                 $remote = $wgParserTestRemote;
 383                 $revId = SpecialVersion::getSvnRevision( $IP );
 384                 $jsonResults = FormatJson::encode( $results );
 385
 386                 if ( !$remote ) {
 387                         print "Can't do remote upload without configuring \$wgParserTestRemote!\n";
 388                         exit( 1 );
 389                 }
 390
 391                 // Generate a hash MAC to validate our credentials
 392                 $message = array(
 393                         $remote['repo'],
 394                         $remote['suite'],
 395                         $revId,
 396                         $status,
 397                 );
 398
 399                 if ( $status == "complete" ) {
 400                         $message[] = $jsonResults;
 401                 }
 402                 $hmac = hash_hmac( "sha1", implode( "|", $message ), $remote['secret'] );
 403
 404                 $postData = array(
 405                         'action' => 'codetestupload',
 406                         'format' => 'json',
 407                         'repo'   => $remote['repo'],
 408                         'suite'  => $remote['suite'],
 409                         'rev'    => $revId,
 410                         'status' => $status,
 411                         'hmac'   => $hmac,
 412                 );
 413
 414                 if ( $status == "complete" ) {
 415                         $postData['results'] = $jsonResults;
 416                 }
 417
 418                 $response = $this->post( $remote['api-url'], $postData );
 419
 420                 if ( $response === false ) {
 421                         print "CodeReview info upload failed to reach server.\n";
 422                         exit( 1 );
 423                 }
 424
 425                 $responseData = FormatJson::decode( $response, true );
 426
 427                 if ( !is_array( $responseData ) ) {
 428                         print "CodeReview API response not recognized...\n";
 429                         wfDebug( "Unrecognized CodeReview API response: $response\n" );
 430                         exit( 1 );
 431                 }
 432
 433                 if ( isset( $responseData['error'] ) ) {
 434                         $code = $responseData['error']['code'];
 435                         $info = $responseData['error']['info'];
 436                         print "CodeReview info upload failed: $code $info\n";
 437                         exit( 1 );
 438                 }
 439         }
 440
 441         function post( $url, $data ) {
 442                 return Http::post( $url, array( 'postData' => $data ) );
 443         }
 444 }
 445
 446 class TestFileIterator implements Iterator {
 447         private $file;
 448         private $fh;
 449         private $parserTest; /* An instance of ParserTest (parserTests.php) or MediaWikiParserTest (phpunit) */
 450         private $index = 0;
 451         private $test;
 452         private $lineNum;
 453         private $eof;
 454
 455         function __construct( $file, $parserTest = null ) {
 456                 global $IP;
 457
 458                 $this->file = $file;
 459                 $this->fh = fopen( $this->file, "rt" );
 460
 461                 if ( !$this->fh ) {
 462                         wfDie( "Couldn't open file '$file'\n" );
 463                 }
 464
 465                 $this->parserTest = $parserTest;
 466
 467                 if ( $this->parserTest ) {
 468                         $this->parserTest->showRunFile( wfRelativePath( $this->file, $IP ) );
 469                 }
 470
 471                 $this->lineNum = $this->index = 0;
 472         }
 473
 474         function rewind() {
 475                 if ( fseek( $this->fh, 0 ) ) {
 476                         wfDie( "Couldn't fseek to the start of '$this->file'\n" );
 477                 }
 478
 479                 $this->index = -1;
 480                 $this->lineNum = 0;
 481                 $this->eof = false;
 482                 $this->next();
 483
 484                 return true;
 485         }
 486
 487         function current() {
 488                 return $this->test;
 489         }
 490
 491         function key() {
 492                 return $this->index;
 493         }
 494
 495         function next() {
 496                 if ( $this->readNextTest() ) {
 497                         $this->index++;
 498                         return true;
 499                 } else {
 500                         $this->eof = true;
 501                 }
 502         }
 503
 504         function valid() {
 505                 return $this->eof != true;
 506         }
 507
 508         function readNextTest() {
 509                 $data = array();
 510                 $section = null;
 511
 512                 while ( false !== ( $line = fgets( $this->fh ) ) ) {
 513                         $this->lineNum++;
 514                         $matches = array();
 515
 516                         if ( preg_match( '/^!!\s*(\w+)/', $line, $matches ) ) {
 517                                 $section = strtolower( $matches[1] );
 518
 519                                 if ( $section == 'endarticle' ) {
 520                                         if ( !isset( $data['text'] ) ) {
 521                                                 wfDie( "'endarticle' without 'text' at line {$this->lineNum} of $this->file\n" );
 522                                         }
 523
 524                                         if ( !isset( $data['article'] ) ) {
 525                                                 wfDie( "'endarticle' without 'article' at line {$this->lineNum} of $this->file\n" );
 526                                         }
 527
 528                                         if ( $this->parserTest ) {
 529                                                 $this->parserTest->addArticle( ParserTest::chomp( $data['article'] ), $data['text'], $this->lineNum );
 530                                         } else {wfDie("JAJA");
 531                                                 ParserTest::addArticle( $data['article'], $data['text'], $this->lineNum );
 532                                         }
 533                                         $data = array();
 534                                         $section = null;
 535
 536                                         continue;
 537                                 }
 538
 539                                 if ( $section == 'endhooks' ) {
 540                                         if ( !isset( $data['hooks'] ) ) {
 541                                                 wfDie( "'endhooks' without 'hooks' at line {$this->lineNum} of $this->file\n" );
 542                                         }
 543
 544                                         foreach ( explode( "\n", $data['hooks'] ) as $line ) {
 545                                                 $line = trim( $line );
 546
 547                                                 if ( $line ) {
 548                                                         if ( $this->parserTest && !$this->parserTest->requireHook( $line ) ) {
 549                                                                 return false;
 550                                                         }
 551                                                 }
 552                                         }
 553
 554                                         $data = array();
 555                                         $section = null;
 556
 557                                         continue;
 558                                 }
 559
 560                                 if ( $section == 'endfunctionhooks' ) {
 561                                         if ( !isset( $data['functionhooks'] ) ) {
 562                                                 wfDie( "'endfunctionhooks' without 'functionhooks' at line {$this->lineNum} of $this->file\n" );
 563                                         }
 564
 565                                         foreach ( explode( "\n", $data['functionhooks'] ) as $line ) {
 566                                                 $line = trim( $line );
 567
 568                                                 if ( $line ) {
 569                                                         if ( $this->parserTest && !$this->parserTest->requireFunctionHook( $line ) ) {
 570                                                                 return false;
 571                                                         }
 572                                                 }
 573                                         }
 574
 575                                         $data = array();
 576                                         $section = null;
 577
 578                                         continue;
 579                                 }
 580
 581                                 if ( $section == 'end' ) {
 582                                         if ( !isset( $data['test'] ) ) {
 583                                                 wfDie( "'end' without 'test' at line {$this->lineNum} of $this->file\n" );
 584                                         }
 585
 586                                         if ( !isset( $data['input'] ) ) {
 587                                                 wfDie( "'end' without 'input' at line {$this->lineNum} of $this->file\n" );
 588                                         }
 589
 590                                         if ( !isset( $data['result'] ) ) {
 591                                                 wfDie( "'end' without 'result' at line {$this->lineNum} of $this->file\n" );
 592                                         }
 593
 594                                         if ( !isset( $data['options'] ) ) {
 595                                                 $data['options'] = '';
 596                                         }
 597
 598                                         if ( !isset( $data['config'] ) )
 599                                                 $data['config'] = '';
 600
 601                                         if ( $this->parserTest
 602                                                  && ( ( preg_match( '/\\bdisabled\\b/i', $data['options'] ) && !$this->parserTest->runDisabled )
 603                                                          || !preg_match( "/" . $this->parserTest->regex . "/i", $data['test'] ) )  ) {
 604                                                 # disabled test
 605                                                 $data = array();
 606                                                 $section = null;
 607
 608                                                 continue;
 609                                         }
 610
 611                                         global $wgUseTeX;
 612
 613                                         if ( $this->parserTest &&
 614                                                  preg_match( '/\\bmath\\b/i', $data['options'] ) && !$wgUseTeX ) {
 615                                                 # don't run math tests if $wgUseTeX is set to false in LocalSettings
 616                                                 $data = array();
 617                                                 $section = null;
 618
 619                                                 continue;
 620                                         }
 621
 622                                         if ( $this->parserTest ) {
 623                                                 $this->test = array(
 624                                                         'test' => ParserTest::chomp( $data['test'] ),
 625                                                         'input' => ParserTest::chomp( $data['input'] ),
 626                                                         'result' => ParserTest::chomp( $data['result'] ),
 627                                                         'options' => ParserTest::chomp( $data['options'] ),
 628                                                         'config' => ParserTest::chomp( $data['config'] ) );
 629                                         } else {
 630                                                 $this->test['test'] = $data['test'];
 631                                         }
 632
 633                                         return true;
 634                                 }
 635
 636                                 if ( isset ( $data[$section] ) ) {
 637                                         wfDie( "duplicate section '$section' at line {$this->lineNum} of $this->file\n" );
 638                                 }
 639
 640                                 $data[$section] = '';
 641
 642                                 continue;
 643                         }
 644
 645                         if ( $section ) {
 646                                 $data[$section] .= $line;
 647                         }
 648                 }
 649
 650                 return false;
 651         }
 652 }