tests/testHelpers.inc

   1 <?php
   2
   3 /**
   4  * @ingroup Testing
   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                 $this->db->begin();
 306
 307                 if ( ! $this->db->tableExists( 'testrun' )
 308                         or ! $this->db->tableExists( 'testitem' ) )
 309                 {
 310                         print "WARNING> `testrun` table not found in database. Trying to create table.\n";
 311                         $this->db->sourceFile( $this->db->patchPath( 'patch-testrun.sql' ) );
 312                         echo "OK, resuming.\n";
 313                 }
 314
 315                 parent::start();
 316
 317                 $this->db->insert( 'testrun',
 318                         array(
 319                                 'tr_date'        => $this->db->timestamp(),
 320                                 'tr_mw_version'  => $this->version,
 321                                 'tr_php_version' => phpversion(),
 322                                 'tr_db_version'  => $this->db->getServerVersion(),
 323                                 'tr_uname'       => php_uname()
 324                         ),
 325                         __METHOD__ );
 326                         if ( $this->db->getType() === 'postgres' ) {
 327                                 $this->curRun = $this->db->currentSequenceValue( 'testrun_id_seq' );
 328                         } else {
 329                                 $this->curRun = $this->db->insertId();
 330                         }
 331         }
 332
 333         /**
 334          * Record an individual test item's success or failure to the db
 335          *
 336          * @param $test String
 337          * @param $result Boolean
 338          */
 339         function record( $test, $result ) {
 340                 parent::record( $test, $result );
 341
 342                 $this->db->insert( 'testitem',
 343                         array(
 344                                 'ti_run'     => $this->curRun,
 345                                 'ti_name'    => $test,
 346                                 'ti_success' => $result ? 1 : 0,
 347                         ),
 348                         __METHOD__ );
 349         }
 350 }
 351
 352 class RemoteTestRecorder extends TestRecorder {
 353         function start() {
 354                 parent::start();
 355
 356                 $this->results = array();
 357                 $this->ping( 'running' );
 358         }
 359
 360         function record( $test, $result ) {
 361                 parent::record( $test, $result );
 362                 $this->results[$test] = (bool)$result;
 363         }
 364
 365         function end() {
 366                 $this->ping( 'complete', $this->results );
 367                 parent::end();
 368         }
 369
 370         /**
 371          * Inform a CodeReview instance that we've started or completed a test run...
 372          *
 373          * @param $status string: "running" - tell it we've started
 374          *                        "complete" - provide test results array
 375          *                        "abort" - something went horribly awry
 376          * @param $results array of test name => true/false
 377          */
 378         function ping( $status, $results = false ) {
 379                 global $wgParserTestRemote, $IP;
 380
 381                 $remote = $wgParserTestRemote;
 382                 $revId = SpecialVersion::getSvnRevision( $IP );
 383                 $jsonResults = FormatJson::encode( $results );
 384
 385                 if ( !$remote ) {
 386                         print "Can't do remote upload without configuring \$wgParserTestRemote!\n";
 387                         exit( 1 );
 388                 }
 389
 390                 // Generate a hash MAC to validate our credentials
 391                 $message = array(
 392                         $remote['repo'],
 393                         $remote['suite'],
 394                         $revId,
 395                         $status,
 396                 );
 397
 398                 if ( $status == "complete" ) {
 399                         $message[] = $jsonResults;
 400                 }
 401                 $hmac = hash_hmac( "sha1", implode( "|", $message ), $remote['secret'] );
 402
 403                 $postData = array(
 404                         'action' => 'codetestupload',
 405                         'format' => 'json',
 406                         'repo'   => $remote['repo'],
 407                         'suite'  => $remote['suite'],
 408                         'rev'    => $revId,
 409                         'status' => $status,
 410                         'hmac'   => $hmac,
 411                 );
 412
 413                 if ( $status == "complete" ) {
 414                         $postData['results'] = $jsonResults;
 415                 }
 416
 417                 $response = $this->post( $remote['api-url'], $postData );
 418
 419                 if ( $response === false ) {
 420                         print "CodeReview info upload failed to reach server.\n";
 421                         exit( 1 );
 422                 }
 423
 424                 $responseData = FormatJson::decode( $response, true );
 425
 426                 if ( !is_array( $responseData ) ) {
 427                         print "CodeReview API response not recognized...\n";
 428                         wfDebug( "Unrecognized CodeReview API response: $response\n" );
 429                         exit( 1 );
 430                 }
 431
 432                 if ( isset( $responseData['error'] ) ) {
 433                         $code = $responseData['error']['code'];
 434                         $info = $responseData['error']['info'];
 435                         print "CodeReview info upload failed: $code $info\n";
 436                         exit( 1 );
 437                 }
 438         }
 439
 440         function post( $url, $data ) {
 441                 return Http::post( $url, array( 'postData' => $data ) );
 442         }
 443 }
 444
 445 class TestFileIterator implements Iterator {
 446         private $file;
 447         private $fh;
 448         private $parserTest; /* An instance of ParserTest (parserTests.php) or MediaWikiParserTest (phpunit) */
 449         private $index = 0;
 450         private $test;
 451         private $lineNum;
 452         private $eof;
 453
 454         function __construct( $file, $parserTest ) {
 455                 global $IP;
 456
 457                 $this->file = $file;
 458                 $this->fh = fopen( $this->file, "rt" );
 459
 460                 if ( !$this->fh ) {
 461                         wfDie( "Couldn't open file '$file'\n" );
 462                 }
 463
 464                 $this->parserTest = $parserTest;
 465                 $this->parserTest->showRunFile( wfRelativePath( $this->file, $IP ) );
 466
 467                 $this->lineNum = $this->index = 0;
 468         }
 469
 470         function rewind() {
 471                 if ( fseek( $this->fh, 0 ) ) {
 472                         wfDie( "Couldn't fseek to the start of '$this->file'\n" );
 473                 }
 474
 475                 $this->index = -1;
 476                 $this->lineNum = 0;
 477                 $this->eof = false;
 478                 $this->next();
 479
 480                 return true;
 481         }
 482
 483         function current() {
 484                 return $this->test;
 485         }
 486
 487         function key() {
 488                 return $this->index;
 489         }
 490
 491         function next() {
 492                 if ( $this->readNextTest() ) {
 493                         $this->index++;
 494                         return true;
 495                 } else {
 496                         $this->eof = true;
 497                 }
 498         }
 499
 500         function valid() {
 501                 return $this->eof != true;
 502         }
 503
 504         function readNextTest() {
 505                 $data = array();
 506                 $section = null;
 507
 508                 while ( false !== ( $line = fgets( $this->fh ) ) ) {
 509                         $this->lineNum++;
 510                         $matches = array();
 511
 512                         if ( preg_match( '/^!!\s*(\w+)/', $line, $matches ) ) {
 513                                 $section = strtolower( $matches[1] );
 514
 515                                 if ( $section == 'endarticle' ) {
 516                                         if ( !isset( $data['text'] ) ) {
 517                                                 wfDie( "'endarticle' without 'text' at line {$this->lineNum} of $this->file\n" );
 518                                         }
 519
 520                                         if ( !isset( $data['article'] ) ) {
 521                                                 wfDie( "'endarticle' without 'article' at line {$this->lineNum} of $this->file\n" );
 522                                         }
 523
 524                                         $this->parserTest->addArticle( ParserTest::chomp( $data['article'] ), $data['text'], $this->lineNum );
 525
 526                                         $data = array();
 527                                         $section = null;
 528
 529                                         continue;
 530                                 }
 531
 532                                 if ( $section == 'endhooks' ) {
 533                                         if ( !isset( $data['hooks'] ) ) {
 534                                                 wfDie( "'endhooks' without 'hooks' at line {$this->lineNum} of $this->file\n" );
 535                                         }
 536
 537                                         foreach ( explode( "\n", $data['hooks'] ) as $line ) {
 538                                                 $line = trim( $line );
 539
 540                                                 if ( $line ) {
 541                                                         if ( !$this->parserTest->requireHook( $line ) ) {
 542                                                                 return false;
 543                                                         }
 544                                                 }
 545                                         }
 546
 547                                         $data = array();
 548                                         $section = null;
 549
 550                                         continue;
 551                                 }
 552
 553                                 if ( $section == 'endfunctionhooks' ) {
 554                                         if ( !isset( $data['functionhooks'] ) ) {
 555                                                 wfDie( "'endfunctionhooks' without 'functionhooks' at line {$this->lineNum} of $this->file\n" );
 556                                         }
 557
 558                                         foreach ( explode( "\n", $data['functionhooks'] ) as $line ) {
 559                                                 $line = trim( $line );
 560
 561                                                 if ( $line ) {
 562                                                         if ( !$this->parserTest->requireFunctionHook( $line ) ) {
 563                                                                 return false;
 564                                                         }
 565                                                 }
 566                                         }
 567
 568                                         $data = array();
 569                                         $section = null;
 570
 571                                         continue;
 572                                 }
 573
 574                                 if ( $section == 'end' ) {
 575                                         if ( !isset( $data['test'] ) ) {
 576                                                 wfDie( "'end' without 'test' at line {$this->lineNum} of $this->file\n" );
 577                                         }
 578
 579                                         if ( !isset( $data['input'] ) ) {
 580                                                 wfDie( "'end' without 'input' at line {$this->lineNum} of $this->file\n" );
 581                                         }
 582
 583                                         if ( !isset( $data['result'] ) ) {
 584                                                 wfDie( "'end' without 'result' at line {$this->lineNum} of $this->file\n" );
 585                                         }
 586
 587                                         if ( !isset( $data['options'] ) ) {
 588                                                 $data['options'] = '';
 589                                         }
 590
 591                                         if ( !isset( $data['config'] ) )
 592                                                 $data['config'] = '';
 593
 594                                         if ( ( ( preg_match( '/\\bdisabled\\b/i', $data['options'] ) && !$this->parserTest->runDisabled )
 595                                                          || !preg_match( "/" . $this->parserTest->regex . "/i", $data['test'] ) )  ) {
 596                                                 # disabled test
 597                                                 $data = array();
 598                                                 $section = null;
 599
 600                                                 continue;
 601                                         }
 602
 603                                         $this->test = array(
 604                                                 'test' => ParserTest::chomp( $data['test'] ),
 605                                                 'input' => ParserTest::chomp( $data['input'] ),
 606                                                 'result' => ParserTest::chomp( $data['result'] ),
 607                                                 'options' => ParserTest::chomp( $data['options'] ),
 608                                                 'config' => ParserTest::chomp( $data['config'] ) );
 609
 610                                         return true;
 611                                 }
 612
 613                                 if ( isset ( $data[$section] ) ) {
 614                                         wfDie( "duplicate section '$section' at line {$this->lineNum} of $this->file\n" );
 615                                 }
 616
 617                                 $data[$section] = '';
 618
 619                                 continue;
 620                         }
 621
 622                         if ( $section ) {
 623                                 $data[$section] .= $line;
 624                         }
 625                 }
 626
 627                 return false;
 628         }
 629 }