DatabaseTestHelper.php

Go to the documentation of this file.

<?php
 
use Wikimedia\Rdbms\TransactionProfiler;
use Wikimedia\Rdbms\DatabaseDomain;
use Wikimedia\Rdbms\Database;
 
class DatabaseTestHelper extends Database {
 
    protected $testName = [];
 
    protected $lastSqls = [];
 
    protected $nextResult = [];
 
    protected $nextError = null;
    protected $lastError = null;
 
    protected $tablesExists;
 
    protected $unionSupportsOrderAndLimit = true;
 
    public function __construct( $testName, array $opts = [] ) {
        $this->testName = $testName;
 
        $this->profiler = null;
        $this->trxProfiler = new TransactionProfiler();
        $this->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->deprecationLogger = function ( $msg ) {
            wfWarn( $msg );
        };
        $this->currentDomain = DatabaseDomain::newUnspecified();
        $this->open( 'localhost', 'testuser', 'password', 'testdb', null, '' );
    }
 
    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;
    }
 
    public function forceNextQueryError( $errno, $error, $options = [] ) {
        $this->nextError = [ 'errno' => $errno, 'error' => $error ] + $options;
    }
 
    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 ( $fname === 'Wikimedia\\Rdbms\\Database::close' ) {
            return; // no $fname parameter
        }
 
        // Handle some internal calls from the Database class
        $check = $fname;
        if ( preg_match(
            '/^Wikimedia\\\\Rdbms\\\\Database::(?:query|beginIfImplied) \((.+)\)$/',
            $fname,
            $m
        ) ) {
            $check = $m[1];
        }
 
        if ( substr( $check, 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 = '', $flags = 0 ) {
        $this->checkFunctionName( $fname );
 
        return parent::query( $sql, $fname, $flags );
    }
 
    public function tableExists( $table, $fname = __METHOD__ ) {
        $tableRaw = $this->tableName( $table, 'raw' );
        if ( isset( $this->sessionTempTables[$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 ) {
        parent::nativeReplace( $table, $rows, $fname );
    }
 
    function getType() {
        return 'test';
    }
 
    function open( $server, $user, $password, $dbName, $schema, $tablePrefix ) {
        $this->conn = (object)[ 'test' ];
 
        return true;
    }
 
    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 $this->lastError ? $this->lastError['errno'] : -1;
    }
 
    function lastError() {
        return $this->lastError ? $this->lastError['error'] : 'test';
    }
 
    protected function wasKnownStatementRollbackError() {
        return $this->lastError['wasKnownStatementRollbackError'] ?? false;
    }
 
    function fieldInfo( $table, $field ) {
        return false;
    }
 
    function indexInfo( $table, $index, $fname = 'Database::indexInfo' ) {
        return false;
    }
 
    function fetchAffectedRowCount() {
        return -1;
    }
 
    function getSoftwareLink() {
        return 'test';
    }
 
    function getServerVersion() {
        return 'test';
    }
 
    function getServerInfo() {
        return 'test';
    }
 
    function isOpen() {
        return $this->conn ? true : false;
    }
 
    function ping( &$rtt = null ) {
        $rtt = 0.0;
        return true;
    }
 
    protected function closeConnection() {
        return true;
    }
 
    protected function doQuery( $sql ) {
        $sql = preg_replace( '< /\* .+?  \*/>', '', $sql );
        $this->addSql( $sql );
 
        if ( $this->nextError ) {
            $this->lastError = $this->nextError;
            $this->nextError = null;
            return false;
        }
 
        $res = $this->nextResult;
        $this->nextResult = [];
        $this->lastError = null;
 
        return new FakeResultWrapper( $res );
    }
 
    public function unionSupportsOrderAndLimit() {
        return $this->unionSupportsOrderAndLimit;
    }
 
    public function setUnionSupportsOrderAndLimit( $v ) {
        $this->unionSupportsOrderAndLimit = (bool)$v;
    }
}