testHelpers.inc

Go to the documentation of this file.

<?php
interface ITestRecorder {

    public function start();

    public function record( $test, $result );

    public function report();

    public function end();

}

class TestRecorder implements ITestRecorder {
    var $parent;
    var $term;

    function __construct( $parent ) {
        $this->parent = $parent;
        $this->term = $parent->term;
    }

    function start() {
        $this->total = 0;
        $this->success = 0;
    }

    function record( $test, $result ) {
        $this->total++;
        $this->success += ( $result ? 1 : 0 );
    }

    function end() {
        // dummy
    }

    function report() {
        if ( $this->total > 0 ) {
            $this->reportPercentage( $this->success, $this->total );
        } else {
            throw new MWException( "No tests found.\n" );
        }
    }

    function reportPercentage( $success, $total ) {
        $ratio = wfPercent( 100 * $success / $total );
        print $this->term->color( 1 ) . "Passed $success of $total tests ($ratio)... ";

        if ( $success == $total ) {
            print $this->term->color( 32 ) . "ALL TESTS PASSED!";
        } else {
            $failed = $total - $success;
            print $this->term->color( 31 ) . "$failed tests failed!";
        }

        print $this->term->reset() . "\n";

        return ( $success == $total );
    }
}

class DbTestPreviewer extends TestRecorder {
    protected $lb; // /< Database load balancer
    protected $db; // /< Database connection to the main DB
    protected $curRun; // /< run ID number for the current run
    protected $prevRun; // /< run ID number for the previous run, if any
    protected $results; // /< Result array

    function __construct( $parent ) {
        parent::__construct( $parent );

        $this->lb = wfGetLBFactory()->newMainLB();
        // This connection will have the wiki's table prefix, not parsertest_
        $this->db = $this->lb->getConnection( DB_MASTER );
    }

    function start() {
        parent::start();

        if ( !$this->db->tableExists( 'testrun', __METHOD__ )
            || !$this->db->tableExists( 'testitem', __METHOD__ )
        ) {
            print "WARNING> `testrun` table not found in database.\n";
            $this->prevRun = false;
        } else {
            // We'll make comparisons against the previous run later...
            $this->prevRun = $this->db->selectField( 'testrun', 'MAX(tr_id)' );
        }

        $this->results = array();
    }

    function record( $test, $result ) {
        parent::record( $test, $result );
        $this->results[$test] = $result;
    }

    function report() {
        if ( $this->prevRun ) {
            // f = fail, p = pass, n = nonexistent
            // codes show before then after
            $table = array(
                'fp' => 'previously failing test(s) now PASSING! :)',
                'pn' => 'previously PASSING test(s) removed o_O',
                'np' => 'new PASSING test(s) :)',

                'pf' => 'previously passing test(s) now FAILING! :(',
                'fn' => 'previously FAILING test(s) removed O_o',
                'nf' => 'new FAILING test(s) :(',
                'ff' => 'still FAILING test(s) :(',
            );

            $prevResults = array();

            $res = $this->db->select( 'testitem', array( 'ti_name', 'ti_success' ),
                array( 'ti_run' => $this->prevRun ), __METHOD__ );

            foreach ( $res as $row ) {
                if ( !$this->parent->regex
                    || preg_match( "/{$this->parent->regex}/i", $row->ti_name )
                ) {
                    $prevResults[$row->ti_name] = $row->ti_success;
                }
            }

            $combined = array_keys( $this->results + $prevResults );

            # Determine breakdown by change type
            $breakdown = array();
            foreach ( $combined as $test ) {
                if ( !isset( $prevResults[$test] ) ) {
                    $before = 'n';
                } elseif ( $prevResults[$test] == 1 ) {
                    $before = 'p';
                } else /* if ( $prevResults[$test] == 0 )*/ {
                    $before = 'f';
                }

                if ( !isset( $this->results[$test] ) ) {
                    $after = 'n';
                } elseif ( $this->results[$test] == 1 ) {
                    $after = 'p';
                } else /*if ( $this->results[$test] == 0 ) */ {
                    $after = 'f';
                }

                $code = $before . $after;

                if ( isset( $table[$code] ) ) {
                    $breakdown[$code][$test] = $this->getTestStatusInfo( $test, $after );
                }
            }

            # Write out results
            foreach ( $table as $code => $label ) {
                if ( !empty( $breakdown[$code] ) ) {
                    $count = count( $breakdown[$code] );
                    printf( "\n%4d %s\n", $count, $label );

                    foreach ( $breakdown[$code] as $differing_test_name => $statusInfo ) {
                        print "      * $differing_test_name  [$statusInfo]\n";
                    }
                }
            }
        } else {
            print "No previous test runs to compare against.\n";
        }

        print "\n";
        parent::report();
    }

