ReverseProxyMiddlewareTest.php in Zircon Profile 8
File
core/tests/Drupal/Tests/Core/StackMiddleware/ReverseProxyMiddlewareTest.php
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;
public function setUp() {
$this->mockHttpKernel = $this
->getMock('Symfony\\Component\\HttpKernel\\HttpKernelInterface');
}
public function testNoProxy() {
$settings = new Settings(array());
$this
->assertEquals(0, $settings
->get('reverse_proxy'));
$middleware = new ReverseProxyMiddleware($this->mockHttpKernel, $settings);
$request = $this
->getMock('Symfony\\Component\\HttpFoundation\\Request', array(
'setTrustedHeaderName',
'setTrustedProxies',
));
$request
->expects($this
->never())
->method('setTrustedHeaderName');
$middleware
->handle($request);
}
public function testReverseProxyEnabled($provided_settings) {
$settings = new Settings(array(
'reverse_proxy' => 1,
) + $provided_settings);
$this
->trustedHeadersAreSet($settings);
}
public function reverseProxyEnabledProvider() {
return array(
array(
array(
'reverse_proxy_header' => 'X_FORWARDED_FOR_CUSTOMIZED',
'reverse_proxy_proto_header' => 'X_FORWARDED_PROTO_CUSTOMIZED',
'reverse_proxy_host_header' => 'X_FORWARDED_HOST_CUSTOMIZED',
'reverse_proxy_port_header' => 'X_FORWARDED_PORT_CUSTOMIZED',
'reverse_proxy_forwarded_header' => 'FORWARDED_CUSTOMIZED',
'reverse_proxy_addresses' => array(
'127.0.0.2',
'127.0.0.3',
),
),
),
);
}
protected function trustedHeadersAreSet(Settings $settings) {
$middleware = new ReverseProxyMiddleware($this->mockHttpKernel, $settings);
$request = new Request();
$middleware
->handle($request);
$this
->assertSame($settings
->get('reverse_proxy_header'), $request
->getTrustedHeaderName($request::HEADER_CLIENT_IP));
$this
->assertSame($settings
->get('reverse_proxy_proto_header'), $request
->getTrustedHeaderName($request::HEADER_CLIENT_PROTO));
$this
->assertSame($settings
->get('reverse_proxy_host_header'), $request
->getTrustedHeaderName($request::HEADER_CLIENT_HOST));
$this
->assertSame($settings
->get('reverse_proxy_port_header'), $request
->getTrustedHeaderName($request::HEADER_CLIENT_PORT));
$this
->assertSame($settings
->get('reverse_proxy_forwarded_header'), $request
->getTrustedHeaderName($request::HEADER_FORWARDED));
$this
->assertSame($settings
->get('reverse_proxy_addresses'), $request
->getTrustedProxies());
}
}