Tests fixes

spiceai · Jan 5, 2024 · 2eedd7a · 2eedd7a
1 parent c316b0b
commit 2eedd7a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/tests/client_test.rs b/tests/client_test.rs
@@ -8,7 +8,9 @@ mod tests {
     async fn new_client() -> Client {
         dotenv::from_path(Path::new(".env.local")).ok();
         let api_key = env::var("API_KEY").expect("API_KEY not found");
-        Client::new(&api_key).await
+        Client::new(&api_key)
+            .await
+            .expect("Failed to create client")
     }
 
     #[tokio::test]

diff --git a/tests/price_test.rs b/tests/price_test.rs
@@ -8,7 +8,9 @@ mod tests {
     async fn new_client() -> Client {
         dotenv::from_path(Path::new(".env.local")).ok();
         let api_key = env::var("API_KEY").expect("API_KEY not found");
-        Client::new(&api_key).await
+        Client::new(&api_key)
+            .await
+            .expect("Failed to create client")
     }
 
     #[tokio::test]

diff --git a/tests/readme_test.rs b/tests/readme_test.rs
@@ -12,7 +12,7 @@ mod tests {
         dotenv::from_path(Path::new(".env.local")).ok();
         let api_key = env::var("API_KEY").expect("API_KEY not found");
 
-        let mut client = Client::new(&api_key).await;
+        let mut client = Client::new(&api_key).await.unwrap();
         let data = client
             .query("SELECT * FROM eth.recent_blocks LIMIT 10;")
             .await;