Mercurial > hg > nginx-tests
view stream_js.t @ 1427:eb1d883305ea
Tests: avoid edge cases in upstream random two test.
Unavailable servers contribute to the number of attempts, if selected,
before the balancer would fall back to the default round-robin method.
This means that it's quite possible to get server with more connections.
To facilitate with selecting two alive servers, down server was removed
from the upstream configuration at the cost of slightly worse coverage.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Thu, 10 Jan 2019 17:42:34 +0300 |
parents | f168fc46c7a4 |
children | 2e81f7788cf7 |
line wrap: on
line source
#!/usr/bin/perl # (C) Andrey Zelenkov # (C) Dmitry Volyntsev # (C) Nginx, Inc. # Tests for stream njs 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/ dgram stream /; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new()->has(qw/http proxy rewrite stream stream_return udp/) ->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } http { %%TEST_GLOBALS_HTTP%% js_include test.js; server { listen 127.0.0.1:8079; server_name localhost; location /njs { js_content test_njs; } location /p/ { proxy_pass http://127.0.0.1:8095/; } location /return { return 200 $http_foo; } } } stream { js_set $js_addr js_addr; js_set $js_var js_var; js_set $js_log js_log; js_set $js_unk js_unk; js_set $js_req_line js_req_line; js_set $js_sess_unk js_sess_unk; js_include test.js; server { listen 127.0.0.1:8080; return $js_addr; } server { listen 127.0.0.1:8081; return $js_log; } server { listen 127.0.0.1:8082; return $js_var; } server { listen 127.0.0.1:8083; return $js_unk; } server { listen 127.0.0.1:8084; return $js_sess_unk; } server { listen 127.0.0.1:%%PORT_8985_UDP%% udp; return $js_addr; } server { listen 127.0.0.1:8086; js_access js_access_step; js_preread js_preread_step; js_filter js_filter_step; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8087; js_access js_access_undecided; return OK; } server { listen 127.0.0.1:8088; js_access js_access_allow; return OK; } server { listen 127.0.0.1:8089; js_access js_access_deny; return OK; } server { listen 127.0.0.1:8091; js_preread js_preread_async; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8092; js_preread js_preread_data; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8093; js_preread js_preread_req_line; return $js_req_line; } server { listen 127.0.0.1:8094; js_filter js_filter_empty; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8095; js_filter js_filter_header_inject; proxy_pass 127.0.0.1:8079; } server { listen 127.0.0.1:8096; js_filter js_filter_search; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8097; js_access js_access_except; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8098; js_preread js_preread_except; proxy_pass 127.0.0.1:8090; } server { listen 127.0.0.1:8099; js_filter js_filter_except; proxy_pass 127.0.0.1:8090; } } EOF $t->write_file('test.js', <<EOF); function test_njs(r) { r.return(200, njs.version); } function js_addr(s) { return 'addr=' + s.remoteAddress; } function js_var(s) { return 'variable=' + s.variables.remote_addr; } function js_sess_unk(s) { return 'sess_unk=' + s.unk; } function js_log(s) { s.log("SEE-THIS"); } var res = ''; function js_access_step(s) { res += '1'; setTimeout(function() { if (s.remoteAddress.match('127.0.0.1')) { s.allow(); } }, 1); } function js_preread_step(s) { s.on('upload', function (data) { res += '2'; if (res.length >= 3) { s.done(); } }); } function js_filter_step(s) { s.on('upload', function(data, flags) { s.send(data); res += '3'; }); s.on('download', function(data, flags) { if (!flags.last) { res += '4'; s.send(data); } else { res += '5'; s.send(res, {last:1}); s.off('download'); } }); } function js_access_undecided(s) { s.decline(); } function js_access_allow(s) { if (s.remoteAddress.match('127.0.0.1')) { s.done(); return; } s.abort(); } function js_access_deny(s) { if (s.remoteAddress.match('127.0.0.1')) { s.abort(); return; } s.allow(); } function js_preread_async(s) { setTimeout(function() { s.done(); }, 1); } function js_preread_data(s) { s.on('upload', function (data, flags) { if (data.indexOf('z') != -1) { s.done(); } }); } var line = ''; function js_preread_req_line(s) { s.on('upload', function (data, flags) { var n = data.indexOf('\n'); if (n != -1) { line = data.substr(0, n); s.done(); } }); } function js_req_line(s) { return line; } function js_filter_empty(s) { } function js_filter_header_inject(s) { var req = ''; s.on('upload', function(data, flags) { req += data; var n = req.search('\n'); if (n != -1) { var rest = req.substr(n + 1); req = req.substr(0, n + 1); s.send(req + 'Foo: foo' + '\r\n' + rest, flags); s.off('upload'); } }); } function js_filter_search(s) { s.on('download', function(data, flags) { var n = data.search('y'); if (n != -1) { s.send('z'); } }); s.on('upload', function(data, flags) { var n = data.search('x'); if (n != -1) { s.send('y'); } }); } function js_access_except(s) { function done() {return s.a.a}; setTimeout(done, 1); setTimeout(done, 2); } function js_preread_except(s) { var fs = require('fs'); fs.readFileSync(); } function js_filter_except(s) { s.on('unknown', function() {}); } EOF $t->run_daemon(\&stream_daemon, port(8090)); $t->try_run('no stream njs available')->plan(19); $t->waitforsocket('127.0.0.1:' . port(8090)); ############################################################################### is(stream('127.0.0.1:' . port(8080))->read(), 'addr=127.0.0.1', 's.remoteAddress'); is(dgram('127.0.0.1:' . port(8985))->io('.'), 'addr=127.0.0.1', 's.remoteAddress udp'); is(stream('127.0.0.1:' . port(8081))->read(), 'undefined', 's.log'); is(stream('127.0.0.1:' . port(8082))->read(), 'variable=127.0.0.1', 's.variables'); is(stream('127.0.0.1:' . port(8083))->read(), '', 'stream js unknown function'); is(stream('127.0.0.1:' . port(8084))->read(), 'sess_unk=undefined', 's.unk'); TODO: { local $TODO = 'not yet' unless get('/njs') =~ /^([.0-9]+)$/m && $1 ge '0.2.4'; is(stream('127.0.0.1:' . port(8086))->io('0'), '0122345', 'async handlers order'); is(stream('127.0.0.1:' . port(8087))->io('#'), 'OK', 'js_access_undecided'); is(stream('127.0.0.1:' . port(8088))->io('#'), 'OK', 'js_access_allow'); is(stream('127.0.0.1:' . port(8089))->io('#'), '', 'js_access_deny'); is(stream('127.0.0.1:' . port(8091))->io('#'), '#', 'js_preread_async'); is(stream('127.0.0.1:' . port(8092))->io('#z'), '#z', 'js_preread_async_data'); is(stream('127.0.0.1:' . port(8093))->io("xy\na"), 'xy', 'js_preread_req_line'); is(stream('127.0.0.1:' . port(8094))->io('x'), 'x', 'js_filter_empty'); like(get('/p/return'), qr/foo/, 'js_filter_injected_header'); is(stream('127.0.0.1:' . port(8096))->io('x'), 'z', 'js_filter_search'); } stream('127.0.0.1:' . port(8097))->io('x'); stream('127.0.0.1:' . port(8098))->io('x'); stream('127.0.0.1:' . port(8099))->io('x'); $t->stop(); ok(index($t->read_file('error.log'), 'SEE-THIS') > 0, 'stream js log'); ok(index($t->read_file('error.log'), 'at fs.readFileSync') > 0, 'stream js_preread backtrace'); ok(index($t->read_file('error.log'), 'at js_filter_except') > 0, 'stream js_filter backtrace'); ############################################################################### sub stream_daemon { my $server = IO::Socket::INET->new( Proto => 'tcp', LocalAddr => '127.0.0.1:' . port(8090), Listen => 5, Reuse => 1 ) or die "Can't create listening socket: $!\n"; local $SIG{PIPE} = 'IGNORE'; while (my $client = $server->accept()) { $client->autoflush(1); log2c("(new connection $client)"); $client->sysread(my $buffer, 65536) or next; log2i("$client $buffer"); log2o("$client $buffer"); $client->syswrite($buffer); close $client; } } sub log2i { Test::Nginx::log_core('|| <<', @_); } sub log2o { Test::Nginx::log_core('|| >>', @_); } sub log2c { Test::Nginx::log_core('||', @_); } sub get { my ($url, %extra) = @_; my $s = IO::Socket::INET->new( Proto => 'tcp', PeerAddr => '127.0.0.1:' . port(8079) ) or die "Can't connect to nginx: $!\n"; return http_get($url, socket => $s); } ###############################################################################