1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
<?php
use MediaWiki\Json\JsonCodec;
use MediaWiki\Parser\ParserOptions;
use MediaWiki\Parser\RevisionOutputCache;
use MediaWiki\PoolCounter\PoolWorkArticleViewOld;
use MediaWiki\Revision\RevisionRecord;
use MediaWiki\Status\Status;
use Psr\Log\NullLogger;
use Wikimedia\ObjectCache\BagOStuff;
use Wikimedia\ObjectCache\HashBagOStuff;
use Wikimedia\ObjectCache\WANObjectCache;
use Wikimedia\Stats\StatsFactory;
use Wikimedia\UUID\GlobalIdGenerator;
/**
* @covers \MediaWiki\PoolCounter\PoolWorkArticleViewOld
* @group Database
*/
class PoolWorkArticleViewOldTest extends PoolWorkArticleViewTest {
/** @var RevisionOutputCache */
private $cache = null;
/**
* @param WikiPage $page
* @param RevisionRecord|null $rev
* @param ParserOptions|null $options
*
* @return PoolWorkArticleViewOld
*/
protected function newPoolWorkArticleView(
WikiPage $page,
?RevisionRecord $rev = null,
$options = null
) {
if ( !$options ) {
$options = ParserOptions::newFromAnon();
}
if ( !$rev ) {
$rev = $page->getRevisionRecord();
}
if ( !$this->cache ) {
$this->installRevisionOutputCache();
}
$renderer = $this->getServiceContainer()->getRevisionRenderer();
return new PoolWorkArticleViewOld(
'test:' . $rev->getId(),
$this->cache,
$rev,
$options,
$renderer,
$this->getLoggerSpi()
);
}
/**
* @param BagOStuff|null $bag
*
* @return RevisionOutputCache
*/
private function installRevisionOutputCache( $bag = null ) {
$globalIdGenerator = $this->createMock( GlobalIdGenerator::class );
$globalIdGenerator->method( 'newUUIDv1' )->willReturn( 'uuid-uuid' );
$this->cache = new RevisionOutputCache(
'test',
new WANObjectCache( [ 'cache' => $bag ?: new HashBagOStuff() ] ),
60 * 60,
'20200101223344',
new JsonCodec(),
StatsFactory::newNull(),
new NullLogger(),
$globalIdGenerator
);
return $this->cache;
}
public function testUpdateCachedOutput() {
$options = ParserOptions::newFromAnon();
$page = $this->getExistingTestPage( __METHOD__ );
$cache = $this->installRevisionOutputCache();
$work = $this->newPoolWorkArticleView( $page, null, $options );
/** @var Status $status */
$status = $work->execute();
$this->assertStatusGood( $status );
$cachedOutput = $cache->get( $page->getRevisionRecord(), $options );
$this->assertNotEmpty( $cachedOutput );
$this->assertSame( $status->getValue()->getRawText(),
$cachedOutput->getRawText() );
}
public function testDoesNotCacheNotSafe() {
$page = $this->getExistingTestPage( __METHOD__ );
$cache = $this->installRevisionOutputCache();
$parserOptions = ParserOptions::newFromAnon();
$parserOptions->setWrapOutputClass( 'wrapwrap' ); // Not safe to cache!
$work = $this->newPoolWorkArticleView( $page, null, $parserOptions );
/** @var Status $status */
$status = $work->execute();
$this->assertStatusGood( $status );
$this->assertFalse( $cache->get( $page->getRevisionRecord(), $parserOptions ) );
}
public function testDoWorkWithFakeRevision() {
// PoolWorkArticleViewOld caches the results, but things with null revid should
// not be cached.
$this->expectException( InvalidArgumentException::class );
parent::testDoWorkWithFakeRevision();
}
}
|
