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
|
<?php
namespace MediaWiki\Tests\Rest\Handler;
use Exception;
use MediaWiki\Block\BlockErrorFormatter;
use MediaWiki\Context\IContextSource;
use MediaWiki\Edit\ParsoidOutputStash;
use MediaWiki\Edit\ParsoidRenderID;
use MediaWiki\Edit\SimpleParsoidOutputStash;
use MediaWiki\Permissions\Authority;
use MediaWiki\Permissions\UserAuthority;
use MediaWiki\Request\FauxRequest;
use MediaWiki\Rest\RequestData;
use MediaWiki\User\User;
use Wikimedia\ObjectCache\HashBagOStuff;
use WikiPage;
/**
* This trait is used in PageHTMLHandlerTest.php & RevisionHTMLHandlerTest.php
* to construct requests and perform stashing for the Parsoid Output stash feature.
*/
trait HTMLHandlerTestTrait {
/** @var ParsoidOutputStash|null */
private $parsoidOutputStash = null;
private function getParsoidOutputStash(): ParsoidOutputStash {
if ( !$this->parsoidOutputStash ) {
$chFactory = $this->getServiceContainer()->getContentHandlerFactory();
$this->parsoidOutputStash = new SimpleParsoidOutputStash( $chFactory, new HashBagOStuff(), 120 );
}
return $this->parsoidOutputStash;
}
private function getAuthority(): Authority {
$services = $this->getServiceContainer();
return new UserAuthority(
// We need a newly created user because we want IP and newbie to apply.
new User(),
new FauxRequest(),
$this->createMock( IContextSource::class ),
$services->getPermissionManager(),
$services->getRateLimiter(),
$this->createMock( BlockErrorFormatter::class )
);
}
/**
* @param WikiPage $page
* @param array $queryParams
* @param array $config
*
* @return array
* @throws Exception
*/
private function executePageHTMLRequest(
WikiPage $page,
array $queryParams = [],
array $config = [],
?Authority $authority = null
): array {
$handler = $this->newHandler();
$request = new RequestData( [
'pathParams' => [ 'title' => $page->getTitle()->getPrefixedDBkey() ],
'queryParams' => $queryParams,
] );
$result = $this->executeHandler(
$handler,
$request,
$config + [ 'format' => 'html' ],
[],
[],
[],
$authority
);
$etag = $result->getHeaderLine( 'ETag' );
$stashKey = ParsoidRenderID::newFromETag( $etag );
return [ $result->getBody()->getContents(), $etag, $stashKey ];
}
/**
* @param int $revId
* @param array $queryParams
* @param array $config
*
* @return array
* @throws Exception
*/
private function executeRevisionHTMLRequest(
int $revId,
array $queryParams = [],
array $config = [],
?Authority $authority = null
): array {
$handler = $this->newHandler();
$request = new RequestData( [
'pathParams' => [ 'id' => $revId ],
'queryParams' => $queryParams,
] );
$result = $this->executeHandler(
$handler,
$request,
$config + [ 'format' => 'html' ],
[],
[],
[],
$authority
);
$etag = $result->getHeaderLine( 'ETag' );
$stashKey = ParsoidRenderID::newFromETag( $etag );
return [ $result->getBody()->getContents(), $etag, $stashKey ];
}
}
|
