Mercurial > hg > nginx-tests

comparison stream_udp_limit_conn.t @ 868:d2cb9ed7412e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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 ###############################################################################