From 290a57291f92acc65251b47f261414838089c1f2 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Sun, 17 Nov 2024 16:54:11 -0800 Subject: [PATCH] fix int tests --- crates/llm_gateway/tests/integration.rs | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/crates/llm_gateway/tests/integration.rs b/crates/llm_gateway/tests/integration.rs index a40389aa..c39debd6 100644 --- a/crates/llm_gateway/tests/integration.rs +++ b/crates/llm_gateway/tests/integration.rs @@ -217,6 +217,8 @@ fn llm_gateway_successful_request_to_open_ai_chat_completions() { ) .expect_get_buffer_bytes(Some(BufferType::HttpRequestBody)) .returning(Some(chat_completions_request_body)) + .expect_get_current_time_nanos() + .returning(Some(0)) .expect_log(Some(LogLevel::Trace), None) .expect_log(Some(LogLevel::Debug), None) .expect_metric_record("input_sequence_length", 21) @@ -279,6 +281,8 @@ fn llm_gateway_bad_request_to_open_ai_chat_completions() { ) .expect_get_buffer_bytes(Some(BufferType::HttpRequestBody)) .returning(Some(incomplete_chat_completions_request_body)) + .expect_get_current_time_nanos() + .returning(Some(0)) .expect_log(Some(LogLevel::Debug), None) .expect_send_local_response( Some(StatusCode::BAD_REQUEST.as_u16().into()), @@ -337,6 +341,8 @@ fn llm_gateway_request_ratelimited() { ) .expect_get_buffer_bytes(Some(BufferType::HttpRequestBody)) .returning(Some(chat_completions_request_body)) + .expect_get_current_time_nanos() + .returning(Some(0)) // The actual call is not important in this test, we just need to grab the token_id .expect_log(Some(LogLevel::Trace), None) .expect_log(Some(LogLevel::Debug), None) @@ -403,6 +409,8 @@ fn llm_gateway_request_not_ratelimited() { ) .expect_get_buffer_bytes(Some(BufferType::HttpRequestBody)) .returning(Some(chat_completions_request_body)) + .expect_get_current_time_nanos() + .returning(Some(0)) // The actual call is not important in this test, we just need to grab the token_id .expect_log(Some(LogLevel::Trace), None) .expect_log(Some(LogLevel::Debug), None)