tests/phpunit/includes/search/ParserOutputSearchDataExtractorTest.php

   1 <?php
   2
   3 use MediaWiki\Parser\ParserOutput;
   4 use MediaWiki\Search\ParserOutputSearchDataExtractor;
   5 use MediaWiki\Title\Title;
   6
   7 /**
   8  * @group Search
   9  * @covers \MediaWiki\Search\ParserOutputSearchDataExtractor
  10  * @group Database
  11  */
  12 class ParserOutputSearchDataExtractorTest extends MediaWikiLangTestCase {
  13
  14         public function testGetCategories() {
  15                 $categories = [
  16                         'Foo_bar' => 'Bar',
  17                         'New_page' => ''
  18                 ];
  19
  20                 $parserOutput = new ParserOutput( '', [], $categories );
  21
  22                 $searchDataExtractor = new ParserOutputSearchDataExtractor();
  23
  24                 $this->assertEquals(
  25                         [ 'Foo bar', 'New page' ],
  26                         $searchDataExtractor->getCategories( $parserOutput )
  27                 );
  28         }
  29
  30         public function testGetExternalLinks() {
  31                 $parserOutput = new ParserOutput();
  32
  33                 $parserOutput->addExternalLink( 'https://foo' );
  34                 $parserOutput->addExternalLink( 'https://bar' );
  35
  36                 $searchDataExtractor = new ParserOutputSearchDataExtractor();
  37
  38                 $this->assertEquals(
  39                         [ 'https://foo', 'https://bar' ],
  40                         $searchDataExtractor->getExternalLinks( $parserOutput )
  41                 );
  42         }
  43
  44         public function testGetOutgoingLinks() {
  45                 $parserOutput = new ParserOutput();
  46
  47                 $parserOutput->addLink( Title::makeTitle( NS_MAIN, 'Foo_bar' ), 1 );
  48                 $parserOutput->addLink( Title::makeTitle( NS_HELP, 'Contents' ), 2 );
  49
  50                 $searchDataExtractor = new ParserOutputSearchDataExtractor();
  51
  52                 // this indexes links with db key
  53                 $this->assertEquals(
  54                         [ 'Foo_bar', 'Help:Contents' ],
  55                         $searchDataExtractor->getOutgoingLinks( $parserOutput )
  56                 );
  57         }
  58
  59         public function testGetTemplates() {
  60                 $title = Title::makeTitle( NS_TEMPLATE, 'Cite_news' );
  61
  62                 $parserOutput = new ParserOutput();
  63                 $parserOutput->addTemplate( $title, 10, 100 );
  64
  65                 $searchDataExtractor = new ParserOutputSearchDataExtractor();
  66
  67                 $this->assertEquals(
  68                         [ 'Template:Cite news' ],
  69                         $searchDataExtractor->getTemplates( $parserOutput )
  70                 );
  71         }
  72
  73 }