    private function getTestStatusInfo( $testname, $after ) {
        // If we're looking at a test that has just been removed, then say when it first appeared.
        if ( $after == 'n' ) {
            $changedRun = $this->db->selectField( 'testitem',
                'MIN(ti_run)',
                array( 'ti_name' => $testname ),
                __METHOD__ );
            $appear = $this->db->selectRow( 'testrun',
                array( 'tr_date', 'tr_mw_version' ),
                array( 'tr_id' => $changedRun ),
                __METHOD__ );

            return "First recorded appearance: "
                . date( "d-M-Y H:i:s", strtotime( $appear->tr_date ) )
                . ", " . $appear->tr_mw_version;
        }

        // Otherwise, this test has previous recorded results.
        // See when this test last had a different result to what we're seeing now.
        $conds = array(
            'ti_name' => $testname,
            'ti_success' => ( $after == 'f' ? "1" : "0" ) );

        if ( $this->curRun ) {
            $conds[] = "ti_run != " . $this->db->addQuotes( $this->curRun );
        }

        $changedRun = $this->db->selectField( 'testitem', 'MAX(ti_run)', $conds, __METHOD__ );

        // If no record of ever having had a different result.
        if ( is_null( $changedRun ) ) {
            if ( $after == "f" ) {
                return "Has never passed";
            } else {
                return "Has never failed";
            }
        }

        // Otherwise, we're looking at a test whose status has changed.
        // (i.e. it used to work, but now doesn't; or used to fail, but is now fixed.)
        // In this situation, give as much info as we can as to when it changed status.
        $pre = $this->db->selectRow( 'testrun',
            array( 'tr_date', 'tr_mw_version' ),
            array( 'tr_id' => $changedRun ),
            __METHOD__ );
        $post = $this->db->selectRow( 'testrun',
            array( 'tr_date', 'tr_mw_version' ),
            array( "tr_id > " . $this->db->addQuotes( $changedRun ) ),
            __METHOD__,
            array( "LIMIT" => 1, "ORDER BY" => 'tr_id' )
        );

        if ( $post ) {
            $postDate = date( "d-M-Y H:i:s", strtotime( $post->tr_date ) ) . ", {$post->tr_mw_version}";
        } else {
            $postDate = 'now';
        }

        return ( $after == "f" ? "Introduced" : "Fixed" ) . " between "
            . date( "d-M-Y H:i:s", strtotime( $pre->tr_date ) ) . ", " . $pre->tr_mw_version
            . " and $postDate";
    }

    function end() {
        $this->lb->commitMasterChanges();
        $this->lb->closeAll();
        parent::end();
    }
}

class DbTestRecorder extends DbTestPreviewer {
    var $version;

    function start() {
        $this->db->begin( __METHOD__ );

        if ( !$this->db->tableExists( 'testrun' )
            || !$this->db->tableExists( 'testitem' )
        ) {
            print "WARNING> `testrun` table not found in database. Trying to create table.\n";
            $this->db->sourceFile( $this->db->patchPath( 'patch-testrun.sql' ) );
            echo "OK, resuming.\n";
        }

        parent::start();

        $this->db->insert( 'testrun',
            array(
                'tr_date' => $this->db->timestamp(),
                'tr_mw_version' => $this->version,
                'tr_php_version' => phpversion(),
                'tr_db_version' => $this->db->getServerVersion(),
                'tr_uname' => php_uname()
            ),
            __METHOD__ );
        if ( $this->db->getType() === 'postgres' ) {
            $this->curRun = $this->db->currentSequenceValue( 'testrun_id_seq' );
        } else {
            $this->curRun = $this->db->insertId();
        }
    }

