From 0b70319e687d703ae146464131ec91ebd5efd347 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Tue, 28 Jan 2025 12:30:24 -0800 Subject: [PATCH] fix unit tests --- crates/llm_gateway/tests/integration.rs | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/crates/llm_gateway/tests/integration.rs b/crates/llm_gateway/tests/integration.rs index 7a74dfa8..73c431bf 100644 --- a/crates/llm_gateway/tests/integration.rs +++ b/crates/llm_gateway/tests/integration.rs @@ -18,6 +18,14 @@ fn wasm_module() -> String { fn request_headers_expectations(module: &mut Tester, http_context: i32) { module .call_proxy_on_request_headers(http_context, 0, false) + .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some(":path")) + .returning(Some("/test_path")) + .expect_log(Some(LogLevel::Debug), Some("request_path: /test_path")) + .expect_get_header_map_value( + Some(MapType::HttpRequestHeaders), + Some("x-arch-llm-provider-hint"), + ) + .returning(Some("default")) .expect_get_header_map_value( Some(MapType::HttpRequestHeaders), Some("x-arch-llm-provider-hint"), @@ -25,9 +33,8 @@ fn request_headers_expectations(module: &mut Tester, http_context: i32) { .returning(Some("default")) .expect_log( Some(LogLevel::Debug), - Some("llm provider hint: Some(Default)"), + Some("llm provider hint: Some(\"default\"), selected llm: open-ai-gpt-4"), ) - .expect_log(Some(LogLevel::Debug), Some("selected llm: open-ai-gpt-4")) .expect_add_header_map_value( Some(MapType::HttpRequestHeaders), Some("x-arch-llm-provider"), @@ -50,11 +57,11 @@ fn request_headers_expectations(module: &mut Tester, http_context: i32) { .returning(None) .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some(":path")) .returning(Some("/v1/chat/completions")) - .expect_log(Some(LogLevel::Debug), None) + // .expect_log(Some(LogLevel::Debug), None) .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("x-request-id")) .returning(None) - .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("traceparent")) - .returning(None) + // .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("traceparent")) + // .returning(None) .execute_and_expect(ReturnType::Action(Action::Continue)) .unwrap(); } @@ -187,7 +194,6 @@ fn llm_gateway_successful_request_to_open_ai_chat_completions() { module .call_proxy_on_context_create(http_context, filter_context) - .expect_log(Some(LogLevel::Debug), None) .execute_and_expect(ReturnType::None) .unwrap();