Mercurial > hg > nginx-tests

comparison stream_udp_upstream_least_conn.t @ 868:d2cb9ed7412e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: stream udp tests.
author Andrey Zelenkov <zelenkov@nginx.com>
date Tue, 15 Mar 2016 16:20:29 +0300
parents
children b9e42c554ba7
comparison
equal deleted inserted replaced
867:89898b20f258 868:d2cb9ed7412e
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Nginx, Inc.
5
6 # Stream tests for upstream least_conn balancer module with datagrams.
7
8 ###############################################################################
9
10 use warnings;
11 use strict;
12
13 use Test::More;
14
15 BEGIN { use FindBin; chdir($FindBin::Bin); }
16
17 use lib 'lib';
18 use Test::Nginx;
19 use Test::Nginx::Stream qw/ dgram /;
20
21 ###############################################################################
22
23 select STDERR; $| = 1;
24 select STDOUT; $| = 1;
25
26 my $t = Test::Nginx->new()->has(qw/stream stream_upstream_least_conn/)
27 ->write_file_expand('nginx.conf', <<'EOF');
28
29 %%TEST_GLOBALS%%
30
31 daemon off;
32
33 events {
34 }
35
36 stream {
37 proxy_responses 1;
38 proxy_timeout 1s;
39
40 upstream u {
41 least_conn;
42 server 127.0.0.1:8081;
43 server 127.0.0.1:8082;
44 }
45
46 server {
47 listen 127.0.0.1:8080 udp;
48 proxy_pass u;
49 }
50 }
51
52 EOF
53
54 $t->run_daemon(\&udp_daemon, 8081, $t);
55 $t->run_daemon(\&udp_daemon, 8082, $t);
56 $t->try_run('no stream udp')->plan(2);
57
58 $t->waitforfile($t->testdir . '/8081');
59 $t->waitforfile($t->testdir . '/8082');
60
61 ###############################################################################
62
63 is(many('.', 10), '8081: 5, 8082: 5', 'balanced');
64
65 my @sockets;
66 for (1 .. 2) {
67 my $s = dgram();
68 $s->write('w');
69 push @sockets, $s;
70 }
71
72 select undef, undef, undef, 0.2;
73
74 is(many('.', 10), '8082: 10', 'least_conn');
75
76 ###############################################################################
77
78 sub many {
79 my ($data, $count, %opts) = @_;
80 my (%ports);
81
82 for (1 .. $count) {
83 if (dgram()->io($data) =~ /(\d+)/) {
84 $ports{$1} = 0 unless defined $ports{$1};
85 $ports{$1}++;
86 }
87 }
88
89 return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports;
90 }
91
92 ###############################################################################
93
94 sub udp_daemon {
95 my ($port, $t) = @_;
96
97 my $server = IO::Socket::INET->new(
98 Proto => 'udp',
99 LocalAddr => '127.0.0.1:' . $port,
100 Reuse => 1,
101 )
102 or die "Can't create listening socket: $!\n";
103
104 # signal we are ready
105
106 open my $fh, '>', $t->testdir() . '/' . $port;
107 close $fh;
108
109 while (1) {
110 $server->recv(my $buffer, 65536);
111
112 my $port = $server->sockport();
113
114 if ($buffer =~ /w/ && $port == 8081) {
115 select undef, undef, undef, 2.5;
116 }
117
118 $buffer = $port;
119
120 $server->send($buffer);
121 }
122 }
123
124 ###############################################################################