test.lua

@@ -36,6 +36,19 @@ local function createMockSharedDict()
     }
 end
 
+-- Create a proxy table that dynamically creates mock dictionaries on access
+local function createSharedProxy()
+    local dicts = {}
+    return setmetatable({}, {
+        __index = function(_, key)
+            if not dicts[key] then
+                dicts[key] = createMockSharedDict()
+            end
+            return dicts[key]
+        end
+    })
+end
+
 function createNgx()
     local ngx = {
         status = nil
@@ -79,11 +92,8 @@ function createNgx()
     ngx.re.match = rex.match
     ngx.re.search = rex.find
 
-    -- shared memory dictionaries for testing
-    ngx.shared = {
-        aproxy_bans = createMockSharedDict(),
-        aproxy_tokens = createMockSharedDict()
-    }
+    -- shared memory dictionaries for testing (dynamically created on access)
+    ngx.shared = createSharedProxy()
 
     -- time function for testing
     ngx.time = function() return 1000000 end