tests(helpers): supporting optional Redis password

Kong · Sep 11, 2024 · 3e05739 · 3e05739
1 parent c937edc
commit 3e05739
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/spec/helpers/redis_helper.lua b/spec/helpers/redis_helper.lua
@@ -4,16 +4,19 @@ local version = require "version"
 local DEFAULT_TIMEOUT = 2000
 
 
-local function connect(host, port)
+local function connect(host, port, password)
   local redis_client = redis:new()
   redis_client:set_timeout(DEFAULT_TIMEOUT)
   assert(redis_client:connect(host, port))
+  if password then
+    assert(redis_client:auth(password))
+  end
   local red_version = string.match(redis_client:info(), 'redis_version:([%g]+)\r\n')
   return redis_client, assert(version(red_version))
 end
 
-local function reset_redis(host, port)
-  local redis_client = connect(host, port)
+local function reset_redis(host, port, password)
+  local redis_client = connect(host, port, password)
   redis_client:flushall()
   redis_client:close()
 end