MockSearchEngine.php
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

MockSearchEngine.php
View Options

	<?php

	class MockSearchEngine extends SearchEngine {
	use LinkCacheTestTrait;

	/** @var SearchResult[][] */
	private static $results = [];
	/** @var ISearchResultSet[][] */
	private static $interwikiResults = [];

	public static function clearMockResults() {
	self::$results = [];
	self::$interwikiResults = [];
	}

	/**
	* @param string $query The query searched for after initial
	* transformations have been applied.
	* @param SearchResult[] $results The results to return for $query
	*/
	public static function addMockResults( $query, array $results ) {
	$mockSearchEngine = new self();
	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?
	$mockSearchEngine->addGoodLinkObject( mt_rand(), $result->getTitle() );
	}
	self::$results[$query] = $results;
	}

	/**
	* @param ISearchResultSet[][] $interwikiResults
	*/
	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 );
	}
	}