Mercurial > hg > nginx-tests
view stream_proxy.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 | ba6cc90e3d67 |
children | 824754da4afc |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for stream proxy module. ############################################################################### 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 { server { listen 127.0.0.1:8080; proxy_pass 127.0.0.1:8081; proxy_connect_timeout 1s; } } EOF $t->run_daemon(\&stream_daemon); $t->run()->waitforsocket('127.0.0.1:8081'); ############################################################################### my $s = stream_connect(); stream_write($s, 'foo1'); is(stream_read($s), 'bar1', 'proxy connection'); stream_write($s, 'foo3'); is(stream_read($s), 'bar3', 'proxy connection again'); stream_write($s, 'close'); is(stream_read($s), 'close', 'proxy connection close'); stream_write($s, 'test'); is(stream_read($s), '', 'proxy connection closed'); ############################################################################### sub stream_connect { my $peer = shift; my $s = IO::Socket::INET->new( Proto => 'tcp', PeerAddr => $peer || '127.0.0.1:8080' ) 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 $server = IO::Socket::INET->new( Proto => 'tcp', LocalAddr => '127.0.0.1:8081', 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 =~ s/foo/bar/g; log2o("$client $buffer"); $client->syswrite($buffer); return $buffer =~ /close/; } sub log2i { Test::Nginx::log_core('|| <<', @_); } sub log2o { Test::Nginx::log_core('|| >>', @_); } sub log2c { Test::Nginx::log_core('||', @_); } ###############################################################################