126 lines
4.5 KiB
PHP
126 lines
4.5 KiB
PHP
<?php
|
|
|
|
use MediaWiki\MediaWikiServices;
|
|
use Wikimedia\Http\TelemetryHeadersInterface;
|
|
use Wikimedia\TestingAccessWrapper;
|
|
|
|
/**
|
|
* @covers \MWHttpRequest
|
|
*/
|
|
class MWHttpRequestTest extends PHPUnit\Framework\TestCase {
|
|
|
|
/**
|
|
* Feeds URI to test a long regular expression in MWHttpRequest::isValidURI
|
|
*/
|
|
public static function provideURI() {
|
|
/** Format: 'boolean expectation', 'URI to test', 'Optional message' */
|
|
return [
|
|
[ false, '¿non sens before!! http://a', 'Allow anything before URI' ],
|
|
|
|
# (http|https) - only two schemes allowed
|
|
[ true, 'http://www.example.org/' ],
|
|
[ true, 'https://www.example.org/' ],
|
|
[ true, 'http://www.example.org', 'URI without directory' ],
|
|
[ true, 'http://a', 'Short name' ],
|
|
[ true, 'http://étoile', 'Allow UTF-8 in hostname' ], # 'étoile' is french for 'star'
|
|
[ false, '\\host\directory', 'CIFS share' ],
|
|
[ false, 'gopher://host/dir', 'Reject gopher scheme' ],
|
|
[ false, 'telnet://host', 'Reject telnet scheme' ],
|
|
|
|
# :\/\/ - double slashes
|
|
[ false, 'http//example.org', 'Reject missing colon in protocol' ],
|
|
[ false, 'http:/example.org', 'Reject missing slash in protocol' ],
|
|
[ false, 'http:example.org', 'Must have two slashes' ],
|
|
# Following fail since hostname can be made of anything
|
|
[ false, 'http:///example.org', 'Must have exactly two slashes, not three' ],
|
|
|
|
# (\w+:{0,1}\w*@)? - optional user:pass
|
|
[ true, 'http://user@host', 'Username provided' ],
|
|
[ true, 'http://user:@host', 'Username provided, no password' ],
|
|
[ true, 'http://user:pass@host', 'Username and password provided' ],
|
|
|
|
# (\S+) - host part is made of anything not whitespaces
|
|
// commented these out in order to remove @group Broken
|
|
// @todo are these valid tests? if so, fix MWHttpRequest::isValidURI so it can handle them
|
|
// [ false, 'http://!"èèè¿¿¿~~\'', 'hostname is made of any non whitespace' ],
|
|
// [ false, 'http://exam:ple.org/', 'hostname cannot use colons!' ],
|
|
|
|
# (:[0-9]+)? - port number
|
|
[ true, 'http://example.org:80/' ],
|
|
[ true, 'https://example.org:80/' ],
|
|
[ true, 'http://example.org:443/' ],
|
|
[ true, 'https://example.org:443/' ],
|
|
|
|
# Part after the hostname is / or / with something else
|
|
[ true, 'http://example/#' ],
|
|
[ true, 'http://example/!' ],
|
|
[ true, 'http://example/:' ],
|
|
[ true, 'http://example/.' ],
|
|
[ true, 'http://example/?' ],
|
|
[ true, 'http://example/+' ],
|
|
[ true, 'http://example/=' ],
|
|
[ true, 'http://example/&' ],
|
|
[ true, 'http://example/%' ],
|
|
[ true, 'http://example/@' ],
|
|
[ true, 'http://example/-' ],
|
|
[ true, 'http://example//' ],
|
|
[ true, 'http://example/&' ],
|
|
|
|
# Fragment
|
|
[ true, 'http://exam#ple.org', ], # This one is valid, really!
|
|
[ true, 'http://example.org:80#anchor' ],
|
|
[ true, 'http://example.org/?id#anchor' ],
|
|
[ true, 'http://example.org/?#anchor' ],
|
|
|
|
[ false, 'http://a ¿non !!sens after', 'Allow anything after URI' ],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* T29854 : MWHttpRequest::isValidURI is too lax
|
|
* @dataProvider provideURI
|
|
* @covers \MWHttpRequest::isValidURI
|
|
*/
|
|
public function testIsValidUri( $expect, $uri, $message = '' ) {
|
|
$this->assertSame( $expect, MWHttpRequest::isValidURI( $uri ), $message );
|
|
}
|
|
|
|
public function testSetReverseProxy() {
|
|
$req = TestingAccessWrapper::newFromObject(
|
|
MediaWikiServices::getInstance()->getHttpRequestFactory()->create( 'https://example.org/path?query=string' )
|
|
);
|
|
$req->setReverseProxy( 'http://localhost:1234' );
|
|
$this->assertSame( 'http://localhost:1234/path?query=string', $req->url );
|
|
$this->assertSame( 'example.org', $req->reqHeaders['Host'] );
|
|
}
|
|
|
|
public function testItInjectsTelemetryHeaders() {
|
|
$telemetry = $this->createMock( TelemetryHeadersInterface::class );
|
|
$telemetry->expects( $this->once() )
|
|
->method( 'getRequestHeaders' )
|
|
->willReturn( [
|
|
'X-Request-Id' => 'request_identifier',
|
|
'tracestate' => 'tracestate_value',
|
|
'traceparent' => 'traceparent_value',
|
|
] );
|
|
|
|
$httpRequest = $this->getMockForAbstractClass(
|
|
MWHttpRequest::class,
|
|
[
|
|
'http://localhost/test',
|
|
[
|
|
'timeout' => 30,
|
|
'connectTimeout' => 30
|
|
]
|
|
]
|
|
);
|
|
$httpRequest->addTelemetry( $telemetry );
|
|
|
|
$accessWrapper = TestingAccessWrapper::newFromObject( $httpRequest );
|
|
$requestHeaders = $accessWrapper->reqHeaders;
|
|
|
|
$this->assertEquals( 'request_identifier', $requestHeaders['X-Request-Id'] );
|
|
$this->assertEquals( 'tracestate_value', $requestHeaders['tracestate'] );
|
|
$this->assertEquals( 'traceparent_value', $requestHeaders['traceparent'] );
|
|
}
|
|
}
|