Mercurial > hg > nginx-tests
comparison 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 |
comparison
equal
deleted
inserted
replaced
867:89898b20f258 | 868:d2cb9ed7412e |
---|---|
1 #!/usr/bin/perl | |
2 | |
3 # (C) Andrey Zelenkov | |
4 # (C) Nginx, Inc. | |
5 | |
6 # Tests for stream limit_conn module with datagrams. | |
7 | |
8 ############################################################################### | |
9 | |
10 use warnings; | |
11 use strict; | |
12 | |
13 use Test::More; | |
14 | |
15 BEGIN { use FindBin; chdir($FindBin::Bin); } | |
16 | |
17 use lib 'lib'; | |
18 use Test::Nginx; | |
19 use Test::Nginx::Stream qw/ dgram /; | |
20 | |
21 ############################################################################### | |
22 | |
23 select STDERR; $| = 1; | |
24 select STDOUT; $| = 1; | |
25 | |
26 my $t = Test::Nginx->new()->has(qw/stream stream_limit_conn shmem/) | |
27 ->write_file_expand('nginx.conf', <<'EOF'); | |
28 | |
29 %%TEST_GLOBALS%% | |
30 | |
31 daemon off; | |
32 | |
33 events { | |
34 } | |
35 | |
36 stream { | |
37 limit_conn_zone $binary_remote_addr zone=zone:1m; | |
38 limit_conn_zone $binary_remote_addr zone=zone2:1m; | |
39 | |
40 proxy_responses 1; | |
41 proxy_timeout 1s; | |
42 | |
43 server { | |
44 listen 127.0.0.1:8081 udp; | |
45 proxy_pass 127.0.0.1:8080; | |
46 | |
47 limit_conn zone 1; | |
48 proxy_responses 2; | |
49 } | |
50 | |
51 server { | |
52 listen 127.0.0.1:8082 udp; | |
53 proxy_pass 127.0.0.1:8080; | |
54 limit_conn zone2 1; | |
55 } | |
56 | |
57 server { | |
58 listen 127.0.0.1:8083 udp; | |
59 proxy_pass 127.0.0.1:8080; | |
60 limit_conn zone 5; | |
61 } | |
62 | |
63 server { | |
64 listen 127.0.0.1:8084 udp; | |
65 proxy_pass 127.0.0.1:8081; | |
66 limit_conn zone2 1; | |
67 } | |
68 | |
69 server { | |
70 listen 127.0.0.1:8085 udp; | |
71 proxy_pass 127.0.0.1:8081; | |
72 limit_conn zone 1; | |
73 } | |
74 } | |
75 | |
76 EOF | |
77 | |
78 $t->try_run('no stream udp')->plan(8); | |
79 $t->run_daemon(\&udp_daemon, $t); | |
80 $t->waitforfile($t->testdir . '/8080'); | |
81 | |
82 ############################################################################### | |
83 | |
84 my $str = '.'; | |
85 | |
86 # same and other zones | |
87 | |
88 my $s = dgram('127.0.0.1:8081'); | |
89 | |
90 is($s->io('1'), '1', 'passed'); | |
91 | |
92 is(dgram('127.0.0.1:8081')->io('1'), '', 'rejected same zone'); | |
93 is(dgram('127.0.0.1:8082')->io('1'), '1', 'passed different zone'); | |
94 is(dgram('127.0.0.1:8083')->io('1'), '1', 'passed same zone unlimited'); | |
95 | |
96 is($s->io('2', read => 2), '12', 'long connection'); | |
97 | |
98 is(dgram('127.0.0.1:8081')->io('2', read => 2), '12', 'passed 2'); | |
99 | |
100 # zones proxy chain | |
101 | |
102 is(dgram('127.0.0.1:8084')->io('1'), '1', 'passed proxy'); | |
103 is(dgram('127.0.0.1:8085')->io('1'), '', 'rejected proxy'); | |
104 | |
105 ############################################################################### | |
106 | |
107 sub udp_daemon { | |
108 my $t = shift; | |
109 | |
110 my $server = IO::Socket::INET->new( | |
111 Proto => 'udp', | |
112 LocalAddr => '127.0.0.1:8080', | |
113 Reuse => 1, | |
114 ) | |
115 or die "Can't create listening socket: $!\n"; | |
116 | |
117 # signal we are ready | |
118 | |
119 open my $fh, '>', $t->testdir() . '/8080'; | |
120 close $fh; | |
121 | |
122 while (1) { | |
123 $server->recv(my $buffer, 65536); | |
124 $server->send($_) for (1 .. $buffer); | |
125 } | |
126 } | |
127 | |
128 ############################################################################### |