4 * @covers PageDataRequestHandler
7 class PageDataRequestHandlerTest
extends \MediaWikiTestCase
{
12 private $interfaceTitle;
19 protected function setUp() {
22 $this->interfaceTitle
= Title
::newFromText( "Special:PageDataRequestHandlerTest" );
24 $this->obLevel
= ob_get_level();
27 protected function tearDown() {
28 $obLevel = ob_get_level();
30 while ( ob_get_level() > $this->obLevel
) {
34 if ( $obLevel !== $this->obLevel
) {
35 $this->fail( "Test changed output buffer level: was {$this->obLevel}" .
36 "before test, but $obLevel after test."
44 * @return PageDataRequestHandler
46 protected function newHandler() {
47 return new PageDataRequestHandler( 'json' );
51 * @param array $params
52 * @param string[] $headers
56 protected function makeOutputPage( array $params, array $headers ) {
58 $request = new FauxRequest( $params );
59 $request->response()->header( 'Status: 200 OK', true, 200 ); // init/reset
61 foreach ( $headers as $name => $value ) {
62 $request->setHeader( strtoupper( $name ), $value );
65 // construct Context and OutputPage
66 $context = new DerivativeContext( RequestContext
::getMain() );
67 $context->setRequest( $request );
69 $output = new OutputPage( $context );
70 $output->setTitle( $this->interfaceTitle
);
71 $context->setOutput( $output );
76 public function handleRequestProvider() {
79 $cases[] = [ '', [], [], '!!', 400 ];
81 $cases[] = [ '', [ 'target' => 'Helsinki' ], [], '!!', 303, [ 'Location' => '!.+!' ] ];
84 foreach ( $cases as $c ) {
88 if ( isset( $case[1]['target'] ) ) {
89 $case[0] .= $case[1]['target'];
90 unset( $case[1]['target'] );
93 $subpageCases[] = $case;
96 $cases = array_merge( $cases, $subpageCases );
100 [ 'target' => 'Helsinki' ],
101 [ 'Accept' => 'text/HTML' ],
104 [ 'Location' => '!Helsinki$!' ]
110 'target' => 'Helsinki',
111 'revision' => '4242',
113 [ 'Accept' => 'text/HTML' ],
116 [ 'Location' => '!Helsinki(\?|&)oldid=4242!' ]
125 [ 'Location' => '!Helsinki&action=raw!' ]
128 // #31: /Q5 with "Accept: text/foobar" triggers a 406
132 [ 'Accept' => 'text/foobar' ],
141 [ 'Accept' => 'text/HTML' ],
144 [ 'Location' => '!Helsinki$!' ]
150 [ 'Accept' => 'text/HTML' ],
153 [ 'Location' => '!Helsinki$!' ]
159 [ 'Accept' => 'text/HTML' ],
162 [ 'Location' => '!AC/DC$!' ]
169 * @dataProvider handleRequestProvider
171 * @param string $subpage The subpage to request (or '')
172 * @param array $params Request parameters
173 * @param array $headers Request headers
174 * @param string $expectedOutput Regex to match the output against.
175 * @param int $expectedStatusCode Expected HTTP status code.
176 * @param string[] $expectedHeaders Expected HTTP response headers.
178 public function testHandleRequest(
183 $expectedStatusCode = 200,
184 array $expectedHeaders = []
186 $output = $this->makeOutputPage( $params, $headers );
187 $request = $output->getRequest();
189 /* @var FauxResponse $response */
190 $response = $request->response();
193 $handler = $this->newHandler();
197 $handler->handleRequest( $subpage, $request, $output );
199 if ( $output->getRedirect() !== '' ) {
200 // hack to apply redirect to web response
204 $text = ob_get_contents();
207 $this->assertEquals( $expectedStatusCode, $response->getStatusCode(), 'status code' );
208 $this->assertRegExp( $expectedOutput, $text, 'output' );
210 foreach ( $expectedHeaders as $name => $exp ) {
211 $value = $response->getHeader( $name );
212 $this->assertNotNull( $value, "header: $name" );
213 $this->assertInternalType( 'string', $value, "header: $name" );
214 $this->assertRegExp( $exp, $value, "header: $name" );
216 } catch ( HttpError
$e ) {
218 $this->assertEquals( $expectedStatusCode, $e->getStatusCode(), 'status code' );
219 $this->assertRegExp( $expectedOutput, $e->getHTML(), 'error output' );
222 // We always set "Access-Control-Allow-Origin: *"
223 $this->assertSame( '*', $response->getHeader( 'Access-Control-Allow-Origin' ) );
226 public function provideHttpContentNegotiation() {
227 $helsinki = Title
::newFromText( 'Helsinki' );
229 'Accept Header of HTML' => [
231 [ 'ACCEPT' => 'text/html' ], // headers
234 'Accept Header without weights' => [
236 [ 'ACCEPT' => '*/*, text/html, text/x-wiki' ],
237 'Helsinki&action=raw'
239 'Accept Header with weights' => [
241 [ 'ACCEPT' => 'text/*; q=0.5, text/json; q=0.7, application/rdf+xml; q=0.8' ],
242 'Helsinki&action=raw'
244 'Accept Header accepting evertyhing and HTML' => [
246 [ 'ACCEPT' => 'text/html, */*' ],
247 'Helsinki&action=raw'
249 'No Accept Header' => [
252 'Helsinki&action=raw'
258 * @dataProvider provideHttpContentNegotiation
260 * @param Title $title
261 * @param array $headers Request headers
262 * @param string $expectedRedirectSuffix Expected suffix of the HTTP Location header.
266 public function testHttpContentNegotiation(
269 $expectedRedirectSuffix
271 /* @var FauxResponse $response */
272 $output = $this->makeOutputPage( [], $headers );
273 $request = $output->getRequest();
275 $handler = $this->newHandler();
276 $handler->httpContentNegotiation( $request, $output, $title );
278 $this->assertStringEndsWith(
279 $expectedRedirectSuffix,
280 $output->getRedirect(),