MockSearchEngine.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
 
class MockSearchEngine extends SearchEngine {
    private static $results = [];
    private static $interwikiResults = [];
 
    public static function clearMockResults() {
        self::$results = [];
        self::$interwikiResults = [];
    }
 
    public static function addMockResults( $query, array $results ) {
        $lc = MediaWikiServices::getInstance()->getLinkCache();
        foreach ( $results as &$result ) {
            // Resolve deferred results; needed to work around T203279
            if ( is_callable( $result ) ) {
                $result = $result();
            }
 
            // TODO: better page ids? Does it matter?
            $lc->addGoodLinkObj( mt_rand(), $result->getTitle() );
        }
        self::$results[$query] = $results;
    }
 
    public static function setMockInterwikiResults( array $interwikiResults ) {
        self::$interwikiResults = $interwikiResults;
    }
 
    protected function doSearchText( $term ) {
        if ( isset( self::$results[ $term ] ) ) {
            $results = array_slice( self::$results[ $term ], $this->offset, $this->limit );
        } else {
            $results = [];
        }
        return new MockSearchResultSet( $results, self::$interwikiResults );
    }
}