Mercurial > hg > nginx-tests
view stream_udp_proxy_requests.t @ 1990:85d88cd5091c
Tests: SSI include stub tests.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Thu, 04 Jul 2024 17:42:55 +0300 |
parents | f3ba4c74de31 |
children |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for stream proxy module, the proxy_requests directive. ############################################################################### 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 udp/)->plan(26) ->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } stream { %%TEST_GLOBALS_STREAM%% proxy_timeout 2100ms; log_format status $status; server { listen 127.0.0.1:%%PORT_8980_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8990_UDP%%; proxy_requests 0; } server { listen 127.0.0.1:%%PORT_8981_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8990_UDP%%; proxy_requests 1; } server { listen 127.0.0.1:%%PORT_8982_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8990_UDP%%; proxy_requests 2; } server { listen 127.0.0.1:%%PORT_8983_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8990_UDP%%; } server { listen 127.0.0.1:%%PORT_8984_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8991_UDP%%; proxy_requests 2; access_log %%TESTDIR%%/s.log status; } server { listen 127.0.0.1:%%PORT_8985_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8990_UDP%%; proxy_requests 2; proxy_responses 2; access_log %%TESTDIR%%/s2.log status; } } EOF $t->run_daemon(\&udp_daemon, $t, port(8990)); $t->run_daemon(\&udp_daemon, $t, port(8991)); $t->run(); $t->waitforfile($t->testdir . '/' . port(8990)); $t->waitforfile($t->testdir . '/' . port(8991)); ############################################################################### # proxy_requests 0, binding is not dropped across streams my $s = dgram('127.0.0.1:' . port(8980)); my $n = $s->io('1', read => 1); ok($n, 'requests 0 create'); is($s->read(), '1', 'requests 0 create - response'); is($s->io('1', read => 1), $n, 'requests 0 second - binding saved'); is($s->read(), '1', 'requests 0 second - response'); is($s->io('1', read => 1), $n, 'requests 0 follow - binding saved'); is($s->read(), '1', 'requests 0 follow - response'); # proxy_requests 1, binding is dropped on every next stream $s = dgram('127.0.0.1:' . port(8981)); $n = $s->io('1', read => 1); ok($n, 'requests 1 create'); is($s->read(), '1', 'requests 1 create - response'); isnt($s->io('1', read => 1), $n, 'requests 1 second - binding lost'); is($s->read(), '1', 'requests 1 second - response'); # proxy_requests 2, binding is dropped on every second stream $s = dgram('127.0.0.1:' . port(8982)); $n = $s->io('1', read => 1); ok($n, 'requests 2 create'); is($s->read(), '1', 'requests 2 create - response'); is($s->io('1', read => 1), $n, 'requests 2 second - binding saved'); is($s->read(), '1', 'requests 2 second - response'); isnt($s->io('1', read => 1), $n, 'requests 2 follow - binding lost'); is($s->read(), '1', 'requests 2 follow - response'); # proxy_requests unset, binding is not dropped across streams $s = dgram('127.0.0.1:' . port(8983)); $n = $s->io('1', read => 1); ok($n, 'requests unset create'); is($s->read(), '1', 'requests unset create - response'); is($s->io('1', read => 1), $n, 'requests unset second - binding saved'); is($s->read(), '1', 'requests unset second - response'); is($s->io('1', read => 1), $n, 'requests unset follow - binding saved'); is($s->read(), '1', 'requests unset follow - response'); # proxy_requests 2, with slow backend # client sends 5 packets, each responded with 3 packets # expects all packets proxied from backend, the last (uneven) session succeed $s = dgram('127.0.0.1:' . port(8984)); $s->write('2') for 1 .. 5; my $b = join ' ', map { $s->read() } (1 .. 15); like($b, qr/^(\d+ 1 2) \1 (?!\1)(\d+ 1 2) \2 (?!\2)\d+ 1 2$/, 'slow backend'); # proxy_requests 2, proxy_responses 2 # client sends 5 packets, each responded with 2 packets # expects all packets proxied from backend, the last (uneven) session succeed $s = dgram('127.0.0.1:' . port(8985)); $s->write('1') for 1 .. 5; $b = join ' ', map { $s->read() } (1 .. 10); SKIP: { skip 'session could early terminate', 1 unless $ENV{TEST_NGINX_UNSAFE}; like($b, qr/^(\d+ 1) \1 (?!\1)(\d+ 1) \2 (?!\2)\d+ 1$/, 'requests - responses'); } $t->stop(); is($t->read_file('s.log'), <<EOF, 'uneven session status - slow backend'); 200 200 200 EOF is($t->read_file('s2.log'), <<EOF, 'uneven session status - responses'); 200 200 200 EOF ############################################################################### sub udp_daemon { my ($t, $port) = @_; my $server = IO::Socket::INET->new( Proto => 'udp', LocalAddr => "127.0.0.1:$port", Reuse => 1 ) or die "Can't create listening socket: $!\n"; # signal we are ready open my $fh, '>', $t->testdir() . "/$port"; close $fh; my $slp = 1 if $port == port(8991); while (1) { $server->recv(my $buffer, 65536); sleep 1, $slp = 0 if $slp; $server->send($server->peerport()); $server->send($_) for (1 .. $buffer); } } ###############################################################################