Mercurial > hg > nginx-tests

comparison stream_upstream_random.t @ 1350:fda743e92b31

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: upstream random tests.
author Sergey Kandaurov <pluknet@nginx.com>
date Fri, 15 Jun 2018 13:59:08 +0300
parents
children a37d8526e691
comparison
equal deleted inserted replaced
1349:99a83f7e7755 1350:fda743e92b31
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Nginx, Inc.
5
6 # Tests for upstream random balancer module.
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 qw/ :DEFAULT http_end /;
19
20 ###############################################################################
21
22 select STDERR; $| = 1;
23 select STDOUT; $| = 1;
24
25 my $t = Test::Nginx->new()
26 ->has(qw/stream stream_upstream_zone stream_upstream_random/)
27 ->write_file_expand('nginx.conf', <<'EOF');
28
29 %%TEST_GLOBALS%%
30
31 daemon off;
32 worker_processes 2;
33
34 events {
35 }
36
37 stream {
38 upstream u {
39 zone z 1m;
40 random;
41 server 127.0.0.1:8081;
42 server 127.0.0.1:8082;
43 }
44
45 upstream lc {
46 zone lc 1m;
47 random two;
48 server 127.0.0.1:8081;
49 server 127.0.0.1:8082;
50 }
51
52 upstream w {
53 zone w 1m;
54 random two least_conn;
55 server 127.0.0.1:8081;
56 server 127.0.0.1:8082 weight=2;
57 }
58
59 upstream mc {
60 zone mc 1m;
61 random;
62 server 127.0.0.1:8081 max_conns=2;
63 server 127.0.0.1:8082 max_conns=1;
64 }
65
66 upstream mc2 {
67 zone mc 1m;
68 random two;
69 server 127.0.0.1:8081 max_conns=2;
70 server 127.0.0.1:8082 max_conns=1;
71 }
72
73 upstream one {
74 random;
75 server 127.0.0.1:8081;
76 }
77
78 upstream two {
79 random two;
80 server 127.0.0.1:8081;
81 }
82
83 upstream zone {
84 zone z 1m;
85 random;
86 server 127.0.0.1:8081;
87 }
88
89 upstream ztwo {
90 zone z 1m;
91 random two;
92 server 127.0.0.1:8081;
93 }
94
95 upstream fail {
96 zone fail 1m;
97 random;
98 server 127.0.0.1:8096;
99 server 127.0.0.1:8083 down;
100 server 127.0.0.1:8082;
101 }
102
103 upstream fail2 {
104 zone fail2 1m;
105 random two;
106 server 127.0.0.1:8096;
107 server 127.0.0.1:8083 down;
108 server 127.0.0.1:8082;
109 }
110
111 server {
112 listen 127.0.0.1:8080;
113 proxy_pass u;
114 }
115
116 server {
117 listen 127.0.0.1:8083;
118 proxy_pass lc;
119 }
120
121 server {
122 listen 127.0.0.1:8084;
123 proxy_pass w;
124 }
125
126 server {
127 listen 127.0.0.1:8085;
128 proxy_pass mc;
129 }
130
131 server {
132 listen 127.0.0.1:8086;
133 proxy_pass mc2;
134 }
135
136 server {
137 listen 127.0.0.1:8087;
138 proxy_pass one;
139 }
140
141 server {
142 listen 127.0.0.1:8088;
143 proxy_pass two;
144 }
145
146 server {
147 listen 127.0.0.1:8089;
148 proxy_pass zone;
149 }
150
151 server {
152 listen 127.0.0.1:8090;
153 proxy_pass ztwo;
154 }
155
156 proxy_connect_timeout 1s;
157
158 server {
159 listen 127.0.0.1:8091;
160 proxy_pass fail;
161 }
162
163 server {
164 listen 127.0.0.1:8092;
165 proxy_pass fail2;
166 }
167 }
168
169 EOF
170
171 $t->run_daemon(\&http_daemon, port(8081));
172 $t->run_daemon(\&http_daemon, port(8082));
173 $t->try_run('no upstream random')->plan(12);
174
175 $t->waitforsocket('127.0.0.1:' . port(8081));
176 $t->waitforsocket('127.0.0.1:' . port(8082));
177
178 ###############################################################################
179
180 my @ports = my ($port1, $port2) = (port(8081), port(8082));
181
182 like(get(8080, '/'), qr/X-Port: ($port1|$port2)/, 'random');
183 like(get(8083, '/'), qr/X-Port: ($port1|$port2)/, 'random two');
184
185 my $s = get(8083, '/w', start => 1, sleep => 0.2);
186 my $r = get(8083, '/');
187 my ($p) = http_end($s) =~ /X-Port: (\d+)/;
188 like($r, qr/X-Port: (?!$p)/, 'random wait');
189
190 SKIP: {
191 skip 'long test', 3 unless $ENV{TEST_NGINX_UNSAFE};
192
193 is(parallel(8084, '/w', 3), "$port1: 1, $port2: 2", 'random weight');
194
195 is(parallel(8085, '/w', 4), "$port1: 2, $port2: 1", 'max_conns');
196 is(parallel(8086, '/w', 4), "$port1: 2, $port2: 1", 'max_conns two');
197
198 }
199
200 # single variants
201
202 like(get(8087, '/'), qr/X-Port: $port1/, 'single one');
203 like(get(8088, '/'), qr/X-Port: $port1/, 'single two');
204 like(get(8089, '/'), qr/X-Port: $port1/, 'zone one');
205 like(get(8090, '/'), qr/X-Port: $port1/, 'zone two');
206
207 like(many(8091, '/', 10), qr/$port2: 10/, 'failures');
208 like(many(8092, '/', 10), qr/$port2: 10/, 'failures two');
209
210 ###############################################################################
211
212 sub get {
213 my ($port, $uri, %opts) = @_;
214 my $s = IO::Socket::INET->new(
215 Proto => 'tcp',
216 PeerAddr => '127.0.0.1',
217 PeerPort => port($port),
218 )
219 or die "Can't connect to nginx: $!\n";
220
221 http_get($uri, socket => $s, %opts);
222 }
223
224 sub many {
225 my ($port, $uri, $count, %opts) = @_;
226 my %ports;
227
228 for (1 .. $count) {
229 if (get($port, $uri) =~ /X-Port: (\d+)/) {
230 $ports{$1} = 0 unless defined $ports{$1};
231 $ports{$1}++;
232 }
233
234 select undef, undef, undef, $opts{delay} if $opts{delay};
235 }
236
237 my @keys = map { my $p = $_; grep { $p == $_ } keys %ports } @ports;
238 return join ', ', map { $_ . ": " . $ports{$_} } @keys;
239 }
240
241 sub parallel {
242 my ($port, $uri, $n) = @_;
243 my %ports;
244
245 my @s = map { get($port, $uri, start => 1, sleep => 0.1) } (1 .. $n);
246
247 for (@s) {
248 if (http_end($_) =~ /X-Port: (\d+)/) {
249 $ports{$1} = 0 unless defined $ports{$1};
250 $ports{$1}++;
251 }
252 }
253
254 my @keys = map { my $p = $_; grep { $p == $_ } keys %ports } @ports;
255 return join ', ', map { $_ . ": " . $ports{$_} } @keys;
256 }
257
258 ###############################################################################
259
260 sub http_daemon {
261 my ($port) = @_;
262
263 my $server = IO::Socket::INET->new(
264 Proto => 'tcp',
265 LocalHost => '127.0.0.1',
266 LocalPort => $port,
267 Listen => 5,
268 Reuse => 1
269 )
270 or die "Can't create listening socket: $!\n";
271
272 local $SIG{PIPE} = 'IGNORE';
273
274 while (my $client = $server->accept()) {
275 $client->autoflush(1);
276
277 my $headers = '';
278 my $uri = '';
279
280 while (<$client>) {
281 $headers .= $_;
282 last if (/^\x0d?\x0a?$/);
283 }
284
285 $uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i;
286
287 if ($uri eq '/w') {
288 Test::Nginx::log_core('||', "$port: sleep(2.5)");
289 select undef, undef, undef, 2.5;
290 }
291
292 if ($uri eq '/close' && $port == port(8081)) {
293 next;
294 }
295
296 Test::Nginx::log_core('||', "$port: response, 200");
297 print $client <<EOF;
298 HTTP/1.1 200 OK
299 Connection: close
300 X-Port: $port
301
302 OK
303 EOF
304
305 close $client;
306 }
307 }
308
309 ###############################################################################