-
Notifications
You must be signed in to change notification settings - Fork 0
/
50reverse-proxy-timeouts.t
102 lines (92 loc) · 3.23 KB
/
50reverse-proxy-timeouts.t
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
use strict;
use warnings;
use Net::EmptyPort qw(check_port empty_port);
use Test::More;
use t::Util;
use File::Temp qw(tempfile);
use Socket;
plan skip_all => 'curl not found'
unless prog_exists('curl');
my $silent_port = empty_port({ host => '0.0.0.0' });
socket(my $silent_server, PF_INET, SOCK_STREAM, getprotobyname("tcp")) || die "socket: $!";
setsockopt($silent_server, SOL_SOCKET, SO_REUSEADDR, pack("l", 1)) || die "setsockopt: $!";
bind($silent_server, sockaddr_in($silent_port, INADDR_ANY)) || die "bind: $!";
listen($silent_server, SOMAXCONN) || die "listen: $!";
my $upstream_port = empty_port({ host => '0.0.0.0' });
my $upstream = spawn_server(
argv => [
qw(
plackup -s Standalone --ssl=1 --ssl-key-file=examples/h2o/server.key --ssl-cert-file=examples/h2o/server.crt --port
),
$upstream_port, ASSETS_DIR . "/upstream.psgi"
],
is_ready => sub {
check_port($upstream_port);
},
);
sub doit {
my $uport = shift;
my $io = shift;
my $connect = shift;
my $first_byte = shift;
my $failure = shift;
my $success = shift;
my $fail_with_chunks = shift;
diag("io: $io, connect: $connect, first_byte: $first_byte");
my $server = spawn_h2o(<< "EOT");
hosts:
default:
paths:
/:
proxy.reverse.url: https://127.0.0.1:$uport
proxy.ssl.verify-peer: OFF
@{[ $io == 0 ? "" : "proxy.timeout.io: $io" ]}
@{[ $connect == 0 ? "" : "proxy.timeout.connect: $connect" ]}
@{[ $first_byte == 0 ? "" : "proxy.timeout.first_byte: $first_byte" ]}
EOT
for my $t ($failure, $success) {
if ($success eq "") {
next;
}
my ($conffh, $conffn) = tempfile(UNLINK => 1);
print $conffh $t;
close($conffh);
my $port = $server->{port};
my $res = `curl -X POST -d \@$conffn -svo /dev/null http://127.0.0.1:$port/custom-perl 2>&1 > /dev/null`;
if ($t eq $failure) {
if ($fail_with_chunks) {
like $res, qr{Illegal or missing hexadecimal sequence in chunked-encoding}, "Truncated transfer";
} else {
like $res, qr{HTTP/1\.1 502 }, "502 response on failure";
}
} else {
like $res, qr{HTTP/1\.1 200 }, "200 response on success";
}
}
};
doit($upstream_port, 2000, 0, 0,
"sleep 3; return [ 200, [], ['ok']];",
"sleep 1; return [ 200, [], ['ok']];", 0);
doit($upstream_port, 2000, 30000, 0,
"sleep 3; return [ 200, [], ['ok']];",
"sleep 1; return [ 200, [], ['ok']];", 0);
doit($silent_port, 0, 2000, 0,
"sleep 3; return [ 200, [], ['ok']];", "", 0);
doit($upstream_port, 2000, 30000, 30000,
'return sub {
my $responder = shift;
my $writer = $responder->([ 200, [ "content-type" => "text/plain" ] ]);
$writer->write("lorem ipsum dolor sit amet");
sleep 3;
$writer->write("lorem ipsum dolor sit amet");
$writer->close;
};',
'return sub {
my $responder = shift;
my $writer = $responder->([ 200, [ "content-type" => "text/plain" ] ]);
$writer->write("lorem ipsum dolor sit amet");
sleep 1;
$writer->write("lorem ipsum dolor sit amet");
$writer->close;
};', 1);
done_testing();