tests(basic-auth): fix flaky on invalidations spec (#13890)

Kong · Nov 19, 2024 · 7c17e3b · 7c17e3b · github-actions · Nov 19, 2024
1 parent 3140f81
commit 7c17e3b
Showing 1 changed file with 39 additions and 30 deletions.
diff --git a/spec/03-plugins/10-basic-auth/04-invalidations_spec.lua b/spec/03-plugins/10-basic-auth/04-invalidations_spec.lua
@@ -72,16 +72,19 @@ for _, strategy in helpers.each_strategy() do
     end)
 
     it("#invalidates credentials when the Consumer is deleted", function()
-      -- populate cache
-      local res = assert(proxy_client:send {
-        method  = "GET",
-        path    = "/",
-        headers = {
-          ["Authorization"] = "Basic Ym9iOmtvbmc=",
-          ["Host"]          = "basic-auth.test"
-        }
-      })
-      assert.res_status(200, res)
+      local res
+      helpers.pwait_until(function()
+        -- populate cache
+        res = assert(proxy_client:send {
+          method  = "GET",
+          path    = "/",
+          headers = {
+            ["Authorization"] = "Basic Ym9iOmtvbmc=",
+            ["Host"]          = "basic-auth.test"
+          }
+        })
+        assert.res_status(200, res)
+      end)
 
       -- ensure cache is populated
       local cache_key = db.basicauth_credentials:cache_key("bob")
@@ -115,16 +118,19 @@ for _, strategy in helpers.each_strategy() do
     end)
 
     it("invalidates credentials from cache when deleted", function()
-      -- populate cache
-      local res = assert(proxy_client:send {
-        method  = "GET",
-        path    = "/",
-        headers = {
-          ["Authorization"] = "Basic Ym9iOmtvbmc=",
-          ["Host"]          = "basic-auth.test"
-        }
-      })
-      assert.res_status(200, res)
+      local res
+      helpers.pwait_until(function()
+        -- populate cache
+        res = assert(proxy_client:send {
+          method  = "GET",
+          path    = "/",
+          headers = {
+            ["Authorization"] = "Basic Ym9iOmtvbmc=",
+            ["Host"]          = "basic-auth.test"
+          }
+        })
+        assert.res_status(200, res)
+      end)
 
       -- ensure cache is populated
       local cache_key = db.basicauth_credentials:cache_key("bob")
@@ -158,16 +164,19 @@ for _, strategy in helpers.each_strategy() do
     end)
 
     it("invalidated credentials from cache when updated", function()
-      -- populate cache
-      local res = assert(proxy_client:send {
-        method  = "GET",
-        path    = "/",
-        headers = {
-          ["Authorization"] = "Basic Ym9iOmtvbmc=",
-          ["Host"]          = "basic-auth.test"
-        }
-      })
-      assert.res_status(200, res)
+      local res
+      helpers.pwait_until(function()
+        -- populate cache
+        res = assert(proxy_client:send {
+          method  = "GET",
+          path    = "/",
+          headers = {
+            ["Authorization"] = "Basic Ym9iOmtvbmc=",
+            ["Host"]          = "basic-auth.test"
+          }
+        })
+        assert.res_status(200, res)
+      end)
 
       -- ensure cache is populated
       local cache_key = db.basicauth_credentials:cache_key("bob")