Mercurial > hg > nginx-tests
view stream_udp_limit_conn.t @ 1236:93f749c1d5c5
Tests: fixed parallel tests execution with UDP.
Previously, when checking ports availability, a UDP socket was always created
first, then a TCP socket was created. On success, one of UDP and TCP sockets
was closed (depending on the "udp" option) and the second one was used to busy
this port in other scripts. This lead to the following problem: in an attempt
to reopen a UDP socket used in a given testing script it could be stolen by
another script as part of checking ports availability.
To solve this problem, UDP and TCP ports were split into two non-overlapping
ranges: TCP ports are only used in the range 8000-8499, and UDP ports - in
the range 8500-8999. In addition, the order of creating sockets in UDP tests
has been reversed: now a TCP socket used as a lock precedes a UDP socket.
author | Andrey Zelenkov <zelenkov@nginx.com> |
---|---|
date | Thu, 26 Oct 2017 18:00:21 +0300 |
parents | 196d33c2bb45 |
children | e4974af3fb12 |
line wrap: on
line source
#!/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 udp/)->plan(9) ->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:%%PORT_8081_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8080_UDP%%; limit_conn zone 1; proxy_responses 2; } server { listen 127.0.0.1:%%PORT_8082_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8080_UDP%%; limit_conn zone2 1; } server { listen 127.0.0.1:%%PORT_8083_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8080_UDP%%; limit_conn zone 5; } server { listen 127.0.0.1:%%PORT_8084_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8081_UDP%%; limit_conn zone2 1; } server { listen 127.0.0.1:%%PORT_8085_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8081_UDP%%; limit_conn zone 1; } } EOF $t->run(); $t->run_daemon(\&udp_daemon, $t); $t->waitforfile($t->testdir . '/' . port(8080)); ############################################################################### # same and other zones my $s = dgram('127.0.0.1:' . port(8081)); is($s->io('1'), '1', 'passed'); # if not all responses were sent to client, then new request # in same socket will be treated as new connection is($s->io('1', read_timeout => 0.1), '', 'rejected new connection'); is(dgram('127.0.0.1:' . port(8081))->io('1', read_timeout => 0.1), '', 'rejected same zone'); is(dgram('127.0.0.1:' . port(8082))->io('1'), '1', 'passed different zone'); is(dgram('127.0.0.1:' . port(8083))->io('1'), '1', 'passed same zone unlimited'); sleep 1; # waiting for proxy_timeout to expire is($s->io('2', read => 2), '12', 'new connection after proxy_timeout'); is(dgram('127.0.0.1:' . port(8081))->io('2', read => 2), '12', 'passed 2'); # zones proxy chain is(dgram('127.0.0.1:' . port(8084))->io('1'), '1', 'passed proxy'); is(dgram('127.0.0.1:' . port(8085))->io('1', read_timeout => 0.1), '', 'rejected proxy'); ############################################################################### sub udp_daemon { my $t = shift; my $server = IO::Socket::INET->new( Proto => 'udp', LocalAddr => '127.0.0.1:' . port(8080), Reuse => 1, ) or die "Can't create listening socket: $!\n"; # signal we are ready open my $fh, '>', $t->testdir() . '/' . port(8080); close $fh; while (1) { $server->recv(my $buffer, 65536); $server->send($_) for (1 .. $buffer); } } ###############################################################################