Mercurial > hg > nginx-tests
view upstream.t @ 549:e4ff43e00d84
Tests: refactored least_conn test.
First, start long requests to busy both backends, and ensure that subsequent
consecutive requests are balanced to the only backend with zero connections.
This change aims to get rid of false negatives due timing issues.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Fri, 17 Apr 2015 17:54:06 +0300 |
parents | e7dc8f4d0a4b |
children | e9064d691790 |
line wrap: on
line source
#!/usr/bin/perl # (C) Maxim Dounin # Tests for upstream module and balancers. ############################################################################### use warnings; use strict; use Test::More; BEGIN { use FindBin; chdir($FindBin::Bin); } use lib 'lib'; use Test::Nginx; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new()->has(qw/http proxy/)->plan(3); $t->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } http { %%TEST_GLOBALS_HTTP%% upstream u { server 127.0.0.1:8081 max_fails=3 fail_timeout=10s; server 127.0.0.1:8082 max_fails=3 fail_timeout=10s; } upstream u2 { server 127.0.0.1:8081 max_fails=3 fail_timeout=10s; server 127.0.0.1:8082 max_fails=3 fail_timeout=10s; } server { listen 127.0.0.1:8080; server_name localhost; location / { proxy_pass http://u; } location /close2 { proxy_pass http://u2; } } } EOF $t->run_daemon(\&http_daemon, 8081); $t->run_daemon(\&http_daemon, 8082); $t->run(); $t->waitforsocket('127.0.0.1:8081'); $t->waitforsocket('127.0.0.1:8082'); ############################################################################### is(many('/', 30), '8081: 15, 8082: 15', 'balanced'); # from 9 first requests to 8081, only 6 will be successful, # 3rd, 6th, and 9th requests will fail; after this the backend # will be considered down and won't be used till fail_timeout passes is(many('/close', 30), '8081: 6, 8082: 24', 'failures'); SKIP: { skip 'long test', 1 unless $ENV{TEST_NGINX_UNSAFE}; # bug: failures counter is reset if first request in a second succeeds # # delay added to make sure first 9 requests will take more than 1s; # note that the test is racy and may unexpectedly succeed is(many('/close2', 30, delay => 0.2), '8081: 6, 8082: 24', 'failures delay'); } ############################################################################### sub many { my ($uri, $count, %opts) = @_; my %ports; for (1 .. $count) { if (http_get($uri) =~ /X-Port: (\d+)/) { $ports{$1} = 0 unless defined $ports{$1}; $ports{$1}++; } select undef, undef, undef, $opts{delay} if $opts{delay}; } return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports; } ############################################################################### sub http_daemon { my ($port) = @_; my $count = 1; 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 =~ 'close' && $port == 8081 && $count++ % 3 == 0) { next; } print $client <<EOF; HTTP/1.1 200 OK Connection: close X-Port: $port OK EOF close $client; } } ###############################################################################