nginx-tests: stream_limit

annotate stream_limit_rate.t @ 637:443848f28a73

Tests: whitespace fix.

author	Maxim Dounin <mdounin@mdounin.ru>
date	Tue, 28 Jul 2015 01:46:32 +0300
parents	40675bfad7d3
children	aed139eaaf3f

rev	line source
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	1 #!/usr/bin/perl
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	2
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	3 # (C) Andrey Zelenkov
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	4 # (C) Nginx, Inc.
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	5
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	6 # Tests for stream proxy module, limit rate directives.
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	7
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	8 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	9
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	10 use warnings;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	11 use strict;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	12
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	13 use Test::More;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	14
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	15 use IO::Select;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	16
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	17 BEGIN { use FindBin; chdir($FindBin::Bin); }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	18
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	19 use lib 'lib';
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	20 use Test::Nginx;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	21
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	22 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	23
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	24 select STDERR; $\| = 1;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	25 select STDOUT; $\| = 1;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	26
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	27 my $t = Test::Nginx->new()->has(qw/stream/)
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	28 ->write_file_expand('nginx.conf', <<'EOF');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	29
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	30 %%TEST_GLOBALS%%
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	31
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	32 daemon off;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	33
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	34 events {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	35 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	36
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	37 stream {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	38 # download and upload rates are set equal to the maximum
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	39 # number of bytes transmitted
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	40
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	41 # proxy_download_rate value comes from following calculations:
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	42 # test string length (1000) + whitespace (1) + time string length (10)
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	43
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	44 proxy_download_rate 1011;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	45 proxy_upload_rate 1000;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	46
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	47 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	48 listen 127.0.0.1:8081;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	49 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	50 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	51
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	52 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	53 listen 127.0.0.1:8082;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	54 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	55 proxy_download_rate 0;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	56 proxy_upload_rate 0;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	57 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	58
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	59 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	60 listen 127.0.0.1:8083;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	61 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	62 proxy_download_rate 1;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	63 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	64
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	65 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	66 listen 127.0.0.1:8084;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	67 proxy_pass 127.0.0.1:8080;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	68 proxy_upload_rate 1;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	69 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	70
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	71 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	72 listen 127.0.0.1:8085;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	73 proxy_pass 127.0.0.1:8080;
634 40675bfad7d3 Tests: fixed stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: 632 diff changeset	74 proxy_download_rate 250;
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	75 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	76
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	77 server {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	78 listen 127.0.0.1:8086;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	79 proxy_pass 127.0.0.1:8090;
634 40675bfad7d3 Tests: fixed stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: 632 diff changeset	80 proxy_upload_rate 250;
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	81 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	82 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	83
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	84 EOF
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	85
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	86 $t->run_daemon(\&stream_daemon, 8080);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	87 $t->run_daemon(\&stream_daemon, 8090);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	88
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	89 $t->try_run('no proxy_download_rate and/or proxy_upload_rate')->plan(8);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	90
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	91 $t->waitforsocket('127.0.0.1:8080');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	92 $t->waitforsocket('127.0.0.1:8090');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	93
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	94 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	95
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	96 my $str = '1234567890' x 100;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	97
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	98 my %r = stream_get($str, peer => '127.0.0.1:8081');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	99 is($r{'data'}, $str, 'exact limit');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	100
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	101 %r = stream_get($str, peer => '127.0.0.1:8082');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	102 is($r{'data'}, $str, 'unlimited');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	103
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	104 SKIP: {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	105 skip 'unsafe on VM', 2 unless $ENV{TEST_NGINX_UNSAFE};
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	106
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	107 # if interaction between backend and client is slow then proxy can add extra
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	108 # bytes to upload/download data
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	109
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	110 %r = stream_get($str, peer => '127.0.0.1:8083', readonce => 1);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	111 is($r{'data'}, '1', 'download - one byte');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	112
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	113 %r = stream_get($str, peer => '127.0.0.1:8084');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	114 is($r{'data'}, '1', 'upload - one byte');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	115
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	116 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	117
634 40675bfad7d3 Tests: fixed stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: 632 diff changeset	118 # Five chunks are split with four 1s delays + 2s error:
637 443848f28a73 Tests: whitespace fix. Maxim Dounin <mdounin@mdounin.ru> parents: 634 diff changeset	119 # the first two chunks are halfs of test string
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	120 # and the third one is some extra data from backend.
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	121
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	122 %r = stream_get($str, peer => '127.0.0.1:8085');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	123 my $diff = time() - $r{'time'};
634 40675bfad7d3 Tests: fixed stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: 632 diff changeset	124 cmp_ok(abs($diff - 4), '<=', 2, 'download - time');
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	125 is($r{'data'}, $str, 'download - data');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	126
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	127 my $time = time();
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	128 %r = stream_get($str . 'close', peer => '127.0.0.1:8086');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	129 $diff = time() - $time;
634 40675bfad7d3 Tests: fixed stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: 632 diff changeset	130 cmp_ok(abs($diff - 4), '<=', 2, 'upload - time');
632 c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	131 is($r{'data'}, $str . 'close', 'upload - data');
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	132
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	133 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	134
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	135 sub stream_get {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	136 my ($data, %extra) = @_;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	137
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	138 my $s = stream_connect($extra{'peer'});
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	139 stream_write($s, $data);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	140
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	141 $data = '';
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	142 while (my $buf = stream_read($s)) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	143 $data .= $buf;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	144 last if $extra{'readonce'};
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	145 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	146 $data =~ /([\S]*)\s?(\d+)?/;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	147
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	148 return ('data' => $1, 'time' => $2);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	149 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	150
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	151 sub stream_connect {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	152 my $peer = shift;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	153 my $s = IO::Socket::INET->new(
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	154 Proto => 'tcp',
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	155 PeerAddr => $peer
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	156 )
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	157 or die "Can't connect to nginx: $!\n";
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	158
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	159 return $s;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	160 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	161
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	162 sub stream_write {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	163 my ($s, $message) = @_;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	164
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	165 local $SIG{PIPE} = 'IGNORE';
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	166
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	167 $s->blocking(0);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	168 while (IO::Select->new($s)->can_write(1.5)) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	169 my $n = $s->syswrite($message);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	170 last unless $n;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	171 $message = substr($message, $n);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	172 last unless length $message;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	173 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	174
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	175 if (length $message) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	176 $s->close();
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	177 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	178 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	179
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	180 sub stream_read {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	181 my ($s) = @_;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	182 my ($buf);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	183
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	184 $s->blocking(0);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	185 if (IO::Select->new($s)->can_read(3)) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	186 $s->sysread($buf, 1024);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	187 };
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	188
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	189 log_in($buf);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	190 return $buf;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	191 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	192
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	193 ###############################################################################
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	194
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	195 sub stream_daemon {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	196 my $port = shift;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	197 my $server = IO::Socket::INET->new(
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	198 Proto => 'tcp',
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	199 LocalAddr => '127.0.0.1',
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	200 LocalPort => $port,
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	201 Listen => 5,
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	202 Reuse => 1
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	203 )
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	204 or die "Can't create listening socket: $!\n";
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	205
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	206 my $sel = IO::Select->new($server);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	207
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	208 local $SIG{PIPE} = 'IGNORE';
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	209
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	210 while (my @ready = $sel->can_read) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	211 foreach my $fh (@ready) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	212 if ($server == $fh) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	213 my $new = $fh->accept;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	214 $new->autoflush(1);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	215 $sel->add($new);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	216
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	217 } elsif (stream_handle_client($fh)) {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	218 $sel->remove($fh);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	219 $fh->close;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	220 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	221 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	222 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	223 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	224
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	225 sub stream_handle_client {
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	226 my ($client) = @_;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	227
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	228 log2c("(new connection $client)");
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	229
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	230 $client->sysread(my $buffer, 65536) or return 1;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	231
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	232 log2i("$client $buffer");
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	233
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	234 $buffer .= " " . time() if $client->sockport() eq 8080;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	235
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	236 log2o("$client $buffer");
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	237
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	238 $client->syswrite($buffer);
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	239
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	240 return $client->sockport() eq 8080 ? 1 : $buffer =~ /close/;
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	241 }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	242
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	243 sub log2i { Test::Nginx::log_core('\|\| <<', @_); }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	244 sub log2o { Test::Nginx::log_core('\|\| >>', @_); }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	245 sub log2c { Test::Nginx::log_core('\|\|', @_); }
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	246
c8d6f816e094 Tests: stream limit rate tests. Andrey Zelenkov <zelenkov@nginx.com> parents: diff changeset	247 ###############################################################################

Mercurial > hg > nginx-tests

annotate stream_limit_rate.t @ 637:443848f28a73