View source
<?php
namespace Drupal\Tests\Core\StackMiddleware;
use Drupal\Core\Site\Settings;
use Drupal\Core\StackMiddleware\ReverseProxyMiddleware;
use Drupal\Tests\UnitTestCase;
use Symfony\Component\HttpFoundation\Request;
class ReverseProxyMiddlewareTest extends UnitTestCase {
protected $mockHttpKernel;
protected function setUp() : void {
$this->mockHttpKernel = $this
->createMock('Symfony\\Component\\HttpKernel\\HttpKernelInterface');
}
public function testNoProxy() {
$settings = new Settings([]);
$this
->assertEquals(0, $settings
->get('reverse_proxy'));
$middleware = new ReverseProxyMiddleware($this->mockHttpKernel, $settings);
$request = $this
->getMockBuilder('Symfony\\Component\\HttpFoundation\\Request')
->setMethods([
'setTrustedProxies',
])
->getMock();
$request
->expects($this
->never())
->method('setTrustedProxies');
$middleware
->handle($request);
}
public function testReverseProxyEnabled($provided_settings, $expected_trusted_header_set) {
$settings = new Settings([
'reverse_proxy' => 1,
] + $provided_settings);
$this
->trustedHeadersAreSet($settings, $expected_trusted_header_set);
}
public function reverseProxyEnabledProvider() {
return [
'Proxy with default trusted headers' => [
[
'reverse_proxy_addresses' => [
'127.0.0.2',
'127.0.0.3',
],
],
Request::HEADER_FORWARDED | Request::HEADER_X_FORWARDED_FOR | Request::HEADER_X_FORWARDED_HOST | Request::HEADER_X_FORWARDED_PORT | Request::HEADER_X_FORWARDED_PROTO,
],
'Proxy with AWS trusted headers' => [
[
'reverse_proxy_addresses' => [
'127.0.0.2',
'127.0.0.3',
],
'reverse_proxy_trusted_headers' => Request::HEADER_X_FORWARDED_AWS_ELB,
],
Request::HEADER_X_FORWARDED_AWS_ELB,
],
'Proxy with custom trusted headers' => [
[
'reverse_proxy_addresses' => [
'127.0.0.2',
'127.0.0.3',
],
'reverse_proxy_trusted_headers' => Request::HEADER_X_FORWARDED_FOR | Request::HEADER_X_FORWARDED_HOST,
],
Request::HEADER_X_FORWARDED_FOR | Request::HEADER_X_FORWARDED_HOST,
],
];
}
protected function trustedHeadersAreSet(Settings $settings, $expected_trusted_header_set) {
$middleware = new ReverseProxyMiddleware($this->mockHttpKernel, $settings);
$request = new Request();
$middleware
->handle($request);
$this
->assertSame($settings
->get('reverse_proxy_addresses'), $request
->getTrustedProxies());
$this
->assertSame($expected_trusted_header_set, $request
->getTrustedHeaderSet());
}
}