7 * @covers SearchEngine<extended>
8 * @note Coverage will only ever show one of on of the Search* classes
10 class SearchEngineTest
extends MediaWikiLangTestCase
{
18 * Checks for database type & version.
19 * Will skip current test if DB does not support search.
21 protected function setUp() {
24 // Search tests require MySQL or SQLite with FTS
25 $dbType = $this->db
->getType();
26 $dbSupported = ( $dbType === 'mysql' )
27 ||
( $dbType === 'sqlite' && $this->db
->getFulltextSearchModule() == 'FTS3' );
29 if ( !$dbSupported ) {
30 $this->markTestSkipped( "MySQL or SQLite with FTS3 only" );
33 $searchType = $this->db
->getSearchEngine();
34 $this->setMwGlobals( [
35 'wgSearchType' => $searchType
38 $this->search
= new $searchType( $this->db
);
41 protected function tearDown() {
42 unset( $this->search
);
47 public function addDBDataOnce() {
48 if ( !$this->isWikitextNS( NS_MAIN
) ) {
49 // @todo cover the case of non-wikitext content in the main namespace
53 // Reset the search type back to default - some extensions may have
55 $this->setMwGlobals( [ 'wgSearchType' => null ] );
57 $this->insertPage( 'Not_Main_Page', 'This is not a main page' );
60 'This is not a talk page to the main page, see [[smithee]]'
62 $this->insertPage( 'Smithee', 'A smithee is one who smiths. See also [[Alan Smithee]]' );
63 $this->insertPage( 'Talk:Smithee', 'This article sucks.' );
64 $this->insertPage( 'Unrelated_page', 'Nothing in this page is about the S word.' );
65 $this->insertPage( 'Another_page', 'This page also is unrelated.' );
66 $this->insertPage( 'Help:Help', 'Help me!' );
67 $this->insertPage( 'Thppt', 'Blah blah' );
68 $this->insertPage( 'Alan_Smithee', 'yum' );
69 $this->insertPage( 'Pages', 'are\'food' );
70 $this->insertPage( 'HalfOneUp', 'AZ' );
71 $this->insertPage( 'FullOneUp', 'AZ' );
72 $this->insertPage( 'HalfTwoLow', 'az' );
73 $this->insertPage( 'FullTwoLow', 'az' );
74 $this->insertPage( 'HalfNumbers', '1234567890' );
75 $this->insertPage( 'FullNumbers', '1234567890' );
76 $this->insertPage( 'DomainName', 'example.com' );
79 protected function fetchIds( $results ) {
80 if ( !$this->isWikitextNS( NS_MAIN
) ) {
81 $this->markTestIncomplete( __CLASS__
. " does no yet support non-wikitext content "
82 . "in the main namespace" );
84 $this->assertTrue( is_object( $results ) );
87 $row = $results->next();
89 $matches[] = $row->getTitle()->getPrefixedText();
90 $row = $results->next();
93 # Search is not guaranteed to return results in a certain order;
94 # sort them numerically so we will compare simply that we received
95 # the expected matches.
101 public function testFullWidth() {
103 [ 'FullOneUp', 'FullTwoLow', 'HalfOneUp', 'HalfTwoLow' ],
104 $this->fetchIds( $this->search
->searchText( 'AZ' ) ),
105 "Search for normalized from Half-width Upper" );
107 [ 'FullOneUp', 'FullTwoLow', 'HalfOneUp', 'HalfTwoLow' ],
108 $this->fetchIds( $this->search
->searchText( 'az' ) ),
109 "Search for normalized from Half-width Lower" );
111 [ 'FullOneUp', 'FullTwoLow', 'HalfOneUp', 'HalfTwoLow' ],
112 $this->fetchIds( $this->search
->searchText( 'AZ' ) ),
113 "Search for normalized from Full-width Upper" );
115 [ 'FullOneUp', 'FullTwoLow', 'HalfOneUp', 'HalfTwoLow' ],
116 $this->fetchIds( $this->search
->searchText( 'az' ) ),
117 "Search for normalized from Full-width Lower" );
120 public function testTextSearch() {
123 $this->fetchIds( $this->search
->searchText( 'smithee' ) ),
124 "Plain search failed" );
127 public function testTextPowerSearch() {
128 $this->search
->setNamespaces( [ 0, 1, 4 ] );
132 'Talk:Not Main Page',
134 $this->fetchIds( $this->search
->searchText( 'smithee' ) ),
135 "Power search failed" );
138 public function testTitleSearch() {
144 $this->fetchIds( $this->search
->searchTitle( 'smithee' ) ),
145 "Title search failed" );
148 public function testTextTitlePowerSearch() {
149 $this->search
->setNamespaces( [ 0, 1, 4 ] );
156 $this->fetchIds( $this->search
->searchTitle( 'smithee' ) ),
157 "Title power search failed" );
161 * @covers SearchEngine::getSearchIndexFields
163 public function testSearchIndexFields() {
165 * @var $mockEngine SearchEngine
167 $mockEngine = $this->getMock( 'SearchEngine', [ 'makeSearchFieldMapping' ] );
169 $mockFieldBuilder = function ( $name, $type ) {
171 $this->getMockBuilder( 'SearchIndexFieldDefinition' )->setConstructorArgs( [
176 $mockField->expects( $this->any() )->method( 'getMapping' )->willReturn( [
177 'testData' => 'test',
182 $mockField->expects( $this->any() )
184 ->willReturn( $mockField );
189 $mockEngine->expects( $this->atLeastOnce() )
190 ->method( 'makeSearchFieldMapping' )
191 ->willReturnCallback( $mockFieldBuilder );
193 // Not using mock since PHPUnit mocks do not work properly with references in params
194 $this->setTemporaryHook( 'SearchIndexFields',
195 function ( &$fields, SearchEngine
$engine ) use ( $mockFieldBuilder ) {
196 $fields['testField'] =
197 $mockFieldBuilder( "testField", SearchIndexField
::INDEX_TYPE_TEXT
);
201 $fields = $mockEngine->getSearchIndexFields();
202 $this->assertArrayHasKey( 'language', $fields );
203 $this->assertArrayHasKey( 'category', $fields );
204 $this->assertInstanceOf( 'SearchIndexField', $fields['testField'] );
206 $mapping = $fields['testField']->getMapping( $mockEngine );
207 $this->assertArrayHasKey( 'testData', $mapping );
208 $this->assertEquals( 'test', $mapping['testData'] );