Mercurial > hg > nginx-tests
changeset 934:d5762cd27ad9
Tests: added tests for proxy_protocol_port variable.
author | Andrey Zelenkov <zelenkov@nginx.com> |
---|---|
date | Mon, 23 May 2016 19:09:03 +0300 |
parents | e94d9436a37f |
children | 25a4b2fdd3fb |
files | proxy_protocol_port.t |
diffstat | 1 files changed, 110 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/proxy_protocol_port.t Mon May 23 19:09:03 2016 +0300 @@ -0,0 +1,110 @@ +#!/usr/bin/perl + +# (C) Andrey Zelenkov +# (C) Nginx, Inc. + +# Tests for proxy_protocol_port variable. + +############################################################################### + +use warnings; +use strict; + +use Test::More; + +use Socket qw/ CRLF /; + +BEGIN { use FindBin; chdir($FindBin::Bin); } + +use lib 'lib'; +use Test::Nginx; + +############################################################################### + +select STDERR; $| = 1; +select STDOUT; $| = 1; + +my $t = Test::Nginx->new()->has(qw/http ipv6 realip/) + ->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +http { + %%TEST_GLOBALS_HTTP%% + + log_format port $proxy_protocol_port; + + server { + listen 127.0.0.1:8080 proxy_protocol; + server_name localhost; + + add_header X-PP-Port $proxy_protocol_port; + add_header X-Remote-Port $remote_port; + + location /pp { + real_ip_header proxy_protocol; + error_page 404 =200 /t; + + location /pp/real { + set_real_ip_from 127.0.0.1/32; + } + } + + location /log { + access_log %%TESTDIR%%/port.log port; + } + } +} + +EOF + +$t->write_file('t', 'SEE-THIS'); +$t->try_run('no proxy_protocol_port')->plan(8); + +############################################################################### + +my $tcp4 = 'PROXY TCP4 192.0.2.1 192.0.2.2 1234 5678' . CRLF; +my $tcp6 = 'PROXY TCP6 2001:Db8::1 2001:Db8::2 1234 5678' . CRLF; +my $unk = 'PROXY UNKNOWN 1 2 3 4 5 6' . CRLF; + +# realip + +like(pp_get('/pp', $tcp4), qr/X-PP-Port: 1234/, 'pp port tcp4'); +like(pp_get('/pp', $tcp6), qr/X-PP-Port: 1234/, 'pp port tcp6'); +unlike(pp_get('/pp', $unk), qr/X-PP-Port/, 'pp port unknown'); + +# remote_port + +like(pp_get('/pp/real', $tcp4), qr/X-Remote-Port: 1234/, 'remote port tcp4'); +unlike(pp_get('/pp', $tcp4), qr/X-Remote-Port: 1234/, 'no remote port tcp4'); +like(pp_get('/pp/real', $tcp6), qr/X-Remote-Port: 1234/, 'remote port tcp6'); +unlike(pp_get('/pp', $tcp6), qr/X-Remote-Port: 1234/, 'no remote port tcp6'); + +# log + +pp_get('/log', $tcp4); + +$t->stop(); + +my $log = $t->read_file('/port.log'); +chomp $log; + +is($log, 1234, 'pp port log'); + +############################################################################### + +sub pp_get { + my ($url, $proxy) = @_; + return http($proxy . <<EOF); +GET $url HTTP/1.0 +Host: localhost + +EOF +} + +###############################################################################