test/apisix
1
0
Fork 0
mirror of https://gitee.com/iresty/apisix.git synced 2024-12-14 17:01:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
9309abaed7
apisix/t/plugin/limit-count.t

458 lines
12 KiB
Perl
Raw Normal View History

bugfix: only accept client IP as key now. (#85)
2019-06-13 19:11:15 +08:00
use t::APISix 'no_plan';
bugfix: added the plugin load. (#95) * bugfix: added the plugin load. * test: added test cases for limit count. * test: added Lua sleep at the end of init_worker. left enough time for read data from etcd, otherwise worker maybe deal with request before get data from etcd. * test: added no_shuffle();
2019-06-14 16:11:14 +08:00
repeat_each(1);
bugfix: only accept client IP as key now. (#85)
2019-06-13 19:11:15 +08:00
no_long_string();
bugfix: added the plugin load. (#95) * bugfix: added the plugin load. * test: added test cases for limit count. * test: added Lua sleep at the end of init_worker. left enough time for read data from etcd, otherwise worker maybe deal with request before get data from etcd. * test: added no_shuffle();
2019-06-14 16:11:14 +08:00
no_shuffle();
bugfix: only accept client IP as key now. (#85)
2019-06-13 19:11:15 +08:00
no_root_location();
run_tests;
__DATA__
=== TEST 1: sanity
--- config
location /t {
content_by_lua_block {
local plugin = require("apisix.plugins.limit-count")
local ok, err = plugin.check_schema({count = 2, time_window = 60, rejected_code = 503, key = 'remote_addr'})
if not ok then
ngx.say(err)
end
ngx.say("done")
}
}
--- request
GET /t
--- response_body
done
--- no_error_log
[error]
=== TEST 2: wrong value of key
--- config
location /t {
content_by_lua_block {
local plugin = require("apisix.plugins.limit-count")
local ok, err = plugin.check_schema({count = 2, time_window = 60, rejected_code = 503, key = 'host'})
if not ok then
ngx.say(err)
end
ngx.say("done")
}
}
--- request
GET /t
--- response_body
invalid "enum" in docuement at pointer "#/key"
done
--- no_error_log
[error]
bugfix: added the plugin load. (#95) * bugfix: added the plugin load. * test: added test cases for limit count. * test: added Lua sleep at the end of init_worker. left enough time for read data from etcd, otherwise worker maybe deal with request before get data from etcd. * test: added no_shuffle();
2019-06-14 16:11:14 +08:00
=== TEST 3: set route(id: 1)
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/routes/1',
ngx.HTTP_PUT,
[[{
"methods": ["GET"],
"plugins": {
"limit-count": {
"count": 2,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.say(body)
}
}
--- request
GET /t
--- response_body
passed
--- no_error_log
[error]
=== TEST 4: up the limit
--- pipelined_requests eval
["GET /hello", "GET /hello", "GET /hello", "GET /hello"]
--- error_code eval
[200, 200, 503, 503]
--- no_error_log
[error]
=== TEST 5: up the limit
--- pipelined_requests eval
["GET /hello1", "GET /hello", "GET /hello2", "GET /hello", "GET /hello"]
--- error_code eval
[404, 200, 404, 200, 503]
--- no_error_log
[error]
feature: used json schema to check the configuration of `limit-count` plugin. (#116) * feature: used json schema to check the configuration of `limit-count` plugin. * travis: if there was any errors, exist directly.
2019-06-17 15:05:19 +08:00
=== TEST 6: invalid route: missing key
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/routes/1',
ngx.HTTP_PUT,
[[{
"plugins": {
"limit-count": {
"count": 2,
"time_window": 60,
"rejected_code": 503
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"required\" in docuement at pointer \"#\""}
--- no_error_log
[error]
=== TEST 7: invalid route: wrong count
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/routes/1',
ngx.HTTP_PUT,
[[{
"plugins": {
"limit-count": {
"count": -100,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"minimum\" in docuement at pointer \"#\/count\""}
--- no_error_log
[error]
=== TEST 8: invalid route: wrong count + POST method
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
change: merged the similar code, and not allowed to specify id when use POST method. (#128) * change: merged the similar code, and not allowed to specify id when use POST method.
2019-06-18 16:15:55 +08:00
local code, body = t('/apisix/admin/routes',
feature: used json schema to check the configuration of `limit-count` plugin. (#116) * feature: used json schema to check the configuration of `limit-count` plugin. * travis: if there was any errors, exist directly.
2019-06-17 15:05:19 +08:00
ngx.HTTP_POST,
[[{
"plugins": {
"limit-count": {
"count": -100,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"minimum\" in docuement at pointer \"#\/count\""}
--- no_error_log
[error]
=== TEST 9: invalid service: missing key
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/services/1',
ngx.HTTP_PUT,
[[{
"plugins": {
"limit-count": {
"count": 2,
"time_window": 60,
"rejected_code": 503
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
}
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"required\" in docuement at pointer \"#\""}
--- no_error_log
[error]
=== TEST 10: invalid service: wrong count
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/services/1',
ngx.HTTP_PUT,
[[{
"plugins": {
"limit-count": {
"count": -100,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
}
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"minimum\" in docuement at pointer \"#\/count\""}
--- no_error_log
[error]
=== TEST 11: invalid service: wrong count + POST method
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
change: merged the similar code, and not allowed to specify id when use POST method. (#128) * change: merged the similar code, and not allowed to specify id when use POST method.
2019-06-18 16:15:55 +08:00
local code, body = t('/apisix/admin/services',
feature: used json schema to check the configuration of `limit-count` plugin. (#116) * feature: used json schema to check the configuration of `limit-count` plugin. * travis: if there was any errors, exist directly.
2019-06-17 15:05:19 +08:00
ngx.HTTP_POST,
[[{
"plugins": {
"limit-count": {
"count": -100,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
}
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.print(body)
}
}
--- request
GET /t
--- error_code: 400
--- response_body
{"error_msg":"failed to check the configuration of plugin limit-count err: invalid \"minimum\" in docuement at pointer \"#\/count\""}
--- no_error_log
[error]
bugfix: got route id from url if not found in post body. (#120)
2019-06-17 18:00:56 +08:00
=== TEST 12: set route without id in post body
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/routes/1',
ngx.HTTP_PUT,
[[{
"plugins": {
"limit-count": {
"count": 2,
"time_window": 60,
"rejected_code": 503,
"key": "remote_addr"
}
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.say(body)
}
}
--- request
GET /t
--- response_body
passed
--- no_error_log
[error]
=== TEST 13: up the limit
--- pipelined_requests eval
["GET /hello", "GET /hello", "GET /hello", "GET /hello"]
--- error_code eval
[200, 200, 503, 503]
--- no_error_log
[error]
test: added test cases for limit conn and limit count. (#131)
2019-06-19 13:06:03 +08:00
=== TEST 14: disable plugin
--- config
location /t {
content_by_lua_block {
local t = require("lib.test_admin").test
local code, body = t('/apisix/admin/routes/1',
ngx.HTTP_PUT,
[[{
"plugins": {
},
"upstream": {
"nodes": {
"127.0.0.1:1980": 1
},
"type": "roundrobin"
},
"uri": "/hello"
}]]
)
if code >= 300 then
ngx.status = code
end
ngx.say(body)
}
}
--- request
GET /t
--- response_body
passed
--- no_error_log
[error]
=== TEST 15: up the limit
--- pipelined_requests eval
["GET /hello", "GET /hello", "GET /hello", "GET /hello"]
--- error_code eval
[200, 200, 200, 200]
--- no_error_log
[error]
Reference in New Issue Copy Permalink