3 namespace MediaWiki\Tests\Storage
;
7 use MediaWiki\Storage\RevisionAccessException
;
8 use MediaWiki\Storage\RevisionStore
;
9 use MediaWiki\Storage\SqlBlobStore
;
10 use MediaWikiTestCase
;
13 use Wikimedia\Rdbms\Database
;
14 use Wikimedia\Rdbms\LoadBalancer
;
16 class RevisionStoreTest
extends MediaWikiTestCase
{
19 * @param LoadBalancer $loadBalancer
20 * @param SqlBlobStore $blobStore
21 * @param WANObjectCache $WANObjectCache
23 * @return RevisionStore
25 private function getRevisionStore(
28 $WANObjectCache = null
30 return new RevisionStore(
31 $loadBalancer ?
$loadBalancer : $this->getMockLoadBalancer(),
32 $blobStore ?
$blobStore : $this->getMockSqlBlobStore(),
33 $WANObjectCache ?
$WANObjectCache : $this->getHashWANObjectCache()
38 * @return \PHPUnit_Framework_MockObject_MockObject|LoadBalancer
40 private function getMockLoadBalancer() {
41 return $this->getMockBuilder( LoadBalancer
::class )
42 ->disableOriginalConstructor()->getMock();
46 * @return \PHPUnit_Framework_MockObject_MockObject|Database
48 private function getMockDatabase() {
49 return $this->getMockBuilder( Database
::class )
50 ->disableOriginalConstructor()->getMock();
54 * @return \PHPUnit_Framework_MockObject_MockObject|SqlBlobStore
56 private function getMockSqlBlobStore() {
57 return $this->getMockBuilder( SqlBlobStore
::class )
58 ->disableOriginalConstructor()->getMock();
61 private function getHashWANObjectCache() {
62 return new WANObjectCache( [ 'cache' => new \
HashBagOStuff() ] );
66 * @covers \MediaWiki\Storage\RevisionStore::getContentHandlerUseDB
67 * @covers \MediaWiki\Storage\RevisionStore::setContentHandlerUseDB
69 public function testGetSetContentHandlerDb() {
70 $store = $this->getRevisionStore();
71 $this->assertTrue( $store->getContentHandlerUseDB() );
72 $store->setContentHandlerUseDB( false );
73 $this->assertFalse( $store->getContentHandlerUseDB() );
74 $store->setContentHandlerUseDB( true );
75 $this->assertTrue( $store->getContentHandlerUseDB() );
78 private function getDefaultQueryFields() {
94 private function getCommentQueryFields() {
96 'rev_comment_text' => 'rev_comment',
97 'rev_comment_data' => 'NULL',
98 'rev_comment_cid' => 'NULL',
102 private function getContentHandlerQueryFields() {
104 'rev_content_format',
109 public function provideGetQueryInfo() {
114 'tables' => [ 'revision' ],
115 'fields' => array_merge(
116 $this->getDefaultQueryFields(),
117 $this->getCommentQueryFields(),
118 $this->getContentHandlerQueryFields()
127 'tables' => [ 'revision' ],
128 'fields' => array_merge(
129 $this->getDefaultQueryFields(),
130 $this->getCommentQueryFields()
139 'tables' => [ 'revision', 'page' ],
140 'fields' => array_merge(
141 $this->getDefaultQueryFields(),
142 $this->getCommentQueryFields(),
153 'page' => [ 'INNER JOIN', [ 'page_id = rev_page' ] ],
161 'tables' => [ 'revision', 'user' ],
162 'fields' => array_merge(
163 $this->getDefaultQueryFields(),
164 $this->getCommentQueryFields(),
170 'user' => [ 'LEFT JOIN', [ 'rev_user != 0', 'user_id = rev_user' ] ],
178 'tables' => [ 'revision', 'text' ],
179 'fields' => array_merge(
180 $this->getDefaultQueryFields(),
181 $this->getCommentQueryFields(),
188 'text' => [ 'INNER JOIN', [ 'rev_text_id=old_id' ] ],
194 [ 'page', 'user', 'text' ],
196 'tables' => [ 'revision', 'page', 'user', 'text' ],
197 'fields' => array_merge(
198 $this->getDefaultQueryFields(),
199 $this->getCommentQueryFields(),
200 $this->getContentHandlerQueryFields(),
214 'page' => [ 'INNER JOIN', [ 'page_id = rev_page' ] ],
215 'user' => [ 'LEFT JOIN', [ 'rev_user != 0', 'user_id = rev_user' ] ],
216 'text' => [ 'INNER JOIN', [ 'rev_text_id=old_id' ] ],
223 * @dataProvider provideGetQueryInfo
224 * @covers \MediaWiki\Storage\RevisionStore::getQueryInfo
226 public function testGetQueryInfo( $contentHandlerUseDb, $options, $expected ) {
227 $store = $this->getRevisionStore();
228 $store->setContentHandlerUseDB( $contentHandlerUseDb );
229 $this->setMwGlobals( 'wgCommentTableSchemaMigrationStage', MIGRATION_OLD
);
230 $this->assertEquals( $expected, $store->getQueryInfo( $options ) );
233 private function getDefaultArchiveFields() {
254 * @covers \MediaWiki\Storage\RevisionStore::getArchiveQueryInfo
256 public function testGetArchiveQueryInfo_contentHandlerDb() {
257 $store = $this->getRevisionStore();
258 $store->setContentHandlerUseDB( true );
259 $this->setMwGlobals( 'wgCommentTableSchemaMigrationStage', MIGRATION_OLD
);
265 'fields' => array_merge(
266 $this->getDefaultArchiveFields(),
268 'ar_comment_text' => 'ar_comment',
269 'ar_comment_data' => 'NULL',
270 'ar_comment_cid' => 'NULL',
277 $store->getArchiveQueryInfo()
282 * @covers \MediaWiki\Storage\RevisionStore::getArchiveQueryInfo
284 public function testGetArchiveQueryInfo_noContentHandlerDb() {
285 $store = $this->getRevisionStore();
286 $store->setContentHandlerUseDB( false );
287 $this->setMwGlobals( 'wgCommentTableSchemaMigrationStage', MIGRATION_OLD
);
293 'fields' => array_merge(
294 $this->getDefaultArchiveFields(),
296 'ar_comment_text' => 'ar_comment',
297 'ar_comment_data' => 'NULL',
298 'ar_comment_cid' => 'NULL',
303 $store->getArchiveQueryInfo()
307 public function testGetTitle_successFromPageId() {
308 $mockLoadBalancer = $this->getMockLoadBalancer();
309 // Title calls wfGetDB() so we have to set the main service
310 $this->setService( 'DBLoadBalancer', $mockLoadBalancer );
312 $db = $this->getMockDatabase();
313 // Title calls wfGetDB() which uses a regular Connection
314 $mockLoadBalancer->expects( $this->atLeastOnce() )
315 ->method( 'getConnection' )
318 // First call to Title::newFromID, faking no result (db lag?)
319 $db->expects( $this->at( 0 ) )
320 ->method( 'selectRow' )
326 ->willReturn( (object)[
327 'page_namespace' => '1',
328 'page_title' => 'Food',
331 $store = $this->getRevisionStore( $mockLoadBalancer );
332 $title = $store->getTitle( 1, 2, RevisionStore
::READ_NORMAL
);
334 $this->assertSame( 1, $title->getNamespace() );
335 $this->assertSame( 'Food', $title->getDBkey() );
338 public function testGetTitle_successFromRevId() {
339 $mockLoadBalancer = $this->getMockLoadBalancer();
340 // Title calls wfGetDB() so we have to set the main service
341 $this->setService( 'DBLoadBalancer', $mockLoadBalancer );
343 $db = $this->getMockDatabase();
344 // Title calls wfGetDB() which uses a regular Connection
345 $mockLoadBalancer->expects( $this->atLeastOnce() )
346 ->method( 'getConnection' )
348 // RevisionStore getTitle uses a ConnectionRef
349 $mockLoadBalancer->expects( $this->atLeastOnce() )
350 ->method( 'getConnectionRef' )
353 // First call to Title::newFromID, faking no result (db lag?)
354 $db->expects( $this->at( 0 ) )
355 ->method( 'selectRow' )
361 ->willReturn( false );
363 // First select using rev_id, faking no result (db lag?)
364 $db->expects( $this->at( 1 ) )
365 ->method( 'selectRow' )
367 [ 'revision', 'page' ],
371 ->willReturn( (object)[
372 'page_namespace' => '1',
373 'page_title' => 'Food2',
376 $store = $this->getRevisionStore( $mockLoadBalancer );
377 $title = $store->getTitle( 1, 2, RevisionStore
::READ_NORMAL
);
379 $this->assertSame( 1, $title->getNamespace() );
380 $this->assertSame( 'Food2', $title->getDBkey() );
384 * @covers \MediaWiki\Storage\RevisionStore::getTitle
386 public function testGetTitle_throwsExceptionAfterFallbacks() {
387 $mockLoadBalancer = $this->getMockLoadBalancer();
388 // Title calls wfGetDB() so we have to set the main service
389 $this->setService( 'DBLoadBalancer', $mockLoadBalancer );
391 $db = $this->getMockDatabase();
392 // Title calls wfGetDB() which uses a regular Connection
393 $mockLoadBalancer->expects( $this->atLeastOnce() )
394 ->method( 'getConnection' )
396 // RevisionStore getTitle uses a ConnectionRef
397 $mockLoadBalancer->expects( $this->atLeastOnce() )
398 ->method( 'getConnectionRef' )
401 // First call to Title::newFromID, faking no result (db lag?)
402 $db->expects( $this->at( 0 ) )
403 ->method( 'selectRow' )
409 ->willReturn( false );
411 // First select using rev_id, faking no result (db lag?)
412 $db->expects( $this->at( 1 ) )
413 ->method( 'selectRow' )
415 [ 'revision', 'page' ],
419 ->willReturn( false );
421 $store = $this->getRevisionStore( $mockLoadBalancer );
423 $this->setExpectedException( RevisionAccessException
::class );
424 $store->getTitle( 1, 2, RevisionStore
::READ_NORMAL
);
427 public function provideNewRevisionFromRow_legacyEncoding_applied() {
428 yield
'windows-1252, old_flags is empty' => [
433 'old_text' => "S\xF6me Content",
438 yield
'windows-1252, old_flags is null' => [
443 'old_text' => "S\xF6me Content",
450 * @dataProvider provideNewRevisionFromRow_legacyEncoding_applied
452 * @covers \MediaWiki\Storage\RevisionStore::newRevisionFromRow
453 * @covers \MediaWiki\Storage\RevisionStore::newRevisionFromRow_1_29
455 public function testNewRevisionFromRow_legacyEncoding_applied( $encoding, $locale, $row, $text ) {
456 $cache = new WANObjectCache( [ 'cache' => new HashBagOStuff() ] );
458 $blobStore = new SqlBlobStore( wfGetLB(), $cache );
459 $blobStore->setLegacyEncoding( $encoding, Language
::factory( $locale ) );
461 $store = new RevisionStore( wfGetLB(), $blobStore, $cache );
463 $record = $store->newRevisionFromRow(
464 $this->makeRow( $row ),
466 Title
::newFromText( __METHOD__
. '-UTPage' )
469 $this->assertSame( $text, $record->getContent( 'main' )->serialize() );
473 * @covers \MediaWiki\Storage\RevisionStore::newRevisionFromRow
474 * @covers \MediaWiki\Storage\RevisionStore::newRevisionFromRow_1_29
476 public function testNewRevisionFromRow_legacyEncoding_ignored() {
478 'old_flags' => 'utf-8',
479 'old_text' => 'Söme Content',
482 $cache = new WANObjectCache( [ 'cache' => new HashBagOStuff() ] );
484 $blobStore = new SqlBlobStore( wfGetLB(), $cache );
485 $blobStore->setLegacyEncoding( 'windows-1252', Language
::factory( 'en' ) );
487 $store = new RevisionStore( wfGetLB(), $blobStore, $cache );
489 $record = $store->newRevisionFromRow(
490 $this->makeRow( $row ),
492 Title
::newFromText( __METHOD__
. '-UTPage' )
494 $this->assertSame( 'Söme Content', $record->getContent( 'main' )->serialize() );
497 private function makeRow( array $array ) {
502 'rev_timestamp' => '20110101000000',
503 'rev_user_text' => 'Tester',
505 'rev_minor_edit' => 0,
508 'rev_parent_id' => 0,
509 'rev_sha1' => 'deadbeef',
510 'rev_comment_text' => 'Testing',
511 'rev_comment_data' => '{}',
512 'rev_comment_cid' => 111,
513 'rev_content_format' => CONTENT_FORMAT_TEXT
,
514 'rev_content_model' => CONTENT_MODEL_TEXT
,
515 'page_namespace' => 0,
516 'page_title' => 'TEST',
519 'page_is_redirect' => 0,
521 'user_name' => 'Tester',
523 'old_text' => 'Hello World',
524 'old_flags' => 'utf-8',