tests/phpunit/includes/page/PageSelectQueryBuilderTest.php

   1 <?php
   2 namespace MediaWiki\Tests\Page;
   3
   4 use Exception;
   5 use MediaWiki\Config\ServiceOptions;
   6 use MediaWiki\MainConfigNames;
   7 use MediaWiki\Page\PageSelectQueryBuilder;
   8 use MediaWiki\Page\PageStore;
   9 use MediaWikiIntegrationTestCase;
  10
  11 /**
  12  * @group Database
  13  */
  14 class PageSelectQueryBuilderTest extends MediaWikiIntegrationTestCase {
  15
  16         public function addDBDataOnce() {
  17                 $this->getExistingTestPage( 'AA' );
  18                 $this->getExistingTestPage( 'AB' );
  19                 $this->getExistingTestPage( 'BB' );
  20
  21                 $this->getExistingTestPage( 'Talk:AA' );
  22                 $this->getExistingTestPage( 'User:AB' );
  23         }
  24
  25         /**
  26          * @return PageStore
  27          * @throws Exception
  28          */
  29         private function getPageStore() {
  30                 $services = $this->getServiceContainer();
  31
  32                 $serviceOptions = new ServiceOptions(
  33                         PageStore::CONSTRUCTOR_OPTIONS,
  34                         [
  35                                 MainConfigNames::LanguageCode => $services->getContentLanguageCode()->toString(),
  36                                 MainConfigNames::PageLanguageUseDB => true,
  37                         ]
  38                 );
  39
  40                 return new PageStore(
  41                         $serviceOptions,
  42                         $services->getDBLoadBalancer(),
  43                         $services->getNamespaceInfo(),
  44                         $services->getTitleParser(),
  45                         $services->getLinkCache(),
  46                         $services->getStatsFactory()
  47                 );
  48         }
  49
  50         /**
  51          * @covers \MediaWiki\Page\PageSelectQueryBuilder::wherePageIds
  52          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecordArray
  53          */
  54         public function testFetchBatchOfPagesById() {
  55                 $pageStore = $this->getPageStore();
  56
  57                 $recAA = $pageStore->getPageByName( NS_MAIN, 'AA' );
  58                 $recAB = $pageStore->getPageByName( NS_MAIN, 'AB' );
  59
  60                 $recs = $pageStore->newSelectQueryBuilder()
  61                         ->wherePageIds( [] )
  62                         ->fetchPageRecordArray();
  63
  64                 $this->assertCount( 0, $recs );
  65
  66                 $recs = $pageStore->newSelectQueryBuilder()
  67                         ->wherePageIds( [ $recAA->getId(), $recAB->getId() ] )
  68                         ->fetchPageRecordArray();
  69
  70                 $this->assertCount( 2, $recs );
  71                 $this->assertSame( 'AA', $recs[ $recAA->getId() ]->getDBkey() );
  72                 $this->assertSame( 'AB', $recs[ $recAB->getId() ]->getDBkey() );
  73         }
  74
  75         /**
  76          * @covers \MediaWiki\Page\PageSelectQueryBuilder::wherePageIds
  77          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecord
  78          */
  79         public function testFetchSinglePageById() {
  80                 $pageStore = $this->getPageStore();
  81
  82                 $recAB = $pageStore->getPageByName( NS_MAIN, 'AB' );
  83
  84                 $rec = $pageStore->newSelectQueryBuilder()
  85                         ->wherePageIds( $recAB->getId() )
  86                         ->fetchPageRecord();
  87
  88                 $this->assertTrue( $recAB->isSamePageAs( $rec ) );
  89
  90                 $rec = $pageStore->newSelectQueryBuilder()
  91                         ->wherePageIds( 348529043 )
  92                         ->fetchPageRecord();
  93
  94                 $this->assertNull( $rec );
  95         }
  96
  97         /**
  98          * @covers \MediaWiki\Page\PageSelectQueryBuilder::whereTitles
  99          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecordArray
 100          * @covers \MediaWiki\Page\PageSelectQueryBuilder::orderByPageId
 101          */
 102         public function testFindBatchOfPageIdsByTitle() {
 103                 $pageStore = $this->getPageStore();
 104
 105                 $recAA = $pageStore->getPageByName( NS_MAIN, 'AA' );
 106                 $recAB = $pageStore->getPageByName( NS_MAIN, 'AB' );
 107                 $recAC = $pageStore->getPageByName( NS_MAIN, 'BB' );
 108
 109                 $recs = $pageStore->newSelectQueryBuilder()
 110                         ->whereTitles( NS_FILE, [ 'AA', 'AB', 'BB' ] )
 111                         ->fetchPageIds();
 112
 113                 $this->assertCount( 0, $recs );
 114
 115                 $recs = $pageStore->newSelectQueryBuilder()
 116                         ->whereTitles( NS_MAIN, [ 'AA', 'AB', 'BB' ] )
 117                         ->orderByPageId( PageSelectQueryBuilder::SORT_DESC )
 118                         ->fetchPageIds();
 119
 120                 $expectedIds = [ $recAA->getId(), $recAB->getId(), $recAC->getId() ];
 121                 sort( $expectedIds );
 122                 $expectedIds = array_reverse( $expectedIds );
 123
 124                 $this->assertSame( $expectedIds, $recs );
 125         }
 126
 127         /**
 128          * @covers \MediaWiki\Page\PageSelectQueryBuilder::whereTitles
 129          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecord
 130          */
 131         public function testFetchSinglePageByTitle() {
 132                 $pageStore = $this->getPageStore();
 133
 134                 $recAB = $pageStore->getPageByName( NS_MAIN, 'AB' );
 135
 136                 $rec = $pageStore->newSelectQueryBuilder()
 137                         ->whereTitles( NS_MAIN, 'AB' )
 138                         ->fetchPageRecord();
 139
 140                 $this->assertTrue( $recAB->isSamePageAs( $rec ) );
 141
 142                 // The page should have ended up in the LinkCache
 143                 $linkCache = $this->getServiceContainer()->getLinkCache();
 144                 $this->assertSame( $rec->getId(), $linkCache->getGoodLinkID( $rec ) );
 145
 146                 $rec = $pageStore->newSelectQueryBuilder()
 147                         ->whereTitles( NS_TALK, 'AB' )
 148                         ->fetchPageRecord();
 149
 150                 $this->assertNull( $rec );
 151         }
 152
 153         /**
 154          * @covers \MediaWiki\Page\PageSelectQueryBuilder::whereNamespace
 155          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecordArray
 156          */
 157         public function testFilterByNamespace() {
 158                 $pageStore = $this->getPageStore();
 159
 160                 $recAA = $pageStore->getPageByName( NS_TALK, 'AA' );
 161
 162                 $recs = $pageStore->newSelectQueryBuilder()
 163                         ->whereNamespace( NS_TALK )
 164                         ->fetchPageRecordArray();
 165
 166                 $this->assertCount( 1, $recs );
 167                 $this->assertSame( 'AA', $recs[ $recAA->getId() ]->getDBkey() );
 168         }
 169
 170         /**
 171          * @covers \MediaWiki\Page\PageSelectQueryBuilder::whereTitlePrefix
 172          * @covers \MediaWiki\Page\PageSelectQueryBuilder::fetchPageRecords
 173          * @covers \MediaWiki\Page\PageSelectQueryBuilder::orderByTitle
 174          */
 175         public function testListPagesByPrefix() {
 176                 $pageStore = $this->getPageStore();
 177
 178                 $recs = $pageStore->newSelectQueryBuilder()
 179                         ->whereTitlePrefix( NS_MAIN, 'A' )
 180                         ->orderByTitle( PageSelectQueryBuilder::SORT_DESC )
 181                         ->fetchPageRecords();
 182
 183                 $recs = iterator_to_array( $recs );
 184
 185                 $this->assertCount( 2, $recs );
 186
 187                 // descending order
 188                 $this->assertSame( 'AB', $recs[0]->getDBkey() );
 189                 $this->assertSame( 'AA', $recs[1]->getDBkey() );
 190
 191                 $recs = $pageStore->newSelectQueryBuilder()
 192                         ->whereTitlePrefix( NS_TALK, 'A' )
 193                         ->fetchPageRecords();
 194
 195                 $this->assertCount( 1, iterator_to_array( $recs ) );
 196
 197                 $recs = $pageStore->newSelectQueryBuilder()
 198                         ->whereTitlePrefix( NS_MAIN, 'XX' )
 199                         ->fetchPageRecords();
 200
 201                 $this->assertCount( 0, iterator_to_array( $recs ) );
 202         }
 203
 204 }