Rename get_latest_prices() to get_prices()

Modify get_prices()
Expose new_spice_client method()
Add tests for get_prices and get_historical_prices

Author: Sevenannn

src/lib.rs

@@ -6,5 +6,6 @@ mod tls;
 
 pub use client::{SpiceClient as Client};
 pub use flight::{SqlFlightClient};
+pub use client::new_spice_client;
 
 // Further public exports and integrations

src/prices.rs

@@ -117,10 +117,13 @@ impl PricesClient {
         Ok(response)
     }
 
-    pub async fn get_latest_prices(&self, pairs: &[&str]) -> Result<LatestPricesResponse, Box<dyn Error>> {
-        let url = format!("{}/v1/prices/latest?pair={}", self.base_url, pairs.join(","));
-        let request = self.client.get(&url);
-        let resp = self.add_headers(request).send().await?;
+    pub async fn get_prices(
+        &self,
+        pair: &str,
+    ) -> Result<LatestPricesResponse, Box<dyn Error>> {
+        let mut url = format!("{}/v1/prices?pairs={}", self.base_url, pair);
+
+        let resp = self.add_headers(self.client.get(&url)).send().await?;
         match resp.status() {
             reqwest::StatusCode::OK => {
                 let response: LatestPricesResponse = resp.json().await?;
@@ -130,7 +133,7 @@ impl PricesClient {
             reqwest::StatusCode::TOO_MANY_REQUESTS => Err(Box::new(std::io::Error::new(std::io::ErrorKind::Other, "Rate limit exceeded, slow down"))),
             reqwest::StatusCode::INTERNAL_SERVER_ERROR => Err(Box::new(std::io::Error::new(std::io::ErrorKind::Other, "Internal server error"))),
             _ => Err(Box::new(std::io::Error::new(std::io::ErrorKind::Other, format!("Unexpected response status: {}", resp.status())))),
-        }
+        }        
     }
 
     pub async fn get_historical_prices(
@@ -140,7 +143,7 @@ impl PricesClient {
         end: Option<i64>, 
         granularity: Option<&str>
     ) -> Result<HashMap<String, Vec<HistoricalPriceData>>, Box<dyn Error>> {
-        let mut url = format!("{}/v1/prices?pair={}", self.base_url, pairs.join(","));
+        let mut url = format!("{}/v1/prices/historical?pairs={}", self.base_url, pairs.join(","));
 
         if let Some(start_time) = start {
             url.push_str(&format!("&start={}", start_time));

tests/price_test.rs

@@ -0,0 +1,62 @@
+
+#[cfg(test)]
+mod tests {
+    use spice_rs::*;
+    use std::env;
+    use std::path::Path;
+
+    #[tokio::test]
+    async fn test_get_prices() {
+        dotenv::from_path(Path::new(".env.local")).ok();
+        let api_key = env::var("API_KEY").expect("API_KEY not found");
+        let http_addr = "https://data.spiceai.io".to_string();
+        let flight_addr = "grpc+tls://flight.spiceai.io".to_string();
+        let firecache_addr = "grpc+tls://firecache.spiceai.io".to_string();
+
+        let pair = "BTC-USD";
+
+        let spice_client: Client = new_spice_client(api_key, http_addr, flight_addr, firecache_addr).await.expect("Failed to initiate spice client");
+
+        let result = spice_client.prices.get_prices(pair).await;
+        assert!(result.is_ok());
+
+        // Code for evaluate results received
+        // match spice_client.prices.get_prices(pair).await {
+        //     Ok(r) => {
+        //         println!("{:?}",r)
+        //     }
+        //     Err(e) => {
+        //         println!("{:?}",e)
+        //     }
+        // }
+    }
+
+    #[tokio::test]
+    async fn test_get_historical_prices() {
+        dotenv::from_path(Path::new(".env.local")).ok();
+        let api_key = env::var("API_KEY").expect("API_KEY not found");
+        let http_addr = "https://data.spiceai.io".to_string();
+        let flight_addr = "grpc+tls://flight.spiceai.io".to_string();
+        let firecache_addr = "grpc+tls://firecache.spiceai.io".to_string();
+
+        let pair1 = "BTC-USD";
+        let pair2 = "ETH-USD";
+
+        let spice_client: Client = new_spice_client(api_key, http_addr, flight_addr, firecache_addr).await.expect("Failed to initiate spice client");
+
+        let result = spice_client.prices.get_historical_prices(&[pair1, pair2], None, None, Some("1h")).await;
+        assert!(result.is_ok());
+        // Code for evaluate results received
+        // match spice_client.prices.get_historical_prices(&[pair1, pair2], Some(1672531200000), Some(1675209600000), Some("1h")).await {
+        // match spice_client.prices.get_historical_prices(&[pair1, pair2], None, None, Some("1h")).await {
+        //     Ok(r) => {
+        //         println!("{:?}",r)
+        //     }
+        //     Err(e) => {
+        //         println!("{:?}",e)
+        //     }
+        // }
+    }
+    
+    // Further tests for the client module
+}