-
Notifications
You must be signed in to change notification settings - Fork 81
/
Copy pathlimit_req2.t
97 lines (65 loc) · 2.48 KB
/
limit_req2.t
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/usr/bin/perl
# (C) Sergey Kandaurov
# (C) Nginx, Inc.
# Tests for nginx limit_req module, multiple limits.
###############################################################################
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 limit_req/)->plan(14);
$t->write_file_expand('nginx.conf', <<'EOF');
%%TEST_GLOBALS%%
daemon off;
events {
}
http {
%%TEST_GLOBALS_HTTP%%
limit_req_zone $arg_a zone=slow:1m rate=1r/m;
limit_req_zone $arg_b zone=fast:1m rate=1000r/s;
server {
listen 127.0.0.1:8080;
server_name localhost;
location / {
limit_req zone=fast;
limit_req zone=slow;
}
location /t2.html {
limit_req zone=fast nodelay;
limit_req zone=slow nodelay;
alias %%TESTDIR%%/t1.html;
}
}
}
EOF
$t->write_file('t1.html', 'XtestX');
$t->run();
###############################################################################
like(http_get('/t1.html?b=1'), qr/^HTTP\/1.. 200 /m, 'fast');
select undef, undef, undef, 0.1;
like(http_get('/t1.html?b=1'), qr/^HTTP\/1.. 200 /m, 'fast - passed');
like(http_get('/t1.html?a=1'), qr/^HTTP\/1.. 200 /m, 'slow');
select undef, undef, undef, 0.1;
like(http_get('/t1.html?a=1'), qr/^HTTP\/1.. 503 /m, 'slow - rejected');
like(http_get('/t1.html?a=2&b=2'), qr/^HTTP\/1.. 200 /m, 'both');
select undef, undef, undef, 0.1;
like(http_get('/t1.html?a=2&b=2'), qr/^HTTP\/1.. 503 /m, 'both - rejected');
like(http_get('/t1.html'), qr/^HTTP\/1.. 200 /m, 'no key');
like(http_get('/t1.html'), qr/^HTTP\/1.. 200 /m, 'no key - passed');
# nodelay
like(http_get('/t2.html?b=3'), qr/^HTTP\/1.. 200 /m, 'nodelay fast');
select undef, undef, undef, 0.1;
like(http_get('/t2.html?b=3'), qr/^HTTP\/1.. 200 /m, 'nodelay fast - passed');
like(http_get('/t2.html?a=3'), qr/^HTTP\/1.. 200 /m, 'nodelay slow');
select undef, undef, undef, 0.1;
like(http_get('/t2.html?a=3'), qr/^HTTP\/1.. 503 /m, 'nodelay slow - rejected');
like(http_get('/t2.html?a=4&b=4'), qr/^HTTP\/1.. 200 /m, 'nodelay both');
select undef, undef, undef, 0.1;
like(http_get('/t2.html?a=4&b=4'), qr/^HTTP\/1.. 503 /m,
'nodelay both - rejected');
###############################################################################