Mercurial > hg > nginx-tests
view stream_upstream_random.t @ 1752:ba6e24e38f03
Tests: improved stop_daemons() to send signal again.
As was observed, it's possible that a signal to complete a uwsgi daemon
can be ignored while it is starting up, which results in tests hang due
to eternal waiting on child processes termination. Notably, it is seen
when running tests with a high number of prove jobs on a low-profile VM
against nginx with broken modules and/or configuration. To reproduce:
$ TEST_NGINX_GLOBALS=ERROR prove -j16 uwsgi*.t
Inspecting uwsgi under ktrace on FreeBSD confirms that a SIGTERM signal
is ignored at the very beginning of uwsgi startup. It is then replaced
with a default action after listen(), thus waiting until uwsgi is ready
to accept new TCP connections doesn't completely solve the hang window.
The fix is to retry sending a signal some time after waitpid(WNOHANG)
continuously demonstrated no progress with reaping a signaled process.
It is modelled after f13ead27f89c that improved stop() for nginx.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Wed, 29 Dec 2021 22:29:23 +0300 |
parents | f3ba4c74de31 |
children |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for upstream random balancer module. ############################################################################### use warnings; use strict; use Test::More; BEGIN { use FindBin; chdir($FindBin::Bin); } use lib 'lib'; use Test::Nginx qw/ :DEFAULT http_end /; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new() ->has(qw/stream stream_upstream_zone stream_upstream_random/)->plan(12) ->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; worker_processes 2; events { } stream { %%TEST_GLOBALS_STREAM%% upstream u { zone z 1m; random; server 127.0.0.1:8081; server 127.0.0.1:8082; } upstream lc { zone lc 1m; random two; server 127.0.0.1:8081; server 127.0.0.1:8082; } upstream w { zone w 1m; random two least_conn; server 127.0.0.1:8081; server 127.0.0.1:8082 weight=2; } upstream mc { zone mc 1m; random; server 127.0.0.1:8081 max_conns=2; server 127.0.0.1:8082 max_conns=1; } upstream mc2 { zone mc 1m; random two; server 127.0.0.1:8081 max_conns=2; server 127.0.0.1:8082 max_conns=1; } upstream one { random; server 127.0.0.1:8081; } upstream two { random two; server 127.0.0.1:8081; } upstream zone { zone z 1m; random; server 127.0.0.1:8081; } upstream ztwo { zone z 1m; random two; server 127.0.0.1:8081; } upstream fail { zone fail 1m; random; server 127.0.0.1:8096; server 127.0.0.1:8083 down; server 127.0.0.1:8082; } upstream fail2 { zone fail2 1m; random two; server 127.0.0.1:8096; server 127.0.0.1:8083 down; server 127.0.0.1:8082; } proxy_connect_timeout 2; server { listen 127.0.0.1:8080; proxy_pass u; } server { listen 127.0.0.1:8083; proxy_pass lc; } server { listen 127.0.0.1:8084; proxy_pass w; } server { listen 127.0.0.1:8085; proxy_pass mc; } server { listen 127.0.0.1:8086; proxy_pass mc2; } server { listen 127.0.0.1:8087; proxy_pass one; } server { listen 127.0.0.1:8088; proxy_pass two; } server { listen 127.0.0.1:8089; proxy_pass zone; } server { listen 127.0.0.1:8090; proxy_pass ztwo; } server { listen 127.0.0.1:8091; proxy_pass fail; } server { listen 127.0.0.1:8092; proxy_pass fail2; } } EOF $t->run_daemon(\&http_daemon, port(8081)); $t->run_daemon(\&http_daemon, port(8082)); $t->run(); $t->waitforsocket('127.0.0.1:' . port(8081)); $t->waitforsocket('127.0.0.1:' . port(8082)); ############################################################################### my @ports = my ($port1, $port2) = (port(8081), port(8082)); like(get(8080, '/'), qr/X-Port: ($port1|$port2)/, 'random'); like(get(8083, '/'), qr/X-Port: ($port1|$port2)/, 'random two'); my $s = get(8083, '/w', start => 1, sleep => 0.5); my $r = get(8083, '/'); my ($p) = http_end($s) =~ /X-Port: (\d+)/; like($r, qr/X-Port: (?!$p)/, 'random wait'); SKIP: { skip 'long test', 3 unless $ENV{TEST_NGINX_UNSAFE}; is(parallel(8084, '/w', 3), "$port1: 1, $port2: 2", 'random weight'); is(parallel(8085, '/w', 4), "$port1: 2, $port2: 1", 'max_conns'); is(parallel(8086, '/w', 4), "$port1: 2, $port2: 1", 'max_conns two'); } # single variants like(get(8087, '/'), qr/X-Port: $port1/, 'single one'); like(get(8088, '/'), qr/X-Port: $port1/, 'single two'); like(get(8089, '/'), qr/X-Port: $port1/, 'zone one'); like(get(8090, '/'), qr/X-Port: $port1/, 'zone two'); like(many(8091, '/', 10), qr/$port2: 10/, 'failures'); like(many(8092, '/', 10), qr/$port2: 10/, 'failures two'); ############################################################################### sub get { my ($port, $uri, %opts) = @_; my $s = IO::Socket::INET->new( Proto => 'tcp', PeerAddr => '127.0.0.1', PeerPort => port($port), ) or die "Can't connect to nginx: $!\n"; http_get($uri, socket => $s, %opts); } sub many { my ($port, $uri, $count, %opts) = @_; my %ports; for (1 .. $count) { if (get($port, $uri) =~ /X-Port: (\d+)/) { $ports{$1} = 0 unless defined $ports{$1}; $ports{$1}++; } select undef, undef, undef, $opts{delay} if $opts{delay}; } my @keys = map { my $p = $_; grep { $p == $_ } keys %ports } @ports; return join ', ', map { $_ . ": " . $ports{$_} } @keys; } sub parallel { my ($port, $uri, $n) = @_; my %ports; my @s = map { get($port, $uri, start => 1, sleep => 0.1) } (1 .. $n); for (@s) { my $r = http_end($_); if ($r && $r =~ /X-Port: (\d+)/) { $ports{$1} = 0 unless defined $ports{$1}; $ports{$1}++; } } my @keys = map { my $p = $_; grep { $p == $_ } keys %ports } @ports; return join ', ', map { $_ . ": " . $ports{$_} } @keys; } ############################################################################### sub http_daemon { my ($port) = @_; my $server = IO::Socket::INET->new( Proto => 'tcp', LocalHost => '127.0.0.1', LocalPort => $port, Listen => 5, Reuse => 1 ) or die "Can't create listening socket: $!\n"; local $SIG{PIPE} = 'IGNORE'; while (my $client = $server->accept()) { $client->autoflush(1); my $headers = ''; my $uri = ''; while (<$client>) { $headers .= $_; last if (/^\x0d?\x0a?$/); } $uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i; if ($uri eq '/w') { Test::Nginx::log_core('||', "$port: sleep(2.5)"); select undef, undef, undef, 2.5; } if ($uri eq '/close' && $port == port(8081)) { next; } Test::Nginx::log_core('||', "$port: response, 200"); print $client <<EOF; HTTP/1.1 200 OK Connection: close X-Port: $port OK EOF close $client; } } ###############################################################################