    function record( $test, $result ) {
        parent::record( $test, $result );

        $this->db->insert( 'testitem',
            array(
                'ti_run' => $this->curRun,
                'ti_name' => $test,
                'ti_success' => $result ? 1 : 0,
            ),
            __METHOD__ );
    }
}

class TestFileIterator implements Iterator {
    private $file;
    private $fh;
    private $parserTest; /* An instance of ParserTest (parserTests.php) or MediaWikiParserTest (phpunit) */
    private $index = 0;
    private $test;
    private $section = null;
    private $sectionData = array();
    private $lineNum;
    private $eof;

    function __construct( $file, $parserTest ) {
        $this->file = $file;
        $this->fh = fopen( $this->file, "rt" );

        if ( !$this->fh ) {
            throw new MWException( "Couldn't open file '$file'\n" );
        }

        $this->parserTest = $parserTest;

        $this->lineNum = $this->index = 0;
    }

    function rewind() {
        if ( fseek( $this->fh, 0 ) ) {
            throw new MWException( "Couldn't fseek to the start of '$this->file'\n" );
        }

        $this->index = -1;
        $this->lineNum = 0;
        $this->eof = false;
        $this->next();

        return true;
    }

    function current() {
        return $this->test;
    }

    function key() {
        return $this->index;
    }

    function next() {
        if ( $this->readNextTest() ) {
            $this->index++;
            return true;
        } else {
            $this->eof = true;
        }
    }

    function valid() {
        return $this->eof != true;
    }

    function readNextTest() {
        $this->clearSection();

        # Create a fake parser tests which never run anything unless
        # asked to do so. This will avoid running hooks for a disabled test
        $delayedParserTest = new DelayedParserTest();

        while ( false !== ( $line = fgets( $this->fh ) ) ) {
            $this->lineNum++;
            $matches = array();

            if ( preg_match( '/^!!\s*(\S+)/', $line, $matches ) ) {
                $this->section = strtolower( $matches[1] );

                if ( $this->section == 'endarticle' ) {
                    $this->checkSection( 'text' );
                    $this->checkSection( 'article' );

                    $this->parserTest->addArticle( ParserTest::chomp( $this->sectionData['article'] ), $this->sectionData['text'], $this->lineNum );

                    $this->clearSection();

                    continue;
                }

                if ( $this->section == 'endhooks' ) {
                    $this->checkSection( 'hooks' );

                    foreach ( explode( "\n", $this->sectionData['hooks'] ) as $line ) {
                        $line = trim( $line );

                        if ( $line ) {
                            $delayedParserTest->requireHook( $line );
                        }
                    }

                    $this->clearSection();

                    continue;
                }

                if ( $this->section == 'endfunctionhooks' ) {
                    $this->checkSection( 'functionhooks' );

                    foreach ( explode( "\n", $this->sectionData['functionhooks'] ) as $line ) {
                        $line = trim( $line );

                        if ( $line ) {
                            $delayedParserTest->requireFunctionHook( $line );
                        }
                    }

                    $this->clearSection();

                    continue;
                }

                if ( $this->section == 'end' ) {
                    $this->checkSection( 'test' );
                    // "input" and "result" are old section names allowed
                    // for backwards-compatibility.
                    $input = $this->checkSection( array( 'wikitext', 'input' ), false );
                    $result = $this->checkSection( array( 'html/php', 'html/*', 'html', 'result' ), false );

                    if ( !isset( $this->sectionData['options'] ) ) {
                        $this->sectionData['options'] = '';
                    }

                    if ( !isset( $this->sectionData['config'] ) ) {
                        $this->sectionData['config'] = '';
                    }

                    if ( $input == false || $result == false ||
                        ( ( preg_match( '/\\bdisabled\\b/i', $this->sectionData['options'] ) && !$this->parserTest->runDisabled )
                        || ( preg_match( '/\\bparsoid\\b/i', $this->sectionData['options'] ) && $result != 'html/php' && !$this->parserTest->runParsoid )
                        || !preg_match( "/" . $this->parserTest->regex . "/i", $this->sectionData['test'] ) )
                    ) {
                        # disabled test
                        $this->clearSection();

                        # Forget any pending hooks call since test is disabled
                        $delayedParserTest->reset();

                        continue;
                    }

                    # We are really going to run the test, run pending hooks and hooks function
                    wfDebug( __METHOD__ . " unleashing delayed test for: {$this->sectionData['test']}" );
                    $hooksResult = $delayedParserTest->unleash( $this->parserTest );
                    if ( !$hooksResult ) {
                        # Some hook reported an issue. Abort.
                        return false;
                    }

                    $this->test = array(
                        'test' => ParserTest::chomp( $this->sectionData['test'] ),
                        'input' => ParserTest::chomp( $this->sectionData[ $input ] ),
                        'result' => ParserTest::chomp( $this->sectionData[ $result ] ),
                        'options' => ParserTest::chomp( $this->sectionData['options'] ),
                        'config' => ParserTest::chomp( $this->sectionData['config'] ),
                    );

                    return true;
                }

                if ( isset( $this->sectionData[$this->section] ) ) {
                    throw new MWException( "duplicate section '$this->section' at line {$this->lineNum} of $this->file\n" );
                }

                $this->sectionData[$this->section] = '';

                continue;
            }

            if ( $this->section ) {
                $this->sectionData[$this->section] .= $line;
            }
        }

        return false;
    }

