Mercurial > hg > nginx-tests

comparison stream_proxy.t @ 555:ba6cc90e3d67

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: basic stream proxy tests.
author Sergey Kandaurov <pluknet@nginx.com>
date Wed, 22 Apr 2015 16:06:44 +0300
parents
children 824754da4afc
comparison
equal deleted inserted replaced
554:4cc65166509e 555:ba6cc90e3d67
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Nginx, Inc.
5
6 # Tests for stream proxy module.
7
8 ###############################################################################
9
10 use warnings;
11 use strict;
12
13 use Test::More;
14
15 use IO::Select;
16
17 BEGIN { use FindBin; chdir($FindBin::Bin); }
18
19 use lib 'lib';
20 use Test::Nginx;
21
22 ###############################################################################
23
24 select STDERR; $| = 1;
25 select STDOUT; $| = 1;
26
27 my $t = Test::Nginx->new()->has(qw/stream/)->plan(4)
28 ->write_file_expand('nginx.conf', <<'EOF');
29
30 %%TEST_GLOBALS%%
31
32 daemon off;
33
34 events {
35 }
36
37 stream {
38 server {
39 listen 127.0.0.1:8080;
40 proxy_pass 127.0.0.1:8081;
41 proxy_connect_timeout 1s;
42 }
43 }
44
45 EOF
46
47 $t->run_daemon(\&stream_daemon);
48 $t->run()->waitforsocket('127.0.0.1:8081');
49
50 ###############################################################################
51
52 my $s = stream_connect();
53
54 stream_write($s, 'foo1');
55 is(stream_read($s), 'bar1', 'proxy connection');
56
57 stream_write($s, 'foo3');
58 is(stream_read($s), 'bar3', 'proxy connection again');
59
60 stream_write($s, 'close');
61 is(stream_read($s), 'close', 'proxy connection close');
62
63 stream_write($s, 'test');
64 is(stream_read($s), '', 'proxy connection closed');
65
66 ###############################################################################
67
68 sub stream_connect {
69 my $peer = shift;
70 my $s = IO::Socket::INET->new(
71 Proto => 'tcp',
72 PeerAddr => $peer || '127.0.0.1:8080'
73 )
74 or die "Can't connect to nginx: $!\n";
75
76 return $s;
77 }
78
79 sub stream_write {
80 my ($s, $message) = @_;
81
82 local $SIG{PIPE} = 'IGNORE';
83
84 $s->blocking(0);
85 while (IO::Select->new($s)->can_write(1.5)) {
86 my $n = $s->syswrite($message);
87 last unless $n;
88 $message = substr($message, $n);
89 last unless length $message;
90 }
91
92 if (length $message) {
93 $s->close();
94 }
95 }
96
97 sub stream_read {
98 my ($s) = @_;
99 my ($buf);
100
101 $s->blocking(0);
102 if (IO::Select->new($s)->can_read(3)) {
103 $s->sysread($buf, 1024);
104 };
105
106 log_in($buf);
107 return $buf;
108 }
109
110 ###############################################################################
111
112 sub stream_daemon {
113 my $server = IO::Socket::INET->new(
114 Proto => 'tcp',
115 LocalAddr => '127.0.0.1:8081',
116 Listen => 5,
117 Reuse => 1
118 )
119 or die "Can't create listening socket: $!\n";
120
121 my $sel = IO::Select->new($server);
122
123 local $SIG{PIPE} = 'IGNORE';
124
125 while (my @ready = $sel->can_read) {
126 foreach my $fh (@ready) {
127 if ($server == $fh) {
128 my $new = $fh->accept;
129 $new->autoflush(1);
130 $sel->add($new);
131
132 } elsif (stream_handle_client($fh)) {
133 $sel->remove($fh);
134 $fh->close;
135 }
136 }
137 }
138 }
139
140 sub stream_handle_client {
141 my ($client) = @_;
142
143 log2c("(new connection $client)");
144
145 $client->sysread(my $buffer, 65536) or return 1;
146
147 log2i("$client $buffer");
148
149 $buffer =~ s/foo/bar/g;
150
151 log2o("$client $buffer");
152
153 $client->syswrite($buffer);
154
155 return $buffer =~ /close/;
156 }
157
158 sub log2i { Test::Nginx::log_core('|| <<', @_); }
159 sub log2o { Test::Nginx::log_core('|| >>', @_); }
160 sub log2c { Test::Nginx::log_core('||', @_); }
161
162 ###############################################################################