Ticket #541: proxy_no_proxy_temp2.t

#!/usr/bin/perl

# (C) Maxim Dounin

# Tests for http proxy and prematurely closed connections.  Incomplete
# responses shouldn't loose information about their incompleteness.

# In particular, incomplete responses:
#
# - shouldn't be cached
#
# - if a response is sent using chunked transfer encoding,
#   final chunk shouldn't be sent

###############################################################################

use warnings;
use strict;

use Test::More;

use Socket qw/ CRLF /;

BEGIN { use FindBin; chdir($FindBin::Bin); }

use lib 'lib';
use Test::Nginx;

###############################################################################

select STDERR; $| = 1;
select STDOUT; $| = 1;

plan(skip_all => 'win32') if $^O eq 'MSWin32';

my $t = Test::Nginx->new()->has(qw/http proxy cache/)->plan(1)
        ->write_file_expand('nginx.conf', <<'EOF');

%%TEST_GLOBALS%%

daemon off;

events {
}

http {
    %%TEST_GLOBALS_HTTP%%

    proxy_cache_path   %%TESTDIR%%/cache  levels=1:2
                       keys_zone=one:1m;

    server {
        listen       127.0.0.1:8080;
        server_name  localhost;

        location / {
            proxy_pass http://127.0.0.1:8081;
            proxy_cache one;
            proxy_buffer_size 1k;
            proxy_buffers 4 1k;
        }
    }
}

EOF

$t->run_daemon(\&http_daemon);
$t->run()->waitforsocket('127.0.0.1:8081');

###############################################################################

TODO: {
local $TODO = 'not yet' unless $t->has_version('1.5.3');

# if disk buffering fails for some reason, there should be
# no final chunk

chmod(0000, $t->testdir() . '/proxy_temp');
like(http_get_11('/proxy/big.html', sleep => 0.5),
        qr/X(?!.*\x0d\x0a?0\x0d\x0a?)/s, 'no proxy temp');

}

###############################################################################

sub http_get_11 {
        my ($uri, %extra) = @_;

        return http(
                "GET $uri HTTP/1.1" . CRLF .
                "Connection: close" . CRLF .
                "Host: localhost" . CRLF . CRLF,
                %extra
        );
}

###############################################################################

sub http_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";

        local $SIG{PIPE} = 'IGNORE';

        while (my $client = $server->accept()) {
                $client->autoflush(1);

                my $headers = '';
                my $uri = '';

                while (<$client>) {
                        $headers .= $_;
                        last if (/^\x0d?\x0a?$/);
                }

                $uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i;

                print $client
                        "HTTP/1.1 200 OK" . CRLF .
                        "Content-Length: 1000010" . CRLF .
                        "Cache-Control: max-age=300" . CRLF .
                        "Connection: close" . CRLF .
                        CRLF .
                        ("X" x (1000*1000)) . "finished" . CRLF;
        }
}

###############################################################################