tests(ai-plugins): fix flaky latency assertion (part 2)

Applies the same fix as 58c580f to similar tests.
Kong · Aug 23, 2024 · b70c138 · b70c138
1 parent 846f222
commit b70c138
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/spec/03-plugins/38-ai-proxy/02-openai_integration_spec.lua b/spec/03-plugins/38-ai-proxy/02-openai_integration_spec.lua
@@ -833,7 +833,7 @@ for _, strategy in helpers.all_strategies() do if strategy ~= "cassandra" then
         first_got.usage.time_per_token = 1
 
         assert.same(first_expected, first_got)
-        assert.is_true(actual_llm_latency > 0)
+        assert.is_true(actual_llm_latency >= 0)
         assert.same(actual_time_per_token, time_per_token)
       end)
 
@@ -1409,7 +1409,7 @@ for _, strategy in helpers.all_strategies() do if strategy ~= "cassandra" then
           },
           body = pl_file.read("spec/fixtures/ai-proxy/openai/llm-v1-chat/requests/good_multi_modal.json"),
         })
-        
+
         -- validate that the request succeeded, response status 200
         local body = assert.res_status(200 , r)
         local json = cjson.decode(body)
@@ -1444,7 +1444,7 @@ for _, strategy in helpers.all_strategies() do if strategy ~= "cassandra" then
         first_got.usage.time_per_token = 1
 
         assert.same(first_expected, first_got)
-        assert.is_true(actual_llm_latency > 0)
+        assert.is_true(actual_llm_latency >= 0)
         assert.same(actual_time_per_token, time_per_token)
       end)
 

diff --git a/spec/03-plugins/39-ai-request-transformer/02-integration_spec.lua b/spec/03-plugins/39-ai-request-transformer/02-integration_spec.lua
@@ -307,7 +307,7 @@ for _, strategy in helpers.all_strategies() do if strategy ~= "cassandra" then
         log_message.ai["ai-request-transformer"].usage.time_per_token = 1
 
         assert.same(_EXPECTED_CHAT_STATS, log_message.ai)
-        assert.is_true(actual_llm_latency > 0)
+        assert.is_true(actual_llm_latency >= 0)
         assert.same(actual_time_per_token, time_per_token)
       end)