public function testShrink()
{
$set = SearchSuggestionSet::emptySuggestionSet();
for ($i = 0; $i < 100; $i++) {
$set->append(new SearchSuggestion(0));
}
$set->shrink(10);
$this->assertEquals(10, $set->getSize());
$set->shrink(0);
$this->assertEquals(0, $set->getSize());
}
/**
* @dataProvider provideSearchBackend
* @covers PrefixSearch::searchBackend
*/
public function testSearchBackend(array $case)
{
$search = $stub = $this->getMockBuilder('SearchEngine')->setMethods(['completionSearchBackend'])->getMock();
$return = SearchSuggestionSet::fromStrings($case['provision']);
$search->expects($this->any())->method('completionSearchBackend')->will($this->returnValue($return));
$search->setLimitOffset(3);
$results = $search->completionSearch($case['query']);
$results = $results->map(function (SearchSuggestion $s) {
return $s->getText();
});
$this->assertEquals($case['results'], $results, $case[0]);
}
/**
* Process completion search results.
* Resolves the titles and rescores.
* @param SearchSuggestionSet $suggestions
* @return SearchSuggestionSet
*/
protected function processCompletionResults($search, SearchSuggestionSet $suggestions)
{
$search = trim($search);
// preload the titles with LinkBatch
$titles = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle();
});
$lb = new LinkBatch($titles);
$lb->setCaller(__METHOD__);
$lb->execute();
$results = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle()->getPrefixedText();
});
if ($this->offset === 0) {
// Rescore results with an exact title match
// NOTE: in some cases like cross-namespace redirects
// (frequently used as shortcuts e.g. WP:WP on huwiki) some
// backends like Cirrus will return no results. We should still
// try an exact title match to workaround this limitation
$rescorer = new SearchExactMatchRescorer();
$rescoredResults = $rescorer->rescore($search, $this->namespaces, $results, $this->limit);
} else {
// No need to rescore if offset is not 0
// The exact match must have been returned at position 0
// if it existed.
$rescoredResults = $results;
}
if (count($rescoredResults) > 0) {
$found = array_search($rescoredResults[0], $results);
if ($found === false) {
// If the first result is not in the previous array it
// means that we found a new exact match
$exactMatch = SearchSuggestion::fromTitle(0, Title::newFromText($rescoredResults[0]));
$suggestions->prepend($exactMatch);
$suggestions->shrink($this->limit);
} else {
// if the first result is not the same we need to rescore
if ($found > 0) {
$suggestions->rescore($found);
}
}
}
return $suggestions;
}
/**
* Process completion search results.
* Resolves the titles and rescores.
* @param SearchSuggestionSet $suggestions
* @return SearchSuggestionSet
*/
protected function processCompletionResults($search, SearchSuggestionSet $suggestions)
{
if ($suggestions->getSize() == 0) {
// If we don't have anything, don't bother
return $suggestions;
}
$search = trim($search);
// preload the titles with LinkBatch
$titles = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle();
});
$lb = new LinkBatch($titles);
$lb->setCaller(__METHOD__);
$lb->execute();
$results = $suggestions->map(function (SearchSuggestion $sugg) {
return $sugg->getSuggestedTitle()->getPrefixedText();
});
// Rescore results with an exact title match
$rescorer = new SearchExactMatchRescorer();
$rescoredResults = $rescorer->rescore($search, $this->namespaces, $results, $this->limit);
if (count($rescoredResults) > 0) {
$found = array_search($rescoredResults[0], $results);
if ($found === false) {
// If the first result is not in the previous array it
// means that we found a new exact match
$exactMatch = SearchSuggestion::fromTitle(0, Title::newFromText($rescoredResults[0]));
$suggestions->prepend($exactMatch);
$suggestions->shrink($this->limit);
} else {
// if the first result is not the same we need to rescore
if ($found > 0) {
$suggestions->rescore($found);
}
}
}
return $suggestions;
}
/**
* Add suggestion set to the end of the current one.
* @param SearchSuggestionSet $set
*/
public function appendAll(SearchSuggestionSet $set)
{
foreach ($set->getSuggestions() as $sugg) {
$this->append($sugg);
}
}
