From 32426b95e676d3e7b7b1a6e3f8216364f5d0d068 Mon Sep 17 00:00:00 2001 From: Francisco Mejia Date: Thu, 23 Aug 2018 14:09:08 -0400 Subject: [PATCH] Add reset_ngx method to sticky_test.lua --- .../etc/nginx/lua/test/balancer/sticky_test.lua | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/rootfs/etc/nginx/lua/test/balancer/sticky_test.lua b/rootfs/etc/nginx/lua/test/balancer/sticky_test.lua index d6b89fae0..9346d6573 100644 --- a/rootfs/etc/nginx/lua/test/balancer/sticky_test.lua +++ b/rootfs/etc/nginx/lua/test/balancer/sticky_test.lua @@ -2,12 +2,18 @@ local sticky = require("balancer.sticky") local cookie = require("resty.cookie") local util = require("util") +local original_ngx = ngx + function mock_ngx(mock) local _ngx = mock setmetatable(_ngx, {__index = _G.ngx}) _G.ngx = _ngx end +local function reset_ngx() + _G.ngx = original_ngx +end + function get_mocked_cookie_new() return function(self) return { @@ -17,7 +23,6 @@ function get_mocked_cookie_new() end end -mock_ngx({ var = {} }) cookie.new = get_mocked_cookie_new() local function get_test_backend() @@ -34,6 +39,14 @@ local function get_test_backend() end 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_endpoint= test_backend.endpoints[1].address .. ":" .. test_backend.endpoints[1].port