MockSearchResultSet.php

Go to the documentation of this file.

<?php
 
class MockSearchResultSet extends SearchResultSet {
    /*
     * @var SearchResultSet[][] Map from result type to list of results for
     *  that type.
     */
    private $interwikiResults;
 
    public function __construct( array $results, array $interwikiResults = [] ) {
        parent::__construct( false, false );
        $this->results = $results;
        $this->interwikiResults = $interwikiResults;
    }
 
    public function numRows() {
        return count( $this->results );
    }
 
    public function hasInterwikiResults( $type = self::SECONDARY_RESULTS ) {
        return isset( $this->interwikiResults[$type] ) &&
            count( $this->interwikiResults[$type] ) > 0;
    }
 
    public function extractResults() {
        $results = parent::extractResults();
 
        foreach ( $results as &$result ) {
            // Resolve deferred results; needed to work around T203279
            if ( is_callable( $result ) ) {
                $result = $result();
            }
        }
 
        return $results;
    }
 
    public function getInterwikiResults( $type = self::SECONDARY_RESULTS ) {
        if ( $this->hasInterwikiResults( $type ) ) {
            return $this->interwikiResults[$type];
        } else {
            return null;
        }
    }
}