changeset 287:ba5b92378653

Tests: add some basic upstream tests.
author Maxim Dounin <mdounin@mdounin.ru>
date Thu, 23 May 2013 19:32:55 +0400
parents f62137d1b5b1
children 56157712d744
files upstream.t
diffstat 1 files changed, 153 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/upstream.t
@@ -0,0 +1,153 @@
+#!/usr/bin/perl
+
+# (C) Maxim Dounin
+
+# Tests for upstream module and balancers.
+
+###############################################################################
+
+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/)->plan(3);
+
+$t->write_file_expand('nginx.conf', <<'EOF');
+
+%%TEST_GLOBALS%%
+
+daemon off;
+
+events {
+}
+
+http {
+    %%TEST_GLOBALS_HTTP%%
+
+    upstream u {
+        server 127.0.0.1:8081 max_fails=3 fail_timeout=10s;
+        server 127.0.0.1:8082 max_fails=3 fail_timeout=10s;
+    }
+
+    upstream u2 {
+        server 127.0.0.1:8081 max_fails=3 fail_timeout=10s;
+        server 127.0.0.1:8082 max_fails=3 fail_timeout=10s;
+    }
+
+    server {
+        listen       127.0.0.1:8080;
+        server_name  localhost;
+
+        location / {
+            proxy_pass http://u;
+        }
+        location /close2 {
+            proxy_pass http://u2;
+        }
+    }
+}
+
+EOF
+
+$t->run_daemon(\&http_daemon, 8081);
+$t->run_daemon(\&http_daemon, 8082);
+$t->run();
+
+###############################################################################
+
+is(many('/', 30), '8081: 15, 8082: 15', 'balanced');
+
+# from 9 first requests to 8081, only 6 will be successfull,
+# 3rd, 6th, and 9th requests will fail; after this the backend
+# will be considered down and won't be used till fail_timeout passes
+
+is(many('/close', 30), '8081: 6, 8082: 24', 'failures');
+
+SKIP: {
+local $TODO = 'broken in 1.3.0';
+
+skip 'long test', 1 unless $ENV{TEST_NGINX_UNSAFE};
+
+# bug: failures counter is reset if first request in a second succeeds
+#
+# delay added to make sure first 9 requests will take more than 1s;
+# note that the test is racy and may unexpectedly succeed
+
+is(many('/close2', 30, delay => 0.2), '8081: 6, 8082: 24', 'failures delay');
+
+}
+
+###############################################################################
+
+sub many {
+	my ($uri, $count, %opts) = @_;
+	my %ports;
+
+	for (1 .. 30) {
+		if (http_get($uri) =~ /X-Port: (\d+)/) {
+			$ports{$1} = 0 unless defined $ports{$1};
+			$ports{$1}++;
+		}
+
+		select undef, undef, undef, $opts{delay} if $opts{delay};
+	}
+
+	return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports;
+}
+
+###############################################################################
+
+sub http_daemon {
+	my ($port) = @_;
+	my $count = 1;
+
+	my $server = IO::Socket::INET->new(
+		Proto => 'tcp',
+		LocalHost => '127.0.0.1',
+		LocalPort => $port,
+		Listen => 5,
+		Reuse => 1
+	)
+		or die "Can't create listening socket: $!\n";
+
+	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;
+
+		if ($uri =~ 'close' && $port == 8081 && $count++ % 3 == 0) {
+			next;
+		}
+
+		print $client <<EOF;
+HTTP/1.1 200 OK
+Connection: close
+X-Port: $port
+
+OK
+EOF
+
+		close $client;
+	}
+}
+
+###############################################################################