Mercurial > hg > nginx-tests
view stream_udp_limit_rate.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 with datagrams, limit rate directives. ############################################################################### 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(8) ->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } stream { %%TEST_GLOBALS_STREAM%% proxy_requests 2; proxy_responses 1; proxy_timeout 3s; server { listen 127.0.0.1:%%PORT_8982_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8980_UDP%%; } server { listen 127.0.0.1:%%PORT_8983_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8980_UDP%%; proxy_download_rate 500; } server { listen 127.0.0.1:%%PORT_8984_UDP%% udp; proxy_pass 127.0.0.1:%%PORT_8980_UDP%%; proxy_upload_rate 500; } } EOF $t->run_daemon(\&udp_daemon, port(8980), $t); $t->run(); $t->waitforfile($t->testdir . '/' . port(8980)); ############################################################################### my $str = '1234567890' x 100; # unlimited my $s = dgram('127.0.0.1:' . port(8982)); is($s->io($str), $str, 'unlimited'); is($s->io($str), $str, 'unlimited 2'); # datagram doesn't get split $s = dgram('127.0.0.1:' . port(8983)); is($s->io($str), $str, 'download'); my $t1 = time(); is($s->io($str), $str, 'download 2'); my $t2 = time(); cmp_ok($t1, '<', $t2, 'download 2 delayed'); # infinite event report before 1.15.9 $s = dgram('127.0.0.1:' . port(8984)); is($s->io($str), $str, 'upload'); is($s->io($str, read_timeout => 0.5), '', 'upload limited'); select undef, undef, undef, 1.6; is($s->io($str), $str, 'upload passed'); ############################################################################### sub udp_daemon { my ($port, $t) = @_; my $server = IO::Socket::INET->new( Proto => 'udp', LocalAddr => "127.0.0.1:$port", ) or die "Can't create listening socket: $!\n"; # signal we are ready open my $fh, '>', $t->testdir() . "/$port"; close $fh; while (1) { $server->recv(my $buffer, 65536); log2i("$server $buffer"); log2o("$server $buffer"); $server->send($buffer); } } sub log2i { Test::Nginx::log_core('|| <<', @_); } sub log2o { Test::Nginx::log_core('|| >>', @_); } sub log2c { Test::Nginx::log_core('||', @_); } ###############################################################################