Mercurial > hg > nginx-tests
view stream_upstream.t @ 836:a9c4cebcfe69
Tests: adjusted http resolver test with resend.
Don't bother with socket error to force resolver resend. Nginx doesn't handle
well ECONNREFUSED send() error, which results in "500 Internal Server Error".
It may deserve a separate test, though.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Fri, 29 Jan 2016 00:40:05 +0300 |
parents | 77359b849cd5 |
children | e9064d691790 |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Stream tests for upstream module and balancers. ############################################################################### use warnings; use strict; use Test::More; use IO::Select; BEGIN { use FindBin; chdir($FindBin::Bin); } use lib 'lib'; use Test::Nginx; use Test::Nginx::Stream qw/ stream /; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new()->has(qw/stream/)->plan(4) ->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } stream { upstream u { server 127.0.0.1:8087; server 127.0.0.1:8088; } upstream u2 { server 127.0.0.1:8089 down; server 127.0.0.1:8089; server 127.0.0.1:8087; server 127.0.0.1:8088; } upstream u3 { server 127.0.0.1:8087; server 127.0.0.1:8088 weight=2; } upstream u4 { server 127.0.0.1:8089; server 127.0.0.1:8087 backup; } proxy_connect_timeout 1s; server { listen 127.0.0.1:8081; proxy_pass u; } server { listen 127.0.0.1:8082; proxy_pass u2; } server { listen 127.0.0.1:8083; proxy_pass u3; } server { listen 127.0.0.1:8084; proxy_pass u4; } } EOF $t->run_daemon(\&stream_daemon, 8087); $t->run_daemon(\&stream_daemon, 8088); $t->run(); $t->waitforsocket('127.0.0.1:8087'); $t->waitforsocket('127.0.0.1:8088'); ############################################################################### is(many('.', 30, peer => '127.0.0.1:8081'), '8087: 15, 8088: 15', 'balanced'); is(many('.', 30, peer => '127.0.0.1:8082'), '8087: 15, 8088: 15', 'failures'); is(many('.', 30, peer => '127.0.0.1:8083'), '8087: 10, 8088: 20', 'weight'); is(many('.', 30, peer => '127.0.0.1:8084'), '8087: 30', 'backup'); ############################################################################### sub many { my ($data, $count, %opts) = @_; my (%ports, $peer); $peer = $opts{peer}; for (1 .. $count) { if (stream($peer)->io($data) =~ /(\d+)/) { $ports{$1} = 0 unless defined $ports{$1}; $ports{$1}++; } } return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports; } ############################################################################### sub stream_daemon { my ($port) = @_; my $server = IO::Socket::INET->new( Proto => 'tcp', LocalAddr => '127.0.0.1', LocalPort => $port, Listen => 5, Reuse => 1 ) or die "Can't create listening socket: $!\n"; my $sel = IO::Select->new($server); local $SIG{PIPE} = 'IGNORE'; while (my @ready = $sel->can_read) { foreach my $fh (@ready) { if ($server == $fh) { my $new = $fh->accept; $new->autoflush(1); $sel->add($new); } elsif (stream_handle_client($fh)) { $sel->remove($fh); $fh->close; } } } } sub stream_handle_client { my ($client) = @_; log2c("(new connection $client)"); $client->sysread(my $buffer, 65536) or return 1; log2i("$client $buffer"); $buffer = $client->sockport(); log2o("$client $buffer"); $client->syswrite($buffer); return 1; } sub log2i { Test::Nginx::log_core('|| <<', @_); } sub log2o { Test::Nginx::log_core('|| >>', @_); } sub log2c { Test::Nginx::log_core('||', @_); } ###############################################################################