ae9d8e6054
The idea is for all entry points to use the MediaWikiEntryPoint base class, to improve consistency and testability. Bug: T354216 Change-Id: I3678afe32c7c1a313d2dcb1808286c25ecd167eb
114 lines
2.6 KiB
PHP
114 lines
2.6 KiB
PHP
<?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
|
|
* @covers \MediaWiki\Rest\Router
|
|
*/
|
|
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/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/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() );
|
|
}
|
|
|
|
}
|