Mercurial > hg > nginx-tests
annotate lib/Test/Nginx/Stream.pm @ 1024:91e64c1ceec9
Tests: stream access_log tests.
| author | Sergey Kandaurov <pluknet@nginx.com> |
|---|---|
| date | Mon, 05 Sep 2016 22:57:40 +0300 |
| parents | 1edb092149e2 |
| children | 3fc6817cd84a |
| rev | line source |
|---|---|
| 816 | 1 package Test::Nginx::Stream; |
| 2 | |
| 3 # (C) Andrey Zelenkov | |
| 4 # (C) Nginx, Inc. | |
| 5 | |
| 6 # Module for nginx stream tests. | |
| 7 | |
| 8 ############################################################################### | |
| 9 | |
| 10 use warnings; | |
| 11 use strict; | |
| 12 | |
| 13 use base qw/ Exporter /; | |
|
868
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
14 our @EXPORT_OK = qw/ stream dgram /; |
| 816 | 15 |
| 16 use Test::More qw//; | |
| 17 use IO::Select; | |
| 18 use IO::Socket; | |
| 19 | |
| 20 use Test::Nginx; | |
| 21 | |
| 22 sub stream { | |
| 23 return Test::Nginx::Stream->new(@_); | |
| 24 } | |
| 25 | |
|
868
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
26 sub dgram { |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
27 unshift(@_, "PeerAddr") if @_ == 1; |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
28 |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
29 return Test::Nginx::Stream->new( |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
30 Proto => "udp", |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
31 @_ |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
32 ); |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
33 } |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
34 |
| 816 | 35 sub new { |
| 36 my $self = {}; | |
| 37 bless $self, shift @_; | |
| 38 | |
| 39 unshift(@_, "PeerAddr") if @_ == 1; | |
| 40 | |
| 41 $self->{_socket} = IO::Socket::INET->new( | |
| 42 Proto => "tcp", | |
|
951
9361c7eddfc1
Tests: parallel tests support.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
935
diff
changeset
|
43 PeerAddr => '127.0.0.1', |
|
969
1edb092149e2
Tests: simplified parallel tests.
Maxim Dounin <mdounin@mdounin.ru>
parents:
951
diff
changeset
|
44 PeerPort => port(8080), |
| 816 | 45 @_ |
| 46 ) | |
| 47 or die "Can't connect to nginx: $!\n"; | |
| 48 | |
| 49 if ({@_}->{'SSL'}) { | |
| 50 require IO::Socket::SSL; | |
| 51 IO::Socket::SSL->start_SSL($self->{_socket}, @_) | |
| 52 or die $IO::Socket::SSL::SSL_ERROR . "\n"; | |
| 53 } | |
| 54 | |
| 55 $self->{_socket}->autoflush(1); | |
| 56 | |
| 57 return $self; | |
| 58 } | |
| 59 | |
| 60 sub write { | |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
61 my ($self, $message, %extra) = @_; |
| 816 | 62 my $s = $self->{_socket}; |
| 63 | |
| 64 local $SIG{PIPE} = 'IGNORE'; | |
| 65 | |
| 66 $s->blocking(0); | |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
67 while (IO::Select->new($s)->can_write($extra{write_timeout} || 1.5)) { |
| 816 | 68 my $n = $s->syswrite($message); |
| 69 log_out(substr($message, 0, $n)); | |
| 70 last unless $n; | |
| 71 | |
| 72 $message = substr($message, $n); | |
| 73 last unless length $message; | |
| 74 } | |
| 75 | |
| 76 if (length $message) { | |
| 77 $s->close(); | |
| 78 } | |
| 79 } | |
| 80 | |
| 81 sub read { | |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
82 my ($self, %extra) = @_; |
| 816 | 83 my ($s, $buf); |
| 84 | |
| 85 $s = $self->{_socket}; | |
| 86 | |
| 87 $s->blocking(0); | |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
88 if (IO::Select->new($s)->can_read($extra{read_timeout} || 5)) { |
| 816 | 89 $s->sysread($buf, 1024); |
| 90 }; | |
| 91 | |
| 92 log_in($buf); | |
| 93 return $buf; | |
| 94 } | |
| 95 | |
| 96 sub io { | |
| 97 my $self = shift; | |
| 98 | |
| 99 my ($data, %extra) = @_; | |
| 100 my $length = $extra{length}; | |
|
868
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
101 my $read = $extra{read}; |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
102 |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
103 $read = 1 if !defined $read |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
104 && $self->{_socket}->socktype() == &SOCK_DGRAM; |
| 816 | 105 |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
106 $self->write($data, %extra); |
| 816 | 107 |
| 108 $data = ''; | |
| 109 while (1) { | |
|
868
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
110 last if defined $read && --$read < 0; |
|
d2cb9ed7412e
Tests: stream udp tests.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
816
diff
changeset
|
111 |
|
935
25a4b2fdd3fb
Tests: I/O timeout options introduced in Stream.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
875
diff
changeset
|
112 my $buf = $self->read(%extra); |
|
875
c380b4b7e2e4
Tests: compatibility with perl < 5.12 in Stream.pm.
Andrey Zelenkov <zelenkov@nginx.com>
parents:
868
diff
changeset
|
113 last unless defined $buf and length($buf); |
| 816 | 114 |
| 115 $data .= $buf; | |
| 116 last if defined $length && length($data) >= $length; | |
| 117 } | |
| 118 | |
| 119 return $data; | |
| 120 } | |
| 121 | |
|
1024
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
122 sub sockaddr { |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
123 my $self = shift; |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
124 return $self->{_socket}->sockaddr(); |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
125 } |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
126 |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
127 sub sockhost { |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
128 my $self = shift; |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
129 return $self->{_socket}->sockhost(); |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
130 } |
|
91e64c1ceec9
Tests: stream access_log tests.
Sergey Kandaurov <pluknet@nginx.com>
parents:
969
diff
changeset
|
131 |
| 816 | 132 sub sockport { |
| 133 my $self = shift; | |
| 134 return $self->{_socket}->sockport(); | |
| 135 } | |
| 136 | |
| 137 ############################################################################### | |
| 138 | |
| 139 1; | |
| 140 | |
| 141 ############################################################################### |