mirror of
https://gitee.com/iresty/apisix.git
synced 2024-12-14 08:51:29 +08:00
808d511dd4
* feat(stream): add limit-conn Signed-off-by: spacewander <spacewanderlzx@gmail.com> * break down long bracket to avoid missing the closing long bracket "]]" https: //github.com/apache/apisix/pull/4515/checks?check_run_id=2962435093 Signed-off-by: spacewander <spacewanderlzx@gmail.com>
2273 lines
65 KiB
Perl
Vendored
2273 lines
65 KiB
Perl
Vendored
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership.
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
# (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
use t::APISIX 'no_plan';
|
|
|
|
repeat_each(1);
|
|
no_long_string();
|
|
no_root_location();
|
|
no_shuffle();
|
|
|
|
run_tests;
|
|
|
|
__DATA__
|
|
|
|
=== TEST 1: schema validation passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local plugin = require("apisix.plugins.traffic-split")
|
|
local ok, err = plugin.check_schema({
|
|
rules = {
|
|
{
|
|
match = {
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "jack"},
|
|
{"arg_age", "!", "<", "16"}
|
|
}
|
|
},
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "rose"},
|
|
{"arg_age", "!", ">", "32"}
|
|
}
|
|
}
|
|
},
|
|
weighted_upstreams = {
|
|
{
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1981"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
},
|
|
{
|
|
upstream = {
|
|
name = "upstream_B",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1982"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
},
|
|
{
|
|
weight = 1
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if not ok then
|
|
ngx.say(err)
|
|
end
|
|
|
|
ngx.say("done")
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
done
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 2: schema validation passed, and `match` configuration is missing
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local plugin = require("apisix.plugins.traffic-split")
|
|
local ok, err = plugin.check_schema({
|
|
rules = {
|
|
{
|
|
weighted_upstreams = {
|
|
{
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1981"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
},
|
|
{
|
|
weight = 1
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if not ok then
|
|
ngx.say(err)
|
|
end
|
|
|
|
ngx.say("done")
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
done
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 3: schema validation failed, `vars` expression operator type is wrong
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local plugin = require("apisix.plugins.traffic-split")
|
|
local ok, err = plugin.check_schema({
|
|
rules = {
|
|
{
|
|
match = {
|
|
{
|
|
vars = {
|
|
{"arg_name", 123, "jack"}
|
|
}
|
|
}
|
|
},
|
|
weighted_upstreams = {
|
|
{
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1981"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
},
|
|
{
|
|
weight = 1
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if not ok then
|
|
ngx.say(err)
|
|
end
|
|
|
|
ngx.say("done")
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body eval
|
|
qr/failed to validate the 'vars' expression: invalid operator '123'/
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 4: missing `rules` configuration, the upstream of the default `route` takes effect
|
|
--- 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,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 5: the upstream of the default `route` takes effect
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1980, 1980, 1980, 1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 6: when `weighted_upstreams` is empty, the upstream of `route` is used by default
|
|
--- 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,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [{}]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 7: the upstream of the default `route` takes effect
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1980, 1980, 1980, 1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 8: single `vars` expression and single plugin `upstream`, and the upstream traffic on `route` accounts for 1/3
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local json = require("toolkit.json")
|
|
local t = require("lib.test_admin").test
|
|
local data = {
|
|
uri = "/server_port",
|
|
plugins = {
|
|
["traffic-split"] = {
|
|
rules = { {
|
|
match = { {
|
|
vars = { { "arg_name", "==", "jack" }, { "arg_age", "!", "<", "16" } }
|
|
} },
|
|
weighted_upstreams = { {
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["127.0.0.1:1981"] = 2
|
|
},
|
|
timeout = {
|
|
connect = 15,
|
|
send = 15,
|
|
read = 15
|
|
}
|
|
},
|
|
weight = 2
|
|
}, {
|
|
weight = 1
|
|
} }
|
|
} }
|
|
}
|
|
},
|
|
upstream = {
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["127.0.0.1:1980"] = 1
|
|
}
|
|
}
|
|
}
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
json.encode(data)
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 9: expression validation failed, return to the default `route` upstream port `1980`
|
|
--- request
|
|
GET /server_port?name=jack&age=14
|
|
--- response_body eval
|
|
1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 10: the expression passes and initiated multiple requests, the upstream traffic of `route` accounts for 1/3, and the upstream traffic of plugins accounts for 2/3
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port?name=jack&age=16', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1981, 1981, 1981, 1981
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 11: Multiple vars rules and multiple plugin upstream
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{"vars": [["arg_name", "==", "jack"], ["arg_age", "~~", "^[1-9]{1,2}"]]},
|
|
{"vars": [["arg_name2", "in", ["jack", "rose"]], ["arg_age2", "!", "<", 18]]}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": {"127.0.0.1:1981":20}}, "weight": 2},
|
|
{"upstream": {"name": "upstream_B", "type": "roundrobin", "nodes": {"127.0.0.1:1982":10}}, "weight": 2},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 12: expression validation failed, return to the default `route` upstream port `1980`
|
|
--- request
|
|
GET /server_port?name=jack&age=0
|
|
--- response_body eval
|
|
1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 13: the expression passes and initiated multiple requests, the upstream traffic of `route` accounts for 1/5, and the upstream traffic of plugins accounts for 4/5
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?name=jack&age=22', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 14: Multiple vars rules and multiple plugin upstream, do not split traffic to the upstream of `route`
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{"vars": [["arg_name", "==", "jack"], ["arg_age", "~~", "^[1-9]{1,2}"]]},
|
|
{"vars": [["arg_name2", "in", ["jack", "rose"]], ["arg_age2", "!", "<", 18]]}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": {"127.0.0.1:1981":20}}, "weight": 2},
|
|
{"upstream": {"name": "upstream_B", "type": "roundrobin", "nodes": {"127.0.0.1:1982":10}}, "weight": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 15: the expression passes and initiated multiple requests, do not split traffic to the upstream of `route`
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port?name=jack&age=22', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1981, 1981, 1982, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 16: support multiple ip configuration of `nodes`, and missing upstream configuration on `route`
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "jack"], ["arg_age", "~~", "^[1-9]{1,2}"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": {"127.0.0.1:1980":1, "127.0.0.1:1981":2, "127.0.0.1:1982":2}, "timeout": {"connect": 15, "send": 15, "read": 15}}, "weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 17: the expression passes and initiated multiple requests, roundrobin the ip of nodes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?name=jack&age=22', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
|
|
|
|
|
|
=== TEST 18: host is domain name
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local json = require("toolkit.json")
|
|
local t = require("lib.test_admin").test
|
|
local data = {
|
|
uri = "/server_port",
|
|
plugins = {
|
|
["traffic-split"] = {
|
|
rules = { {
|
|
weighted_upstreams = { {
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["foo.com:80"] = 0
|
|
}
|
|
},
|
|
weight = 2
|
|
}, {
|
|
weight = 1
|
|
} }
|
|
} }
|
|
}
|
|
},
|
|
upstream = {
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["127.0.0.1:1980"] = 1
|
|
}
|
|
}
|
|
}
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
json.encode(data)
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 19: domain name resolved successfully
|
|
--- request
|
|
GET /server_port
|
|
--- error_code: 502
|
|
--- error_log eval
|
|
qr/dns resolver domain: foo.com to \d+.\d+.\d+.\d+/
|
|
|
|
|
|
|
|
=== TEST 20: mock Grayscale Release
|
|
--- 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,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{
|
|
"upstream": {
|
|
"name": "upstream_A",
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1981":1
|
|
}
|
|
},
|
|
"weight": 2
|
|
},
|
|
{
|
|
"weight": 1
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 21: 2/3 request traffic hits the upstream of the plugin, 1/3 request traffic hits the upstream of `route`
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1981, 1981, 1981, 1981
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 22: mock Blue-green Release
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local json = require("toolkit.json")
|
|
local t = require("lib.test_admin").test
|
|
local data = {
|
|
uri = "/server_port",
|
|
plugins = {
|
|
["traffic-split"] = {
|
|
rules = { {
|
|
match = { {
|
|
vars = { { "http_release", "==", "blue" } }
|
|
} },
|
|
weighted_upstreams = { {
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["127.0.0.1:1981"] = 1
|
|
}
|
|
}
|
|
} }
|
|
} }
|
|
}
|
|
},
|
|
upstream = {
|
|
type = "roundrobin",
|
|
nodes = {
|
|
["127.0.0.1:1980"] = 1
|
|
}
|
|
}
|
|
}
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
json.encode(data)
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 23: release is equal to `blue`
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["release"] = "blue"
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1981, 1981, 1981, 1981, 1981
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 24: release is equal to `green`
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["release"] = "green"
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1980, 1980, 1980, 1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 25: mock Custom Release
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "jack"], ["arg_age", ">", "23"],["http_appkey", "~~", "[a-z]{1,5}"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": {"127.0.0.1:1981":20}}, "weight": 2},
|
|
{"upstream": {"name": "upstream_B", "type": "roundrobin", "nodes": {"127.0.0.1:1982":10}}, "weight": 2},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 26: `match` rule passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["appkey"] = "api-key"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?name=jack&age=36', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 27: `match` rule failed, `age` condition did not match
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["release"] = "green"
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port?name=jack&age=16', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1980, 1980, 1980, 1980
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 28: upstream nodes are array type and node is the domain name
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"foo.com", "port": 80, "weight": 0}]}, "weight": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 29: domain name resolved successfully
|
|
--- request
|
|
GET /server_port
|
|
--- error_code: 502
|
|
--- error_log eval
|
|
qr/dns resolver domain: foo.com to \d+.\d+.\d+.\d+/
|
|
|
|
|
|
|
|
=== TEST 30: the nodes of upstream are array type, with multiple nodes
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "jack"], ["arg_age", ">", "23"],["http_appkey", "~~", "[a-z]{1,5}"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}, {"host":"127.0.0.1", "port":1982, "weight": 2}]}, "weight": 4},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 31: `match` rule passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["appkey"] = "api-key"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?name=jack&age=36', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 32: the upstream node is an array type and has multiple upstream
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "jack"], ["arg_age", ">", "23"],["http_appkey", "~~", "[a-z]{1,5}"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}]}, "weight": 2},
|
|
{"upstream": {"name": "upstream_B", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1982, "weight": 2}]}, "weight": 2},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 33: `match` rule passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["appkey"] = "api-key"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?name=jack&age=36', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 34: multi-upstream, test with unique upstream 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}]}, "weight": 2},
|
|
{"upstream": {"name": "upstream_B", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1982, "weight": 2}]}, "weight": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 35: the upstream `key` is unique
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 2 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1982
|
|
--- grep_error_log eval
|
|
qr/upstream_key: roundrobin#route_1_\d/
|
|
--- grep_error_log_out eval
|
|
qr/(upstream_key: roundrobin#route_1_1
|
|
upstream_key: roundrobin#route_1_2
|
|
|upstream_key: roundrobin#route_1_2
|
|
upstream_key: roundrobin#route_1_1
|
|
)/
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 36: has empty upstream, test the upstream key is unique
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}]}, "weight": 1},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 37: the upstream `key` is unique
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 2 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981
|
|
--- grep_error_log eval
|
|
qr/upstream_key: roundrobin#route_1_\d/
|
|
--- grep_error_log_out
|
|
upstream_key: roundrobin#route_1_1
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 38: schema validation, "additionalProperties = false" to limit the plugin configuration
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local plugin = require("apisix.plugins.traffic-split")
|
|
local ok, err = plugin.check_schema({
|
|
additional_properties = "hello",
|
|
rules = {
|
|
{
|
|
match = {
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "jack"},
|
|
{"arg_age", "!", "<", "16"}
|
|
}
|
|
},
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "rose"},
|
|
{"arg_age", "!", ">", "32"}
|
|
}
|
|
}
|
|
},
|
|
weighted_upstreams = {
|
|
{
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1981"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if not ok then
|
|
ngx.say(err)
|
|
end
|
|
|
|
ngx.say("done")
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body eval
|
|
qr/additional properties forbidden, found additional_properties/
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 39: schema validation, "additionalProperties = false" to limit the "rules" configuration
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local plugin = require("apisix.plugins.traffic-split")
|
|
local ok, err = plugin.check_schema({
|
|
rules = {
|
|
{
|
|
additional_properties = "hello",
|
|
match = {
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "jack"},
|
|
{"arg_age", "!", "<", "16"}
|
|
}
|
|
},
|
|
{
|
|
vars = {
|
|
{"arg_name", "==", "rose"},
|
|
{"arg_age", "!", ">", "32"}
|
|
}
|
|
}
|
|
},
|
|
weighted_upstreams = {
|
|
{
|
|
upstream = {
|
|
name = "upstream_A",
|
|
type = "roundrobin",
|
|
nodes = {["127.0.0.1:1981"]=2},
|
|
timeout = {connect = 15, send = 15, read = 15}
|
|
},
|
|
weight = 2
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if not ok then
|
|
ngx.say(err)
|
|
end
|
|
|
|
ngx.say("done")
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body eval
|
|
qr/property "rules" validation failed: failed to validate item 1: additional properties forbidden, found additional_properties/
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 40: the request header contains horizontal lines("-")
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["http_x-api-appkey", "==", "api-key"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}, {"host":"127.0.0.1", "port":1982, "weight": 2}]}, "weight": 4},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 41: `match` rule passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["x-api-appkey"] = "api-key"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 42: request args and request headers contain horizontal lines("-")
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_x-api-name", "==", "jack"], ["arg_x-api-age", ">", "23"],["http_x-api-appkey", "~~", "[a-z]{1,5}"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream": {"name": "upstream_A", "type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1981, "weight": 2}, {"host":"127.0.0.1", "port":1982, "weight": 2}]}, "weight": 4},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 43: `match` rule passed
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["x-api-appkey"] = "hello"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?x-api-name=jack&x-api-age=36', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 44: set upstream(id: 1)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/upstreams/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"nodes": {
|
|
"127.0.0.1:1981": 1
|
|
},
|
|
"type": "roundrobin",
|
|
"desc": "new upstream"
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 45: set upstream(id: 2)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/upstreams/2',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"nodes": {
|
|
"127.0.0.1:1982": 1
|
|
},
|
|
"type": "roundrobin",
|
|
"desc": "new upstream"
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 46: set route(id: 1, upstream_id: 1, upstream_id in plugin: 2), and `weighted_upstreams` does not have a structure with only `weight`
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "James"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream_id":"1"
|
|
}]=]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 47: when `match` rule passed, use the `upstream_id` in plugin, and when it failed, use the `upstream_id` in route
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
|
|
for i = 1, 5, 2 do
|
|
-- match rule passed
|
|
local _, _, body = t('/server_port?name=James', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
|
|
-- match rule failed
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i+1] = body
|
|
end
|
|
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1981, 1981, 1982, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 48: set route(use upstream for route and upstream_id for plugin), and `weighted_upstreams` does not have a structure with only `weight`
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_name", "==", "James"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 49: when `match` rule passed, use the `upstream_id` in plugin, and when it failed, use the `upstream` in route
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
|
|
for i = 1, 5, 2 do
|
|
-- match rule passed
|
|
local _, _, body = t('/server_port?name=James', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
|
|
-- match rule failed
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i+1] = body
|
|
end
|
|
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1980, 1981, 1981, 1981
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 50: set route(id: 1, upstream_id: 1, upstream_id in plugin: 2), and `weighted_upstreams` has a structure with only `weight`
|
|
--- 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["uri", "==", "/server_port"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 2, "weight": 1},
|
|
{"weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream_id":"1"
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 51: all requests `match` rule passed, proxy requests to the upstream of route based on the structure with only `weight` in `weighted_upstreams`
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 6 do
|
|
local _, _, body = t('/server_port', ngx.HTTP_GET)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1981, 1981, 1982, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 52: the upstream_id is used in the 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_x-api-name", "==", "jack"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1, "weight": 2},
|
|
{"upstream_id": 2, "weight": 1},
|
|
{"weight": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 53: `match` rule passed(upstream_id)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?x-api-name=jack', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1981, 1981, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 54: only use upstream_id in the 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_x-api-name", "==", "jack"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1, "weight": 1},
|
|
{"upstream_id": 2, "weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 55: `match` rule passed(only use upstream_id)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
for i = 1, 4 do
|
|
local _, _, body = t('/server_port?x-api-name=jack', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1981, 1981, 1982, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 56: use upstream and upstream_id in the 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,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"match": [
|
|
{
|
|
"vars": [["arg_x-api-name", "==", "jack"]]
|
|
}
|
|
],
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1, "weight": 2},
|
|
{"upstream": {"type": "roundrobin", "nodes": [{"host":"127.0.0.1", "port":1982, "weight": 1}]}, "weight": 1},
|
|
{"weight": 2}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 57: `match` rule passed(upstream + upstream_id)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local bodys = {}
|
|
local headers = {}
|
|
headers["x-api-appkey"] = "hello"
|
|
for i = 1, 5 do
|
|
local _, _, body = t('/server_port?x-api-name=jack', ngx.HTTP_GET, "", nil, headers)
|
|
bodys[i] = body
|
|
end
|
|
table.sort(bodys)
|
|
ngx.say(table.concat(bodys, ", "))
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
1980, 1980, 1981, 1981, 1982
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 58: set route + upstream (two upstream node: one healthy + one unhealthy)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/upstreams/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1981": 1,
|
|
"127.0.0.1:1970": 1
|
|
},
|
|
"checks": {
|
|
"active": {
|
|
"http_path": "/status",
|
|
"host": "foo.com",
|
|
"healthy": {
|
|
"interval": 1,
|
|
"successes": 1
|
|
},
|
|
"unhealthy": {
|
|
"interval": 1,
|
|
"http_failures": 2
|
|
}
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
ngx.say(body)
|
|
return
|
|
end
|
|
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[=[{
|
|
"uri": "/server_port",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1, "weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 59: hit routes, ensure the checker is bound to the upstream
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port"
|
|
|
|
do
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET", keepalive = false})
|
|
end
|
|
|
|
ngx.sleep(2.5)
|
|
|
|
local ports_count = {}
|
|
for i = 1, 6 do
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET", keepalive = false})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.port > b.port
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
ngx.exit(200)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
[{"count":6,"port":"1981"}]
|
|
--- grep_error_log eval
|
|
qr/\([^)]+\) unhealthy .* for '.*'/
|
|
--- grep_error_log_out
|
|
(upstream#/apisix/upstreams/1) unhealthy TCP increment (1/2) for 'foo.com(127.0.0.1:1970)'
|
|
(upstream#/apisix/upstreams/1) unhealthy TCP increment (2/2) for 'foo.com(127.0.0.1:1970)'
|
|
--- timeout: 10
|
|
|
|
|
|
|
|
=== TEST 60: set upstream(id: 1), by default retries count = number of nodes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/upstreams/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"nodes": {
|
|
"127.0.0.1:1": 1,
|
|
"127.0.0.2:1": 1,
|
|
"127.0.0.3:1": 1
|
|
},
|
|
"type": "roundrobin"
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 61: set route(id: 1, upstream_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,
|
|
[=[{
|
|
"uri": "/hello",
|
|
"plugins": {
|
|
"traffic-split": {
|
|
"rules": [
|
|
{
|
|
"weighted_upstreams": [
|
|
{"upstream_id": 1, "weight": 1}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"upstream": {
|
|
"type": "roundrobin",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]=]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 62: hit routes
|
|
--- request
|
|
GET /hello
|
|
--- error_code: 502
|
|
--- grep_error_log eval
|
|
qr/\([^)]+\) while connecting to upstream/
|
|
--- grep_error_log_out
|
|
(111: Connection refused) while connecting to upstream
|
|
(111: Connection refused) while connecting to upstream
|
|
(111: Connection refused) while connecting to upstream
|