From db202578d3f54f8d857da457703efb7750a90b52 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Tue, 15 Oct 2024 18:53:18 -0700
Subject: [PATCH] fix int and unit tests

---
 crates/llm_gateway/tests/integration.rs    | 2 +-
 crates/prompt_gateway/tests/integration.rs | 2 +-
 crates/public_types/src/configuration.rs   | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/crates/llm_gateway/tests/integration.rs b/crates/llm_gateway/tests/integration.rs
index c628d9c3..5bc76c66 100644
--- a/crates/llm_gateway/tests/integration.rs
+++ b/crates/llm_gateway/tests/integration.rs
@@ -17,7 +17,7 @@ use std::collections::HashMap;
 use std::path::Path;
 
 fn wasm_module() -> String {
-    let wasm_file = Path::new("target/wasm32-wasi/release/intelligent_prompt_gateway.wasm");
+    let wasm_file = Path::new("target/wasm32-wasi/release/llm_gateway.wasm");
     assert!(
         wasm_file.exists(),
         "Run `cargo build --release --target=wasm32-wasi` first"
diff --git a/crates/prompt_gateway/tests/integration.rs b/crates/prompt_gateway/tests/integration.rs
index c628d9c3..49f8576b 100644
--- a/crates/prompt_gateway/tests/integration.rs
+++ b/crates/prompt_gateway/tests/integration.rs
@@ -17,7 +17,7 @@ use std::collections::HashMap;
 use std::path::Path;
 
 fn wasm_module() -> String {
-    let wasm_file = Path::new("target/wasm32-wasi/release/intelligent_prompt_gateway.wasm");
+    let wasm_file = Path::new("target/wasm32-wasi/release/prompt_gateway.wasm");
     assert!(
         wasm_file.exists(),
         "Run `cargo build --release --target=wasm32-wasi` first"
diff --git a/crates/public_types/src/configuration.rs b/crates/public_types/src/configuration.rs
index 74f58ab0..c8bb72de 100644
--- a/crates/public_types/src/configuration.rs
+++ b/crates/public_types/src/configuration.rs
@@ -226,7 +226,7 @@ mod test {
     #[test]
     fn test_deserialize_configuration() {
         let ref_config =
-            fs::read_to_string("../docs/source/resources/includes/arch_config_full_reference.yaml")
+            fs::read_to_string("../../docs/source/resources/includes/arch_config_full_reference.yaml")
                 .expect("reference config file not found");
 
         let config: super::Configuration = serde_yaml::from_str(&ref_config).unwrap();