    private function clearSection() {
        $this->sectionData = array();
        $this->section = null;

    }

    private function checkSection( $tokens, $fatal = true ) {
        if ( is_null( $this->section ) ) {
            throw new MWException( __METHOD__ . " can not verify a null section!\n" );
        }
        if ( !is_array( $tokens ) ) {
            $tokens = array( $tokens );
        }
        if ( count( $tokens ) == 0 ) {
            throw new MWException( __METHOD__ . " can not verify zero sections!\n" );
        }

        $data = $this->sectionData;
        $tokens = array_filter( $tokens, function ( $token ) use ( $data ) {
            return isset( $data[ $token ] );
        } );

        if ( count( $tokens ) == 0 ) {
            if ( !$fatal ) {
                return false;
            }
            throw new MWException( sprintf(
                "'%s' without '%s' at line %s of %s\n",
                $this->section,
                implode( ',', $tokens ),
                $this->lineNum,
                $this->file
            ) );
        }
        if ( count( $tokens ) > 1 ) {
            throw new MWException( sprintf(
                "'%s' with unexpected tokens '%s' at line %s of %s\n",
                $this->section,
                implode( ',', $tokens ),
                $this->lineNum,
                $this->file
            ) );
        }

        $tokens = array_values( $tokens );
        return $tokens[ 0 ];
    }
}

class DelayedParserTest {

    private $hooks;
    private $fnHooks;

    public function __construct() {
        $this->reset();
    }

    public function reset() {
        $this->hooks = array();
        $this->fnHooks = array();
    }

    public function unleash( &$parserTest ) {
        if ( !( $parserTest instanceof ParserTest || $parserTest instanceof NewParserTest ) ) {
            throw new MWException( __METHOD__ . " must be passed an instance of ParserTest or NewParserTest classes\n" );
        }

        # Trigger delayed hooks. Any failure will make us abort
        foreach ( $this->hooks as $hook ) {
            $ret = $parserTest->requireHook( $hook );
            if ( !$ret ) {
                return false;
            }
        }

        # Trigger delayed function hooks. Any failure will make us abort
        foreach ( $this->fnHooks as $fnHook ) {
            $ret = $parserTest->requireFunctionHook( $fnHook );
            if ( !$ret ) {
                return false;
            }
        }

        # Delayed execution was successful.
        return true;
    }

    public function requireHook( $hook ) {
        $this->hooks[] = $hook;
    }

    public function requireFunctionHook( $fnHook ) {
        $this->fnHooks[] = $fnHook;
    }

}