| #!/usr/bin/perl |
| |
| # (C) Andrey Zelenkov |
| # (C) Nginx, Inc. |
| |
| # Tests for http keepalive directives. |
| |
| ############################################################################### |
| |
| 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/http/)->plan(14) |
| ->write_file_expand('nginx.conf', <<'EOF'); |
| |
| %%TEST_GLOBALS%% |
| |
| daemon off; |
| |
| events { |
| } |
| |
| http { |
| %%TEST_GLOBALS_HTTP%% |
| |
| log_format test $sent_http_connection; |
| access_log %%TESTDIR%%/test.log test if=$arg_l; |
| |
| server { |
| listen 127.0.0.1:8080; |
| server_name localhost; |
| |
| keepalive_requests 2; |
| keepalive_timeout 1 9; |
| |
| location / { } |
| location /r { |
| keepalive_requests 4; |
| } |
| |
| location /safari { |
| keepalive_disable safari; |
| } |
| |
| location /none { |
| keepalive_disable none; |
| } |
| |
| location /zero { |
| keepalive_timeout 0; |
| } |
| } |
| } |
| |
| EOF |
| |
| $t->write_file('index.html', ''); |
| $t->write_file('r', ''); |
| $t->write_file('safari', ''); |
| $t->write_file('none', ''); |
| $t->write_file('zero', ''); |
| $t->run(); |
| |
| ############################################################################### |
| |
| # keepalive_requests |
| |
| like(http_keepalive('/'), qr/Connection: keep-alive/, 'keepalive request'); |
| is(count_keepalive(http_keepalive('/?l=ok', req => 2)), 1, 'keepalive limit'); |
| is(count_keepalive(http_keepalive('/r', req => 3)), 3, 'keepalive merge'); |
| is(count_keepalive(http_keepalive('/r', req => 5)), 3, 'keepalive merge limit'); |
| |
| # keepalive_disable |
| |
| like(http_keepalive('/', method => 'POST', ua => "MSIE 5.0"), |
| qr/Connection: close/, 'keepalive disable msie6'); |
| like(http_keepalive('/', ua => "MSIE 5.0"), qr/Connection: keep-alive/, |
| 'keepalive disable msie6 GET'); |
| like(http_keepalive('/', method => 'POST', ua => "MSIE 7.0"), |
| qr/Connection: keep-alive/, 'keepalive disable msie6 modern'); |
| like(http_keepalive('/', ua => "Mac OS X Safari/7534.48.3"), |
| qr/Connection: keep-alive/, 'keepalive disable msie6 safari'); |
| like(http_keepalive('/safari', ua => "Mac OS X Safari/7534.48.3"), |
| qr/Connection: close/, 'keepalive disable safari'); |
| like(http_keepalive('/none', method => 'POST', ua => "MSIE 5.0"), |
| qr/Connection: keep-alive/, 'keepalive disable none'); |
| |
| # keepalive_timeout |
| |
| my $r = http_keepalive('/', req => 2, sleep => 2.1); |
| is(count_keepalive($r), 1, 'keepalive timeout request'); |
| like($r, qr/Keep-Alive: timeout=9/, 'keepalive timeout header'); |
| |
| like(http_keepalive('/zero'), qr/Connection: close/, 'keepalive timeout 0'); |
| |
| $t->stop(); |
| |
| TODO: { |
| local $TODO = 'not yet'; |
| |
| is($t->read_file('test.log'), "keep-alive\nclose\n", 'sent_http_connection'); |
| |
| } |
| |
| ############################################################################### |
| |
| sub http_keepalive { |
| my ($url, %opts) = @_; |
| my $total = ''; |
| |
| $opts{ua} = $opts{ua} || ''; |
| $opts{req} = $opts{req} || 1; |
| $opts{sleep} = $opts{sleep} || 0; |
| $opts{method} = $opts{method} || 'GET'; |
| |
| local $SIG{PIPE} = 'IGNORE'; |
| |
| my $s = http('', start => 1); |
| |
| for my $i (1 .. $opts{req}) { |
| |
| my $sleep = ($i == 1 ? $opts{sleep} : 0); |
| |
| http(<<EOF, socket => $s, start => 1, sleep => $sleep); |
| $opts{method} $url HTTP/1.1 |
| Host: localhost |
| User-Agent: $opts{ua} |
| |
| EOF |
| |
| my $data = ''; |
| |
| while (IO::Select->new($s)->can_read(3)) { |
| sysread($s, my $buffer, 4096) or last; |
| $data .= $buffer; |
| last if $data =~ /^\x0d\x0a/ms; |
| } |
| |
| log_in($data); |
| |
| $total .= $data; |
| } |
| |
| return $total; |
| } |
| |
| sub count_keepalive { |
| my ($str) = @_; |
| return $str =~ s/Connection: keep-alive//g; |
| } |
| |
| ############################################################################### |