wehub-resource-sync
293 lines
11 KiB
Lua
293 lines
11 KiB
Lua
local helpers = require "spec.helpers"
|
|
local cjson = require "cjson"
|
|
|
|
|
|
local TEST_NAME_HEADER = "X-PW-Test"
|
|
local TESTS_FILTER_FILE = nil -- helpers.test_conf.wasm_filters_path .. "/tests.wasm"
|
|
|
|
local fixtures = {
|
|
dns_mock = helpers.dns_mock.new({
|
|
mocks_only = true
|
|
}),
|
|
http_mock = {},
|
|
stream_mock = {}
|
|
}
|
|
|
|
fixtures.dns_mock:A({
|
|
name = "mock.io",
|
|
address = "127.0.0.1"
|
|
})
|
|
|
|
fixtures.dns_mock:A({
|
|
name = "status.io",
|
|
address = "127.0.0.1"
|
|
})
|
|
|
|
|
|
local function add_service_and_route(bp, name, path)
|
|
local service = assert(bp.services:insert({
|
|
name = name,
|
|
url = helpers.mock_upstream_url,
|
|
}))
|
|
|
|
local route = assert(bp.routes:insert({
|
|
name = name .. "-route",
|
|
service = { id = service.id },
|
|
paths = { path },
|
|
hosts = { name },
|
|
protocols = { "https" },
|
|
}))
|
|
|
|
return service, route
|
|
end
|
|
|
|
|
|
local function add_filter_to_service(bp, filter_name, service)
|
|
local filters = {
|
|
{ name = filter_name, enabled = true, config = {} },
|
|
}
|
|
|
|
assert(bp.filter_chains:insert({
|
|
service = { id = service.id }, filters = filters,
|
|
}))
|
|
end
|
|
|
|
|
|
for _, strategy in helpers.each_strategy() do
|
|
-- TODO: replace these test cases with ones that assert the proper behavior
|
|
-- after the feature is removed
|
|
pending("Plugin: prometheus (metrics) [#" .. strategy .. "]", function()
|
|
local admin_client
|
|
|
|
lazy_setup(function()
|
|
local filter_dir = helpers.make_temp_dir()
|
|
local filter_file = filter_dir .. "/tests.wasm"
|
|
local status_api_port = helpers.get_available_port()
|
|
|
|
-- copy filter to a custom location to avoid filter metadata collision
|
|
assert(helpers.file.copy(TESTS_FILTER_FILE, filter_file))
|
|
assert(helpers.file.write(filter_dir .. "/tests.meta.json", cjson.encode({
|
|
config_schema = { type = "object", properties = {} },
|
|
metrics = {
|
|
label_patterns = {
|
|
{ label = "service", pattern = "(_s_id=([0-9a-z%-]+))" },
|
|
{ label = "route", pattern = "(_r_id=([0-9a-z%-]+))" },
|
|
}
|
|
}
|
|
})))
|
|
|
|
require("kong.runloop.wasm").enable({
|
|
{ name = "tests", path = filter_file },
|
|
})
|
|
|
|
local bp = helpers.get_db_utils(strategy, {
|
|
"services", "routes", "plugins", "filter_chains",
|
|
})
|
|
|
|
local service, _ = add_service_and_route(bp, "mock", "/")
|
|
local service2, _ = add_service_and_route(bp, "mock2", "/v2")
|
|
|
|
add_service_and_route(bp, "status.io", "/metrics")
|
|
|
|
add_filter_to_service(bp, "tests", service)
|
|
add_filter_to_service(bp, "tests", service2)
|
|
|
|
assert(bp.plugins:insert({
|
|
name = "prometheus",
|
|
config = { wasm_metrics = true },
|
|
}))
|
|
|
|
assert(helpers.start_kong({
|
|
nginx_conf = "spec/fixtures/custom_nginx.template",
|
|
wasm = true,
|
|
wasm_filters_path = filter_dir,
|
|
plugins = "bundled,prometheus",
|
|
status_listen = '127.0.0.1:' .. status_api_port .. ' ssl',
|
|
status_access_log = "logs/status_access.log",
|
|
status_error_log = "logs/status_error.log"
|
|
}, nil, nil, fixtures))
|
|
|
|
local proxy_client = helpers.proxy_ssl_client()
|
|
|
|
local res = proxy_client:get("/", {
|
|
headers = { host = "mock", [TEST_NAME_HEADER] = "update_metrics" },
|
|
})
|
|
assert.res_status(200, res)
|
|
|
|
res = proxy_client:get("/v2", {
|
|
headers = { host = "mock2", [TEST_NAME_HEADER] = "update_metrics" },
|
|
})
|
|
assert.res_status(200, res)
|
|
|
|
proxy_client:close()
|
|
end)
|
|
|
|
lazy_teardown(function()
|
|
if admin_client then
|
|
admin_client:close()
|
|
end
|
|
|
|
helpers.stop_kong()
|
|
end)
|
|
|
|
before_each(function()
|
|
admin_client = helpers.admin_client()
|
|
end)
|
|
|
|
after_each(function()
|
|
if admin_client then
|
|
admin_client:close()
|
|
end
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm counters", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
local expected_c = '# HELP pw_tests_a_counter\n'
|
|
.. '# TYPE pw_tests_a_counter counter\n'
|
|
.. 'pw_tests_a_counter 2'
|
|
|
|
assert.matches(expected_c, body, nil, true)
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm labeled counters", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
|
|
local expected_c = '# HELP pw_tests_a_labeled_counter\n'
|
|
.. '# TYPE pw_tests_a_labeled_counter counter\n'
|
|
.. 'pw_tests_a_labeled_counter{service="mock2",route="mock2-route"} 1\n'
|
|
.. 'pw_tests_a_labeled_counter{service="mock",route="mock-route"} 1'
|
|
|
|
assert.matches(expected_c, body, nil, true)
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm gauges", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
|
|
local expected_g = '# HELP pw_tests_a_gauge\n'
|
|
.. '# TYPE pw_tests_a_gauge gauge\n'
|
|
.. 'pw_tests_a_gauge 1'
|
|
|
|
assert.matches(expected_g, body, nil, true)
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm labeled gauges", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
|
|
local expected_g = '# HELP pw_tests_a_labeled_gauge\n'
|
|
.. '# TYPE pw_tests_a_labeled_gauge gauge\n'
|
|
.. 'pw_tests_a_labeled_gauge{service="mock2",route="mock2-route"} 1\n'
|
|
.. 'pw_tests_a_labeled_gauge{service="mock",route="mock-route"} 1'
|
|
|
|
assert.matches(expected_g, body, nil, true)
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm histograms", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
|
|
local expected_h = '# HELP pw_tests_a_histogram\n'
|
|
.. '# TYPE pw_tests_a_histogram histogram\n'
|
|
.. 'pw_tests_a_histogram{le="1"} 2\n'
|
|
.. 'pw_tests_a_histogram{le="2"} 4\n'
|
|
.. 'pw_tests_a_histogram{le="4"} 6\n'
|
|
.. 'pw_tests_a_histogram{le="8"} 8\n'
|
|
.. 'pw_tests_a_histogram{le="16"} 10\n'
|
|
.. 'pw_tests_a_histogram{le="32"} 12\n'
|
|
.. 'pw_tests_a_histogram{le="64"} 14\n'
|
|
.. 'pw_tests_a_histogram{le="128"} 16\n'
|
|
.. 'pw_tests_a_histogram{le="256"} 18\n'
|
|
.. 'pw_tests_a_histogram{le="512"} 20\n'
|
|
.. 'pw_tests_a_histogram{le="1024"} 22\n'
|
|
.. 'pw_tests_a_histogram{le="2048"} 24\n'
|
|
.. 'pw_tests_a_histogram{le="4096"} 26\n'
|
|
.. 'pw_tests_a_histogram{le="8192"} 28\n'
|
|
.. 'pw_tests_a_histogram{le="16384"} 30\n'
|
|
.. 'pw_tests_a_histogram{le="32768"} 32\n'
|
|
.. 'pw_tests_a_histogram{le="65536"} 34\n'
|
|
.. 'pw_tests_a_histogram{le="+Inf"} 36\n'
|
|
.. 'pw_tests_a_histogram_sum 524286\n'
|
|
.. 'pw_tests_a_histogram_count 36'
|
|
|
|
assert.matches(expected_h, body, nil, true)
|
|
end)
|
|
|
|
it("exposes Proxy-Wasm labeled histograms", function()
|
|
local res = assert(admin_client:send{
|
|
method = "GET",
|
|
path = "/metrics"
|
|
})
|
|
|
|
local body = assert.res_status(200, res)
|
|
|
|
local expected_h = '# HELP pw_tests_a_labeled_histogram\n'
|
|
.. '# TYPE pw_tests_a_labeled_histogram histogram\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="1"} 1\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="2"} 2\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="4"} 3\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="8"} 4\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="16"} 5\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="32"} 6\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="64"} 7\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="128"} 8\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="256"} 9\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="512"} 10\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="1024"} 11\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="2048"} 12\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="4096"} 13\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="8192"} 14\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="16384"} 15\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="32768"} 16\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="65536"} 17\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock2",route="mock2-route",le="+Inf"} 18\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="1"} 1\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="2"} 2\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="4"} 3\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="8"} 4\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="16"} 5\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="32"} 6\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="64"} 7\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="128"} 8\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="256"} 9\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="512"} 10\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="1024"} 11\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="2048"} 12\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="4096"} 13\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="8192"} 14\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="16384"} 15\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="32768"} 16\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="65536"} 17\n'
|
|
.. 'pw_tests_a_labeled_histogram{service="mock",route="mock-route",le="+Inf"} 18\n'
|
|
.. 'pw_tests_a_labeled_histogram_sum{service="mock2",route="mock2-route"} 262143\n'
|
|
.. 'pw_tests_a_labeled_histogram_sum{service="mock",route="mock-route"} 262143\n'
|
|
.. 'pw_tests_a_labeled_histogram_count{service="mock2",route="mock2-route"} 18\n'
|
|
.. 'pw_tests_a_labeled_histogram_count{service="mock",route="mock-route"} 18'
|
|
|
|
assert.matches(expected_h, body, nil, true)
|
|
end)
|
|
end)
|
|
end
|