Mercurial > hg > nginx-tests
diff stream_udp_limit_conn.t @ 868:d2cb9ed7412e
Tests: stream udp tests.
author | Andrey Zelenkov <zelenkov@nginx.com> |
---|---|
date | Tue, 15 Mar 2016 16:20:29 +0300 |
parents | |
children | 6bb1f2ccd386 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/stream_udp_limit_conn.t @@ -0,0 +1,128 @@ +#!/usr/bin/perl + +# (C) Andrey Zelenkov +# (C) Nginx, Inc. + +# Tests for stream limit_conn module with datagrams. + +############################################################################### + +use warnings; +use strict; + +use Test::More; + +BEGIN { use FindBin; chdir($FindBin::Bin); } + +use lib 'lib'; +use Test::Nginx; +use Test::Nginx::Stream qw/ dgram /; + +############################################################################### + +select STDERR; $| = 1; +select STDOUT; $| = 1; + +my $t = Test::Nginx->new()->has(qw/stream stream_limit_conn shmem/) + ->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +stream { + limit_conn_zone $binary_remote_addr zone=zone:1m; + limit_conn_zone $binary_remote_addr zone=zone2:1m; + + proxy_responses 1; + proxy_timeout 1s; + + server { + listen 127.0.0.1:8081 udp; + proxy_pass 127.0.0.1:8080; + + limit_conn zone 1; + proxy_responses 2; + } + + server { + listen 127.0.0.1:8082 udp; + proxy_pass 127.0.0.1:8080; + limit_conn zone2 1; + } + + server { + listen 127.0.0.1:8083 udp; + proxy_pass 127.0.0.1:8080; + limit_conn zone 5; + } + + server { + listen 127.0.0.1:8084 udp; + proxy_pass 127.0.0.1:8081; + limit_conn zone2 1; + } + + server { + listen 127.0.0.1:8085 udp; + proxy_pass 127.0.0.1:8081; + limit_conn zone 1; + } +} + +EOF + +$t->try_run('no stream udp')->plan(8); +$t->run_daemon(\&udp_daemon, $t); +$t->waitforfile($t->testdir . '/8080'); + +############################################################################### + +my $str = '.'; + +# same and other zones + +my $s = dgram('127.0.0.1:8081'); + +is($s->io('1'), '1', 'passed'); + +is(dgram('127.0.0.1:8081')->io('1'), '', 'rejected same zone'); +is(dgram('127.0.0.1:8082')->io('1'), '1', 'passed different zone'); +is(dgram('127.0.0.1:8083')->io('1'), '1', 'passed same zone unlimited'); + +is($s->io('2', read => 2), '12', 'long connection'); + +is(dgram('127.0.0.1:8081')->io('2', read => 2), '12', 'passed 2'); + +# zones proxy chain + +is(dgram('127.0.0.1:8084')->io('1'), '1', 'passed proxy'); +is(dgram('127.0.0.1:8085')->io('1'), '', 'rejected proxy'); + +############################################################################### + +sub udp_daemon { + my $t = shift; + + my $server = IO::Socket::INET->new( + Proto => 'udp', + LocalAddr => '127.0.0.1:8080', + Reuse => 1, + ) + or die "Can't create listening socket: $!\n"; + + # signal we are ready + + open my $fh, '>', $t->testdir() . '/8080'; + close $fh; + + while (1) { + $server->recv(my $buffer, 65536); + $server->send($_) for (1 .. $buffer); + } +} + +###############################################################################