wehub-resource-sync
127 lines
3.6 KiB
Lua
127 lines
3.6 KiB
Lua
local helpers = require("spec.helpers")
|
|
|
|
local uuid_pattern = "^" .. ("%x"):rep(8) .. "%-" .. ("%x"):rep(4) .. "%-"
|
|
.. ("%x"):rep(4) .. "%-" .. ("%x"):rep(4) .. "%-"
|
|
.. ("%x"):rep(12) .. "$"
|
|
for _, v in ipairs({ {"off", "off"}, {"on", "off"}, {"on", "on"}, }) do
|
|
local rpc, rpc_sync = v[1], v[2]
|
|
|
|
for _, strategy in helpers.each_strategy() do
|
|
describe("PDK: kong.cluster for #" .. strategy .. " rpc_sync=" .. rpc_sync, function()
|
|
local proxy_client
|
|
|
|
local CP_MOCK_PORT
|
|
local DP_MOCK_PORT
|
|
|
|
lazy_setup(function()
|
|
CP_MOCK_PORT = helpers.get_available_port()
|
|
DP_MOCK_PORT = helpers.get_available_port()
|
|
|
|
local fixtures_dp = {
|
|
http_mock = {
|
|
my_server_block = [[
|
|
server {
|
|
server_name my_server;
|
|
listen ]] .. DP_MOCK_PORT .. [[;
|
|
|
|
location = "/hello" {
|
|
content_by_lua_block {
|
|
ngx.print(kong.cluster.get_id())
|
|
}
|
|
}
|
|
}
|
|
]]
|
|
},
|
|
}
|
|
|
|
local fixtures_cp = {
|
|
http_mock = {
|
|
my_server_block = [[
|
|
server {
|
|
server_name my_server;
|
|
listen ]] .. CP_MOCK_PORT .. [[;
|
|
|
|
location = "/hello" {
|
|
content_by_lua_block {
|
|
ngx.print(kong.cluster.get_id())
|
|
}
|
|
}
|
|
}
|
|
]]
|
|
},
|
|
}
|
|
|
|
assert(helpers.get_db_utils(strategy, {
|
|
"plugins",
|
|
"routes",
|
|
"services",
|
|
"upstreams",
|
|
"targets",
|
|
}))
|
|
|
|
assert(helpers.start_kong({
|
|
role = "control_plane",
|
|
cluster_cert = "spec/fixtures/kong_clustering.crt",
|
|
cluster_cert_key = "spec/fixtures/kong_clustering.key",
|
|
database = strategy,
|
|
db_update_frequency = 0.1,
|
|
cluster_listen = "127.0.0.1:9005",
|
|
nginx_conf = "spec/fixtures/custom_nginx.template",
|
|
cluster_rpc = rpc,
|
|
cluster_rpc_sync = rpc_sync,
|
|
}, nil, nil, fixtures_cp))
|
|
|
|
assert(helpers.start_kong({
|
|
role = "data_plane",
|
|
database = "off",
|
|
prefix = "servroot2",
|
|
cluster_cert = "spec/fixtures/kong_clustering.crt",
|
|
cluster_cert_key = "spec/fixtures/kong_clustering.key",
|
|
cluster_control_plane = "127.0.0.1:9005",
|
|
proxy_listen = "0.0.0.0:9002",
|
|
nginx_conf = "spec/fixtures/custom_nginx.template",
|
|
cluster_rpc = rpc,
|
|
cluster_rpc_sync = rpc_sync,
|
|
}, nil, nil, fixtures_dp))
|
|
|
|
if rpc_sync == "on" then
|
|
assert.logfile("servroot2/logs/error.log").has.line("[kong.sync.v2] full sync ends", true, 10)
|
|
end
|
|
end)
|
|
|
|
lazy_teardown(function()
|
|
if proxy_client then
|
|
proxy_client:close()
|
|
end
|
|
|
|
helpers.stop_kong("servroot2")
|
|
helpers.stop_kong()
|
|
end)
|
|
|
|
it("kong.cluster.get_id() in Hybrid mode", function()
|
|
proxy_client = helpers.http_client(helpers.get_proxy_ip(false), CP_MOCK_PORT)
|
|
|
|
local res = proxy_client:get("/hello")
|
|
local cp_cluster_id = assert.response(res).has_status(200)
|
|
|
|
assert.match(uuid_pattern, cp_cluster_id)
|
|
|
|
proxy_client:close()
|
|
|
|
helpers.wait_until(function()
|
|
proxy_client = helpers.http_client(helpers.get_proxy_ip(false), DP_MOCK_PORT)
|
|
local res = proxy_client:get("/hello")
|
|
local body = assert.response(res).has_status(200)
|
|
proxy_client:close()
|
|
|
|
if string.match(body, uuid_pattern) then
|
|
if cp_cluster_id == body then
|
|
return true
|
|
end
|
|
end
|
|
end, 10)
|
|
end)
|
|
end)
|
|
end -- for _, strategy
|
|
end -- for rpc_sync
|