diff proxy_extra_data.t @ 1581:463d6863d360

Tests: tests for extra data and short responses.
author Maxim Dounin <mdounin@mdounin.ru>
date Mon, 06 Jul 2020 18:37:20 +0300
parents
children 6128590b0d46
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/proxy_extra_data.t
@@ -0,0 +1,211 @@
+#!/usr/bin/perl
+
+# (C) Maxim Dounin
+# (C) Nginx, Inc.
+
+# Tests for http backend with extra data.
+
+###############################################################################
+
+use warnings;
+use strict;
+
+use Test::More;
+
+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 cache rewrite addition/)->plan(20)
+	->write_file_expand('nginx.conf', <<'EOF');
+
+%%TEST_GLOBALS%%
+
+daemon off;
+
+events {
+}
+
+http {
+    %%TEST_GLOBALS_HTTP%%
+
+    proxy_cache_path cache keys_zone=one:1m;
+    proxy_cache_key $request_uri;
+    proxy_cache_valid any 1m;
+
+    server {
+        listen       127.0.0.1:8080;
+        server_name  localhost;
+
+        location / {
+            proxy_pass http://127.0.0.1:8081;
+            add_after_body /after;
+        }
+
+        location /unbuf/ {
+            proxy_pass http://127.0.0.1:8081;
+            proxy_buffering off;
+            add_after_body /after;
+        }
+
+        location /head/ {
+            proxy_pass http://127.0.0.1:8081;
+            proxy_cache one;
+            add_after_body /after;
+        }
+
+        location /after {
+            return 200 ":after\n";
+        }
+    }
+}
+
+EOF
+
+$t->run_daemon(\&http_daemon);
+$t->run()->waitforsocket('127.0.0.1:' . port(8081));
+
+###############################################################################
+
+TODO: {
+local $TODO = 'not yet' unless $t->has_version('1.19.1');
+
+like(http_get('/'), qr/SEE-THIS(?!-BUT-NOT-THIS)/, 'response with extra data');
+
+}
+
+like(http_get('/short'), qr/SEE-THIS(?!.*:after)/s, 'too short response');
+like(http_get('/empty'), qr/200 OK(?!.*:after)/s, 'empty too short response');
+
+like(http_head('/'), qr/200 OK(?!.*SEE-THIS)/s, 'no data in HEAD');
+like(http_head('/short'), qr/200 OK(?!.*SEE-THIS)/s, 'too short to HEAD');
+like(http_head('/empty'), qr/200 OK/, 'empty response to HEAD');
+
+# unbuffered responses
+
+TODO: {
+local $TODO = 'not yet' unless $t->has_version('1.19.1');
+
+like(http_get('/unbuf/'), qr/SEE-THIS(?!-BUT-NOT-THIS)/,
+	'unbuffered with extra data');
+
+}
+
+like(http_get('/unbuf/short'), qr/SEE-THIS(?!.*:after)/s,
+	'unbuffered too short response');
+like(http_get('/unbuf/empty'), qr/200 OK(?!.*:after)/s,
+	'unbuffered empty too short response');
+
+like(http_head('/unbuf/'), qr/200 OK(?!.*SEE-THIS)/s,
+	'unbuffered no data in HEAD');
+like(http_head('/unbuf/short'), qr/200 OK(?!.*SEE-THIS)/s,
+	'unbuffered too short response to HEAD');
+like(http_head('/unbuf/empty'), qr/200 OK/,
+	'unbuffered empty response to HEAD');
+
+# caching of responsses to HEAD requests
+
+like(http_head('/head/empty'), qr/200 OK(?!.*SEE-THIS)/s, 'head no body');
+like(http_head('/head/matching'), qr/200 OK(?!.*SEE-THIS)/s, 'head matching');
+like(http_head('/head/extra'), qr/200 OK(?!.*SEE-THIS)/s, 'head extra');
+like(http_head('/head/short'), qr/200 OK(?!.*SEE-THIS)/s, 'head too short');
+
+like(http_get('/head/empty'), qr/SEE-THIS/, 'head no body cached');
+like(http_get('/head/matching'), qr/SEE-THIS/, 'head matching cached');
+
+TODO: {
+local $TODO = 'not yet' unless $t->has_version('1.19.1');
+
+like(http_get('/head/extra'), qr/SEE-THIS(?!-BUT-NOT-THIS)/s,
+	'head extra cached');
+
+}
+
+like(http_get('/head/short'), qr/SEE-THIS(?!.*:after)/s,
+	'head too short cached');
+
+###############################################################################
+
+sub http_daemon {
+	my $server = IO::Socket::INET->new(
+		Proto => 'tcp',
+		LocalHost => '127.0.0.1:' . port(8081),
+		Listen => 5,
+		Reuse => 1
+	)
+		or die "Can't create listening socket: $!\n";
+
+	local $SIG{PIPE} = 'IGNORE';
+
+	my ($uri, $head);
+
+	while (my $c = $server->accept()) {
+		$c->autoflush(1);
+
+		my $headers = '';
+		my $uri = '';
+
+		while (<$c>) {
+			$headers .= $_;
+			last if (/^\x0d?\x0a?$/);
+		}
+
+		$uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i;
+		$uri =~ s!^/unbuf!!;
+
+		$head = ($headers =~ /^HEAD/);
+
+		if ($uri eq '/') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 8\n\n");
+			$c->print("SEE-THIS-BUT-NOT-THIS\n");
+
+		} elsif ($uri eq '/short') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 100\n\n");
+			$c->print("SEE-THIS-TOO-SHORT-RESPONSE\n");
+
+		} elsif ($uri eq '/empty') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 100\n\n");
+
+		} elsif ($uri eq '/head/empty') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 8\n\n");
+			$c->print("SEE-THIS") unless $head;
+
+		} elsif ($uri eq '/head/matching') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 8\n\n");
+			$c->print("SEE-THIS");
+
+		} elsif ($uri eq '/head/extra') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 8\n\n");
+			$c->print("SEE-THIS-BUT-NOT-THIS\n");
+
+		} elsif ($uri eq '/head/short') {
+			$c->print("HTTP/1.1 200 OK\n");
+			$c->print("Content-Type: text/html\n");
+			$c->print("Content-Length: 100\n\n");
+			$c->print("SEE-THIS\n");
+		}
+
+		close $c;
+	}
+}
+
+###############################################################################