Mercurial > hg > nginx-tests
view stream_upstream.t @ 605:a77f19282f63
Tests: switched to using headers to test variables in userid.t.
Tests might behave incorrectly due to the delay in writing to log.
Made tests stricter for different endianness. While here, improved style.
author | Andrey Zelenkov <zelenkov@nginx.com> |
---|---|
date | Thu, 11 Jun 2015 20:56:19 +0300 |
parents | ff49e1c00b35 |
children | 824754da4afc |
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; ############################################################################### 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_get($data, $peer) =~ /(\d+)/) { $ports{$1} = 0 unless defined $ports{$1}; $ports{$1}++; } } return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports; } sub stream_get { my ($data, $peer) = @_; my $s = stream_connect($peer); stream_write($s, $data); $data = ''; while (my $buf = stream_read($s)) { $data .= $buf; } return $data; } sub stream_connect { my $peer = shift; my $s = IO::Socket::INET->new( Proto => 'tcp', PeerAddr => $peer ) or die "Can't connect to nginx: $!\n"; return $s; } sub stream_write { my ($s, $message) = @_; local $SIG{PIPE} = 'IGNORE'; $s->blocking(0); while (IO::Select->new($s)->can_write(1.5)) { my $n = $s->syswrite($message); last unless $n; $message = substr($message, $n); last unless length $message; } if (length $message) { $s->close(); } } sub stream_read { my ($s) = @_; my ($buf); $s->blocking(0); if (IO::Select->new($s)->can_read(3)) { $s->sysread($buf, 1024); }; log_in($buf); return $buf; } ############################################################################### 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('||', @_); } ###############################################################################