DevFW-CICD/ingress-nginx-helm
14
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add reset_ngx method to sticky_test.lua

Browse source
This commit is contained in:
Francisco Mejia 2018-08-23 14:09:08 -04:00
parent c7b75970ca
commit 32426b95e6
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
rootfs/etc/nginx/lua/test/balancer/sticky_test.lua
View file

@ -2,12 +2,18 @@ local sticky = require("balancer.sticky")
local cookie = require("resty.cookie") local cookie = require("resty.cookie")
local util = require("util") local util = require("util")
local original_ngx = ngx
function mock_ngx(mock) function mock_ngx(mock)
local _ngx = mock local _ngx = mock
setmetatable(_ngx, {__index = _G.ngx}) setmetatable(_ngx, {__index = _G.ngx})
_G.ngx = _ngx _G.ngx = _ngx
end end
local function reset_ngx()
_G.ngx = original_ngx
end
function get_mocked_cookie_new() function get_mocked_cookie_new()
return function(self) return function(self)
return { return {
@ -17,7 +23,6 @@ function get_mocked_cookie_new()
end end
end end
mock_ngx({ var = {} })
cookie.new = get_mocked_cookie_new() cookie.new = get_mocked_cookie_new()
local function get_test_backend() local function get_test_backend()
@ -34,6 +39,14 @@ local function get_test_backend()
end end
describe("Sticky", function() describe("Sticky", function()
before_each(function()
mock_ngx({ var = {} })
end)
after_each(function()
reset_ngx()
end)
local test_backend = get_test_backend() local test_backend = get_test_backend()
local test_backend_endpoint= test_backend.endpoints[1].address .. ":" .. test_backend.endpoints[1].port local test_backend_endpoint= test_backend.endpoints[1].address .. ":" .. test_backend.endpoints[1].port