# HG changeset patch # User Sergey Kandaurov # Date 1413479589 -14400 # Node ID c872b2c9645fdf0e8203a5ed6c7a3c2c797c1a59 # Parent b86c05516e6556d49745f3d15d109907a4414b7a Tests: tests for limit_conn and limit_req with complex value. diff --git a/limit_conn_complex.t b/limit_conn_complex.t new file mode 100644 --- /dev/null +++ b/limit_conn_complex.t @@ -0,0 +1,92 @@ +#!/usr/bin/perl + +# (C) Sergey Kandaurov +# (C) Nginx, Inc. + +# limit_req based tests for limit_conn module with complex keys. + +############################################################################### + +use warnings; +use strict; + +use Test::More; + +use IO::Select; + +BEGIN { use FindBin; chdir($FindBin::Bin); } + +use lib 'lib'; +use Test::Nginx; + +############################################################################### + +select STDERR; $| = 1; +select STDOUT; $| = 1; + +plan(skip_all => 'win32') if $^O eq 'MSWin32'; + +my $t = Test::Nginx->new()->has(qw/http proxy limit_conn limit_req/); + +$t->write_file_expand('nginx.conf', <<'EOF'); + +%%TEST_GLOBALS%% + +daemon off; + +events { +} + +http { + %%TEST_GLOBALS_HTTP%% + + limit_req_zone $binary_remote_addr$arg_r zone=req:1m rate=1r/s; + limit_conn_zone $binary_remote_addr$arg_c zone=conn:1m; + + server { + listen 127.0.0.1:8080; + server_name localhost; + + location / { + limit_conn conn 1; + } + + location /w { + limit_conn conn 1; + proxy_pass http://127.0.0.1:8080/req; + } + + location /req { + limit_req zone=req burst=1; + } + } +} + +EOF + +$t->try_run('no complex value')->plan(4); + +############################################################################### + +my $s; + +# charge limit_req + +http_get('/req'); + +# limit_req tests + +$s = http_get('/req', start => 1); +ok(!IO::Select->new($s)->can_read(0.1), 'limit_req same key'); + +$s = http_get('/req?r=2', start => 1); +ok(IO::Select->new($s)->can_read(0.1), 'limit_req different key'); + +# limit_conn tests + +$s = http_get('/w', start => 1); + +like(http_get('/'), qr/^HTTP\/1.. 503 /, 'limit_conn same key'); +unlike(http_get('/?c=2'), qr/^HTTP\/1.. 503 /, 'limit_conn different key'); + +###############################################################################