blob: 9b3416734ff244904cb417727e468e01c9372b5b [file] [log] [blame]
#!/usr/bin/perl
# (C) Maxim Dounin
# (C) Sergey Kandaurov
# (C) Nginx, Inc.
# Tests for grpc module, grpc_next_upstream directive.
###############################################################################
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 http_v2 grpc rewrite/)->plan(9);
$t->write_file_expand('nginx.conf', <<'EOF');
%%TEST_GLOBALS%%
daemon off;
events {
}
http {
%%TEST_GLOBALS_HTTP%%
upstream u {
server 127.0.0.1:8081 max_fails=2;
server 127.0.0.1:8082;
}
upstream u2 {
server 127.0.0.1:8081;
server 127.0.0.1:8082;
}
server {
listen 127.0.0.1:8080;
server_name localhost;
location / {
grpc_pass u;
grpc_next_upstream http_500 http_404 invalid_header;
}
location /all/ {
grpc_pass u2;
grpc_next_upstream http_500 http_404;
error_page 404 /all/404;
grpc_intercept_errors on;
}
location /all/404 {
return 200 "$upstream_addr\n";
}
}
server {
listen 127.0.0.1:8081 http2;
server_name localhost;
location / {
return 404;
}
location /ok {
return 200 "AND-THIS\n";
}
location /500 {
return 500;
}
location /444 {
return 444;
}
location /all/ {
return 404;
}
}
server {
listen 127.0.0.1:8082 http2;
server_name localhost;
location / {
return 200 "TEST-OK-IF-YOU-SEE-THIS\n";
}
location /all/ {
return 404;
}
}
}
EOF
$t->run();
###############################################################################
my ($p1, $p2) = (port(8081), port(8082));
# check if both request fallback to a backend
# which returns valid response
like(http_get('/'), qr/SEE-THIS/, 'grpc request');
like(http_get('/'), qr/SEE-THIS/, 'second request');
# make sure backend isn't switched off after
# grpc_next_upstream http_404
like(http_get('/ok') . http_get('/ok'), qr/AND-THIS/, 'not down');
# next upstream on invalid_header
like(http_get('/444'), qr/SEE-THIS/, 'request 444');
like(http_get('/444'), qr/SEE-THIS/, 'request 444 second');
# next upstream on http_500
like(http_get('/500'), qr/SEE-THIS/, 'request 500');
like(http_get('/500'), qr/SEE-THIS/, 'request 500 second');
# make sure backend switched off with http_500
unlike(http_get('/ok') . http_get('/ok'), qr/AND-THIS/, 'down after 500');
# make sure all backends are tried once
like(http_get('/all/rr'),
qr/^127.0.0.1:($p1, 127.0.0.1:$p2|$p2, 127.0.0.1:$p1)$/mi,
'all tried once');
###############################################################################