changeset 979:ef6be3201851

Tests: basic tests for stream split_clients module.
author Sergey Kandaurov <pluknet@nginx.com>
date Wed, 13 Jul 2016 13:25:33 +0300
parents 04cb1849005a
children d10ec0f0bb6d
files lib/Test/Nginx.pm stream_split_clients.t
diffstat 2 files changed, 79 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/lib/Test/Nginx.pm
+++ b/lib/Test/Nginx.pm
@@ -174,6 +174,8 @@ sub has_module($) {
 			=> '(?s)^(?!.*--without-stream_map_module)',
 		stream_return
 			=> '(?s)^(?!.*--without-stream_return_module)',
+		stream_split_clients
+			=> '(?s)^(?!.*--without-stream_split_clients_module)',
 		stream_upstream_hash
 			=> '(?s)^(?!.*--without-stream_upstream_hash_module)',
 		stream_upstream_least_conn
new file mode 100644
--- /dev/null
+++ b/stream_split_clients.t
@@ -0,0 +1,77 @@
+#!/usr/bin/perl
+
+# (C) Sergey Kandaurov
+# (C) Nginx, Inc.
+
+# Tests for stream split_client module.
+
+###############################################################################
+
+use warnings;
+use strict;
+
+use Test::More;
+
+BEGIN { use FindBin; chdir($FindBin::Bin); }
+
+use lib 'lib';
+use Test::Nginx;
+use Test::Nginx::Stream qw/ stream /;
+
+###############################################################################
+
+select STDERR; $| = 1;
+select STDOUT; $| = 1;
+
+my $t = Test::Nginx->new()->has(qw/stream stream_split_clients stream_return/);
+
+$t->write_file_expand('nginx.conf', <<'EOF');
+
+%%TEST_GLOBALS%%
+
+daemon off;
+
+events {
+}
+
+stream {
+    split_clients $connection $variant {
+        51.2%  "first";
+        10%    "second";
+        *      "third";
+    }
+
+    server {
+        listen  127.0.0.1:8080;
+        return  $variant;
+    }
+}
+
+EOF
+
+$t->try_run('no stream split_clients');
+$t->plan(1);
+
+###############################################################################
+
+# NB: split_clients distribution is a subject to implementation details
+
+like(many('/', 20), qr/first: 12, second: 2, third: 6/, 'split');
+
+###############################################################################
+
+sub many {
+	my ($uri, $count) = @_;
+	my %dist;
+
+	for (1 .. $count) {
+		if (my $data = stream()->read()) {
+			$dist{$data} = 0 unless defined $data;
+			$dist{$data}++;
+		}
+	}
+
+	return join ', ', map { $_ . ": " . $dist{$_} } sort keys %dist;
+}
+
+###############################################################################