# # 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'; log_level('debug'); repeat_each(1); no_long_string(); no_root_location(); run_tests; __DATA__ === TEST 1: sanity --- config location /t { content_by_lua_block { local plugin = require("apisix.plugins.request-id") local ok, err = plugin.check_schema({}) if not ok then ngx.say(err) end ngx.say("done") } } --- request GET /t --- response_body done --- no_error_log [error] === TEST 2: wrong type --- config location /t { content_by_lua_block { local plugin = require("apisix.plugins.request-id") local ok, err = plugin.check_schema({include_in_response = "bad_type"}) if not ok then ngx.say(err) end ngx.say("done") } } --- request GET /t --- response_body property "include_in_response" validation failed: wrong type: expected boolean, got string done --- no_error_log [error] === TEST 3: add plugin with include_in_response true (default true) --- 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": { "request-id": { } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }]], [[{ "node": { "value": { "plugins": { "request-id": { } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }, "key": "/apisix/routes/1" }, "action": "set" }]] ) if code >= 300 then ngx.status = code end ngx.say(body) } } --- request GET /t --- response_body passed --- no_error_log [error] === TEST 4: check for request id in response header (default header name) --- config location /t { content_by_lua_block { local http = require "resty.http" local httpc = http.new() local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/opentracing" local res, err = httpc:request_uri(uri, { method = "GET", headers = { ["Content-Type"] = "application/json", } }) if res.headers["X-Request-Id"] then ngx.say("request header present") else ngx.say("failed") end } } --- request GET /t --- response_body request header present --- no_error_log [error] === TEST 5: check for unique id --- config location /t { content_by_lua_block { local http = require "resty.http" local httpc = http.new() local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/opentracing" local res1, err1 = httpc:request_uri(uri, { method = "GET", headers = { ["Content-Type"] = "application/json", } } ) local res2, err2 = httpc:request_uri(uri, { method = "GET", headers = { ["Content-Type"] = "application/json", } } ) -- ngx.say("res1: ", res1.headers["X-Request-Id"]) -- ngx.say("res2: ", res2.headers["X-Request-Id"]) if res1.headers["X-Request-Id"] == res2.headers["X-Request-Id"] then ngx.say("ids not unique") else ngx.say("true") end } } --- request GET /t --- response_body true --- no_error_log [error] === TEST 6: add plugin with custom header 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, [[{ "plugins": { "request-id": { "header_name": "Custom-Header-Name" } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }]], [[{ "node": { "value": { "plugins": { "request-id": { "header_name": "Custom-Header-Name", "include_in_response": true } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }, "key": "/apisix/routes/1" }, "action": "set" }]] ) if code >= 300 then ngx.status = code end ngx.say(body) } } --- request GET /t --- response_body passed --- no_error_log [error] === TEST 7: check for request id in response header (custom header name) --- config location /t { content_by_lua_block { local http = require "resty.http" local httpc = http.new() local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/opentracing" local res, err = httpc:request_uri(uri, { method = "GET", headers = { ["Content-Type"] = "application/json", } }) if res.headers["Custom-Header-Name"] then ngx.say("request header present") else ngx.say("failed") end } } --- request GET /t --- response_body request header present --- no_error_log [error] === TEST 8: add plugin with include_in_response false (default true) --- 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": { "request-id": { "include_in_response": false } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }]], [[{ "node": { "value": { "plugins": { "request-id": { "include_in_response": false } }, "upstream": { "nodes": { "127.0.0.1:1982": 1 }, "type": "roundrobin" }, "uri": "/opentracing" }, "key": "/apisix/routes/1" }, "action": "set" }]] ) if code >= 300 then ngx.status = code end ngx.say(body) } } --- request GET /t --- response_body passed --- no_error_log [error] === TEST 9: check for request id is not present in the response header --- config location /t { content_by_lua_block { local http = require "resty.http" local httpc = http.new() local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/opentracing" local res, err = httpc:request_uri(uri, { method = "GET", headers = { ["Content-Type"] = "application/json", } }) if not res.headers["X-Request-Id"] then ngx.say("request header not present") else ngx.say("failed") end } } --- request GET /t --- response_body request header not present --- no_error_log [error]