DatabaseTestHelper.php

Go to the documentation of this file.

<?php
 
use Wikimedia\Rdbms\TransactionProfiler;
use Wikimedia\Rdbms\DatabaseDomain;
 
class DatabaseTestHelper extends Database {
 
    protected $testName = [];
 
    protected $lastSqls = [];
 
    protected $nextResult = [];
 
    protected $tablesExists;
 
    protected $unionSupportsOrderAndLimit = true;
 
    public function __construct( $testName, array $opts = [] ) {
        $this->testName = $testName;
 
        $this->profiler = new ProfilerStub( [] );
        $this->trxProfiler = new TransactionProfiler();
        $this->cliMode = isset( $opts['cliMode'] ) ? $opts['cliMode'] : true;
        $this->connLogger = new \Psr\Log\NullLogger();
        $this->queryLogger = new \Psr\Log\NullLogger();
        $this->errorLogger = function ( Exception $e ) {
            wfWarn( get_class( $e ) . ": {$e->getMessage()}" );
        };
        $this->currentDomain = DatabaseDomain::newUnspecified();
    }
 
    public function getLastSqls() {
        $lastSqls = implode( '; ', $this->lastSqls );
        $this->lastSqls = [];
 
        return $lastSqls;
    }
 
    public function setExistingTables( $tablesExists ) {
        $this->tablesExists = (array)$tablesExists;
    }
 
    public function forceNextResult( $res ) {
        $this->nextResult = $res;
    }
 
    protected function addSql( $sql ) {
        // clean up spaces before and after some words and the whole string
        $this->lastSqls[] = trim( preg_replace(
            '/\s{2,}(?=FROM|WHERE|GROUP BY|ORDER BY|LIMIT)|(?<=SELECT|INSERT|UPDATE)\s{2,}/',
            ' ', $sql
        ) );
    }
 
    protected function checkFunctionName( $fname ) {
        if ( substr( $fname, 0, strlen( $this->testName ) ) !== $this->testName ) {
            throw new MWException( 'function name does not start with test class. ' .
                $fname . ' vs. ' . $this->testName . '. ' .
                'Please provide __METHOD__ to database methods.' );
        }
    }
 
    function strencode( $s ) {
        // Choose apos to avoid handling of escaping double quotes in quoted text
        return str_replace( "'", "\'", $s );
    }
 
    public function addIdentifierQuotes( $s ) {
        // no escaping to avoid handling of double quotes in quoted text
        return $s;
    }
 
    public function query( $sql, $fname = '', $tempIgnore = false ) {
        $this->checkFunctionName( $fname );
        $this->addSql( $sql );
 
        return parent::query( $sql, $fname, $tempIgnore );
    }
 
    public function tableExists( $table, $fname = __METHOD__ ) {
        $tableRaw = $this->tableName( $table, 'raw' );
        if ( isset( $this->mSessionTempTables[$tableRaw] ) ) {
            return true; // already known to exist
        }
 
        $this->checkFunctionName( $fname );
 
        return in_array( $table, (array)$this->tablesExists );
    }
 
    // Redeclare parent method to make it public
    public function nativeReplace( $table, $rows, $fname ) {
        return parent::nativeReplace( $table, $rows, $fname );
    }
 
    function getType() {
        return 'test';
    }
 
    function open( $server, $user, $password, $dbName ) {
        return false;
    }
 
    function fetchObject( $res ) {
        return false;
    }
 
    function fetchRow( $res ) {
        return false;
    }
 
    function numRows( $res ) {
        return -1;
    }
 
    function numFields( $res ) {
        return -1;
    }
 
    function fieldName( $res, $n ) {
        return 'test';
    }
 
    function insertId() {
        return -1;
    }
 
    function dataSeek( $res, $row ) {
        /* nop */
    }
 
    function lastErrno() {
        return -1;
    }
 
    function lastError() {
        return 'test';
    }
 
    function fieldInfo( $table, $field ) {
        return false;
    }
 
    function indexInfo( $table, $index, $fname = 'Database::indexInfo' ) {
        return false;
    }
 
    function affectedRows() {
        return -1;
    }
 
    function getSoftwareLink() {
        return 'test';
    }
 
    function getServerVersion() {
        return 'test';
    }
 
    function getServerInfo() {
        return 'test';
    }
 
    function isOpen() {
        return true;
    }
 
    function ping( &$rtt = null ) {
        $rtt = 0.0;
        return true;
    }
 
    protected function closeConnection() {
        return false;
    }
 
    protected function doQuery( $sql ) {
        $res = $this->nextResult;
        $this->nextResult = [];
 
        return new FakeResultWrapper( $res );
    }
 
    public function unionSupportsOrderAndLimit() {
        return $this->unionSupportsOrderAndLimit;
    }
 
    public function setUnionSupportsOrderAndLimit( $v ) {
        $this->unionSupportsOrderAndLimit = (bool)$v;
    }
}