Mercurial > hg > nginx-tests
diff body.t @ 240:462d89f5732a
Tests: request body and chunked transfer encoding tests.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Fri, 09 Nov 2012 07:46:37 +0400 |
parents | |
children | 6a0d934950bc |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/body.t @@ -0,0 +1,147 @@ +#!/usr/bin/perl + +# (C) Maxim Dounin + +# Tests for nginx request body reading. + +############################################################################### + +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; + +my $t = Test::Nginx->new()->has(qw/http proxy rewrite/)->plan(10); + +$t->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +http { + %%TEST_GLOBALS_HTTP%% + + server { + listen 127.0.0.1:8080; + server_name localhost; + + client_header_buffer_size 1k; + + location / { + client_body_buffer_size 2k; + add_header X-Body "$request_body"; + add_header X-Body-File "$request_body_file"; + proxy_pass http://127.0.0.1:8081; + } + location /b { + client_body_buffer_size 2k; + client_body_in_file_only on; + add_header X-Body "$request_body"; + add_header X-Body-File "$request_body_file"; + proxy_pass http://127.0.0.1:8081; + } + location /single { + client_body_in_single_buffer on; + add_header X-Body "$request_body"; + add_header X-Body-File "$request_body_file"; + proxy_pass http://127.0.0.1:8081; + } + location /discard { + return 200 "TEST\n"; + } + } + + server { + listen 127.0.0.1:8081; + server_name localhost; + + location / { + return 200 "TEST\n"; + } + } +} + +EOF + +$t->run(); + +############################################################################### + +unlike(http_get('/'), qr/X-Body:/ms, 'no body'); + +like(http_get_body('/', '0123456789'), + qr/X-Body: 0123456789\x0d?$/ms, 'body'); + +like(http_get_body('/', '0123456789' x 128), + qr/X-Body: (0123456789){128}\x0d?$/ms, 'body in two buffers'); + +like(http_get_body('/', '0123456789' x 512), + qr/X-Body-File/ms, 'body in file'); + +like(read_body_file(http_get_body('/b', '0123456789' x 512)), + qr/^(0123456789){512}$/s, 'body in file only'); + +like(http_get_body('/single', '0123456789' x 128), + qr/X-Body: (0123456789){128}\x0d?$/ms, 'body in single buffer'); + +# pipelined requests + +like(http_get_body('/', '0123456789', '0123456789' x 128, '0123456789' x 512, + 'foobar'), qr/X-Body: foobar\x0d?$/ms, 'body pipelined'); +like(http_get_body('/', '0123456789' x 128, '0123456789' x 512, '0123456789', + 'foobar'), qr/X-Body: foobar\x0d?$/ms, 'body pipelined 2'); + +like(http_get_body('/discard', '0123456789', '0123456789' x 128, + '0123456789' x 512, 'foobar'), qr/(TEST.*){4}/ms, + 'body discard'); +like(http_get_body('/discard', '0123456789' x 128, '0123456789' x 512, + '0123456789', 'foobar'), qr/(TEST.*){4}/ms, + 'body discard 2'); + +############################################################################### + +sub read_body_file { + my ($r) = @_; + return '' unless $r =~ m/X-Body-File: (.*)/; + open FILE, $1 + or return "$!"; + local $/; + my $content = <FILE>; + close FILE; + return $content; +} + +sub http_get_body { + my $uri = shift; + my $last = pop; + return http( join '', (map { + my $body = $_; + "GET $uri HTTP/1.1" . CRLF + . "Host: localhost" . CRLF + . "Content-Length: " . (length $body) . CRLF . CRLF + . $body + } @_), + "GET $uri HTTP/1.1" . CRLF + . "Host: localhost" . CRLF + . "Connection: close" . CRLF + . "Content-Length: " . (length $last) . CRLF . CRLF + . $last + ); +} + +###############################################################################