tests/phpunit/includes/specials/SpecialSearchTestMockResultSet.php

   1 <?php
   2
   3 use MediaWiki\Title\Title;
   4
   5 class SpecialSearchTestMockResultSet extends SearchResultSet {
   6
   7         private ?string $suggestion;
   8         private ?string $rewrittenQuery;
   9
  10         public function __construct(
  11                 ?string $suggestion = null,
  12                 ?string $rewrittenQuery = null,
  13                 array $results = [],
  14                 bool $containedSyntax = false
  15         ) {
  16                 $this->suggestion = $suggestion;
  17                 $this->rewrittenQuery = $rewrittenQuery;
  18                 $this->results = $results;
  19                 $this->containedSyntax = $containedSyntax;
  20         }
  21
  22         public function getTotalHits() {
  23                 return $this->numRows();
  24         }
  25
  26         public function hasSuggestion() {
  27                 return $this->suggestion !== null;
  28         }
  29
  30         public function getSuggestionQuery() {
  31                 return $this->suggestion;
  32         }
  33
  34         public function getSuggestionSnippet() {
  35                 return $this->suggestion;
  36         }
  37
  38         public function hasRewrittenQuery() {
  39                 return $this->rewrittenQuery !== null;
  40         }
  41
  42         public function getQueryAfterRewrite() {
  43                 return $this->rewrittenQuery;
  44         }
  45
  46         public function getQueryAfterRewriteSnippet() {
  47                 return htmlspecialchars( $this->rewrittenQuery );
  48         }
  49
  50         public function getFirstResult(): ?Title {
  51                 $first = reset( $this->results );
  52                 return $first ? $first->getTitle() : null;
  53         }
  54 }