changeset 619:58a67c40a761

Tests: stream limit conn tests.
author Andrey Zelenkov <zelenkov@nginx.com>
date Mon, 29 Jun 2015 21:40:30 +0300
parents 2d9f5f439598
children 7497adad1a50
files lib/Test/Nginx.pm stream_limit_conn.t
diffstat 2 files changed, 138 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/lib/Test/Nginx.pm
+++ b/lib/Test/Nginx.pm
@@ -145,6 +145,8 @@ sub has_module($) {
 		split_clients
 			=> '(?s)^(?!.*--without-http_split_clients_module)',
 		stream	=> '--with-stream(?!\S)',
+		stream_limit_conn =>
+			'(?s)^(?!.*--without-stream_limit_conn_module)',
 		stream_upstream_hash
 			=> '(?s)^(?!.*--without-stream_upstream_hash_module)',
 		stream_upstream_least_conn
new file mode 100644
--- /dev/null
+++ b/stream_limit_conn.t
@@ -0,0 +1,136 @@
+#!/usr/bin/perl
+
+# (C) Andrey Zelenkov
+# (C) Nginx, Inc.
+
+# Tests for stream limit_conn module.
+
+###############################################################################
+
+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 stream stream_limit_conn shmem/)
+	->write_file_expand('nginx.conf', <<'EOF');
+
+%%TEST_GLOBALS%%
+
+daemon off;
+
+events {
+}
+
+stream {
+    limit_conn_zone  $binary_remote_addr  zone=zone:1m;
+    limit_conn_zone  $binary_remote_addr  zone=zone2:1m;
+
+    server {
+        listen           127.0.0.1:8080;
+        proxy_pass       127.0.0.1:8084;
+        limit_conn       zone 1;
+    }
+
+    server {
+        listen           127.0.0.1:8085;
+        proxy_pass       127.0.0.1:8084;
+        limit_conn       zone 5;
+    }
+
+    server {
+        listen           127.0.0.1:8081;
+        proxy_pass       127.0.0.1:8084;
+        limit_conn       zone2 1;
+    }
+
+    server {
+        listen           127.0.0.1:8082;
+        proxy_pass       127.0.0.1:8080;
+        limit_conn       zone2 1;
+    }
+
+    server {
+        listen           127.0.0.1:8083;
+        proxy_pass       127.0.0.1:8080;
+        limit_conn       zone 1;
+    }
+}
+
+http {
+    %%TEST_GLOBALS_HTTP%%
+
+    server {
+        listen       127.0.0.1:8084;
+        server_name  localhost;
+
+        location / { }
+    }
+}
+
+EOF
+
+$t->write_file('index.html', '');
+$t->try_run('no stream limit_conn')->plan(8);
+
+###############################################################################
+
+like(get(), qr/200 OK/, 'passed');
+
+# same and other zones
+
+my $s = http(<<EOF, start => 1, sleep => 0.2);
+GET / HTTP/1.0
+EOF
+
+ok($s, 'long connection');
+
+is(get(), undef, 'rejected same zone');
+like(get('127.0.0.1:8081'), qr/200 OK/, 'passed different zone');
+like(get('127.0.0.1:8085'), qr/200 OK/, 'passed same zone unlimited');
+
+ok(http(<<EOF, socket => $s), 'long connection closed');
+Host: localhost
+
+EOF
+
+# zones proxy chain
+
+like(get('127.0.0.1:8082'), qr/200 OK/, 'passed proxy');
+is(get('127.0.0.1:8083'), undef, 'rejected proxy');
+
+###############################################################################
+
+sub get {
+	my $peer = shift;
+
+	my $r = http_get('/', socket => getconn($peer));
+	if (!$r) {
+		$r = undef;
+	}
+
+	return $r;
+}
+
+sub getconn {
+	my $peer = shift;
+	my $s = IO::Socket::INET->new(
+		Proto => 'tcp',
+		PeerAddr => $peer || '127.0.0.1:8080'
+	)
+		or die "Can't connect to nginx: $!\n";
+
+	return $s;
+}
+
+###############################################################################