# HG changeset patch # User Sergey Kandaurov # Date 1468405533 -10800 # Node ID ef6be32018517755e0f541e96f2a7c42eb1ff83b # Parent 04cb1849005aa165ae1325a9995f41974fd445a3 Tests: basic tests for stream split_clients module. diff --git a/lib/Test/Nginx.pm b/lib/Test/Nginx.pm --- 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 diff --git a/stream_split_clients.t b/stream_split_clients.t 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; +} + +###############################################################################