blob: 2307bacc66c36b90c77a834167a5a2d5757aee7e (
plain) (
blame)
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
|
<?php
namespace MediaWiki\Tests\Rest;
use GuzzleHttp\Psr7\Stream;
use GuzzleHttp\Psr7\Uri;
use MediaWiki\MainConfigNames;
use MediaWiki\Rest\EntryPoint;
use MediaWiki\Rest\Handler;
use MediaWiki\Rest\RequestData;
use MediaWiki\Rest\RequestInterface;
use MediaWiki\Tests\MockEnvironment;
use MediaWikiIntegrationTestCase;
/**
* @covers \MediaWiki\Rest\EntryPoint
*/
class EntryPointTest extends MediaWikiIntegrationTestCase {
use RestTestTrait;
public function setUp(): void {
parent::setUp();
$this->overrideConfigValue( MainConfigNames::RestPath, '/rest' );
}
private function createRouter( RequestInterface $request ) {
return $this->newRouter( [
'request' => $request
] );
}
/**
* @param RequestData $request
* @param MockEnvironment $env
*
* @return EntryPoint
*/
private function getEntryPoint( RequestData $request, MockEnvironment $env ): EntryPoint {
$entryPoint = new EntryPoint(
$request,
$env->makeFauxContext(),
$env,
$this->getServiceContainer()
);
$entryPoint->setRouter( $this->createRouter( $request ) );
return $entryPoint;
}
public static function mockHandlerHeader() {
return new class extends Handler {
public function execute() {
$response = $this->getResponseFactory()->create();
$response->setHeader( 'Foo', 'Bar' );
return $response;
}
};
}
public function testHeader() {
$uri = '/rest/mock/v1/EntryPoint/header';
$request = new RequestData( [ 'uri' => new Uri( $uri ) ] );
$env = new MockEnvironment();
$env->setRequestInfo( $uri );
$entryPoint = $this->getEntryPoint(
$request,
$env
);
$entryPoint->enableOutputCapture();
$entryPoint->run();
$env->assertHeaderValue( 'Bar', 'Foo' );
$env->assertStatusCode( 200 );
}
public static function mockHandlerBodyRewind() {
return new class extends Handler {
public function execute() {
$response = $this->getResponseFactory()->create();
$stream = new Stream( fopen( 'php://memory', 'w+' ) );
$stream->write( 'hello' );
$response->setBody( $stream );
return $response;
}
};
}
/**
* Make sure EntryPoint rewinds a seekable body stream before reading.
*/
public function testBodyRewind() {
$uri = '/rest/mock/v1/EntryPoint/bodyRewind';
$request = new RequestData( [ 'uri' => new Uri( $uri ) ] );
$env = new MockEnvironment();
$env->setRequestInfo( $uri );
$entryPoint = $this->getEntryPoint(
$request,
$env
);
$entryPoint->enableOutputCapture();
$entryPoint->run();
// NOTE: MediaWikiEntryPoint::doPostOutputShutdown flushes all output buffers
$this->assertStringContainsString( 'hello', $entryPoint->getCapturedOutput() );
}
}
|
