Mercurial > hg > nginx-tests
view proxy_next_upstream_tries.t @ 1990:85d88cd5091c
Tests: SSI include stub tests.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Thu, 04 Jul 2024 17:42:55 +0300 |
parents | e4974af3fb12 |
children |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for http proxy module, proxy_next_upstream_tries # and proxy_next_upstream_timeout directives. ############################################################################### 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 rewrite/)->plan(8); $t->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } http { %%TEST_GLOBALS_HTTP%% upstream u { server 127.0.0.1:8081; server 127.0.0.1:8081; server 127.0.0.1:8081; } upstream u2 { server 127.0.0.1:8081; server 127.0.0.1:8081 backup; server 127.0.0.1:8081 backup; } server { listen 127.0.0.1:8080; server_name localhost; proxy_next_upstream http_404; proxy_intercept_errors on; error_page 404 /404; location /tries { proxy_pass http://u; proxy_next_upstream_tries 2; } location /tries/backup { proxy_pass http://u2; proxy_next_upstream_tries 2; } location /tries/resolver { resolver 127.0.0.1:%%PORT_8982_UDP%%; proxy_pass http://$host:%%PORT_8081%%; proxy_next_upstream_tries 2; } location /tries/zero { proxy_pass http://u; proxy_next_upstream_tries 0; } location /timeout { proxy_pass http://u/w2; proxy_next_upstream_timeout 3800ms; } location /timeout/backup { proxy_pass http://u2/w2; proxy_next_upstream_timeout 3800ms; } location /timeout/resolver { resolver 127.0.0.1:%%PORT_8982_UDP%%; proxy_pass http://$host:%%PORT_8081%%/w2; proxy_next_upstream_timeout 3800ms; } location /timeout/zero { proxy_pass http://u/w; proxy_next_upstream_timeout 0; } location /404 { return 200 x${upstream_status}x; } } } EOF $t->run_daemon(\&http_daemon, port(8081)); $t->run_daemon(\&dns_daemon, port(8982), $t); $t->run(); $t->waitforsocket('127.0.0.1:' . port(8081)); $t->waitforfile($t->testdir . '/' . port(8982)); ############################################################################### like(http_get('/tries'), qr/x404, 404x/, 'tries'); like(http_get('/tries/backup'), qr/x404, 404x/, 'tries backup'); like(http_get('/tries/resolver'), qr/x404, 404x/, 'tries resolved'); like(http_get('/tries/zero'), qr/x404, 404, 404x/, 'tries zero'); # two tries fit into 1.9s SKIP: { skip 'long tests', 4 unless $ENV{TEST_NGINX_UNSAFE}; like(http_get('/timeout'), qr/x404, 404x/, 'timeout'); like(http_get('/timeout/backup'), qr/x404, 404x/, 'timeout backup'); like(http_get('/timeout/resolver'), qr/x404, 404x/, 'timeout resolved'); like(http_get('/timeout/zero'), qr/x404, 404, 404x/, 'timeout zero'); } ############################################################################### sub http_daemon { my ($port) = @_; 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"; local $SIG{PIPE} = 'IGNORE'; while (my $client = $server->accept()) { $client->autoflush(1); my $headers = ''; my $uri = ''; while (<$client>) { $headers .= $_; last if (/^\x0d?\x0a?$/); } next if $headers eq ''; $uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i; if ($uri eq '/w') { Test::Nginx::log_core('||', "$port: sleep(1)"); select undef, undef, undef, 1; } if ($uri eq '/w2') { Test::Nginx::log_core('||', "$port: sleep(2)"); select undef, undef, undef, 2; } Test::Nginx::log_core('||', "$port: response, 404"); print $client <<EOF; HTTP/1.1 404 Not Found Connection: close EOF } continue { close $client; } } sub reply_handler { my ($recv_data) = @_; my (@name, @rdata); use constant NOERROR => 0; use constant A => 1; use constant IN => 1; # default values my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600); # decode name my ($len, $offset) = (undef, 12); while (1) { $len = unpack("\@$offset C", $recv_data); last if $len == 0; $offset++; push @name, unpack("\@$offset A$len", $recv_data); $offset += $len; } $offset -= 1; my ($id, $type, $class) = unpack("n x$offset n2", $recv_data); @rdata = map { rd_addr($ttl, '127.0.0.1') } (1 .. 3) if $type == A; $len = @name; pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata, 0, 0, @name, $type, $class) . join('', @rdata); } sub rd_addr { my ($ttl, $addr) = @_; my $code = 'split(/\./, $addr)'; return pack 'n3N', 0xc00c, A, IN, $ttl if $addr eq ''; pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code); } sub dns_daemon { my ($port, $t) = @_; my ($data, $recv_data); my $socket = IO::Socket::INET->new( LocalAddr => '127.0.0.1', LocalPort => $port, Proto => 'udp', ) or die "Can't create listening socket: $!\n"; # signal we are ready open my $fh, '>', $t->testdir() . '/' . $port; close $fh; while (1) { $socket->recv($recv_data, 65536); $data = reply_handler($recv_data); $socket->send($data); } } ###############################################################################