added potential fix to api error issue within the test cases

Author: bhargavakula01

oscps-lib/src/component.rs

@@ -6,6 +6,26 @@ extern crate uom;
 
 extern crate pubchem;
 use anyhow::Result;
+use std::{thread,time::Duration};
+
+// pub enum RequestStatusCodes {
+//     BadRequest,
+//     NotFound,
+//     GatewayError,
+//     Success
+
+// }
+
+// impl RequestStatusCodes {
+//     pub fn value(&self) -> u32 {
+//         match self {
+//             RequestStatusCodes::Success => 200,
+//             RequestStatusCodes::NotFound => 404,
+//             RequestStatusCodes::GatewayError => 504,
+//             RequestStatusCodes::BadRequest => 400
+//         }
+//     }
+// }
 
 #[allow(dead_code)]
 /// This will hold the list of chemicals used within the simulation
@@ -43,9 +63,23 @@ impl Chemical {
             ChemicalIdentifier::PubchemID(id) => pubchem::Compound::new(id),
             ChemicalIdentifier::CompoundName(name) => pubchem::Compound::with_name(name.as_str()),
         };
-
-        let cid_vec = pubchem_chemical_object.cids().unwrap();
-        let cid: i32 = cid_vec[0];
+        let mut request_counter = 0;
+        let mut cid_vec = None;
+        while request_counter <= 10 {
+            match pubchem_chemical_object.cids(){
+                Ok(cid_list) => {
+                    cid_vec = Some(cid_list);
+                    break;
+                },
+                _ => {
+                    request_counter = request_counter + 1;
+                    thread::sleep(Duration::from_secs(10));
+                }
+            };
+        }
+
+        // let cid_vec = pubchem_chemical_object.cids().unwrap();
+        let cid: i32 = cid_vec.unwrap()[0];
         let prop = ChemicalProperties::new(cid).unwrap();
         return Ok(Chemical {
             pubchem_obj: pubchem_chemical_object,
@@ -93,14 +127,16 @@ impl ChemicalProperties {
 #[cfg(test)]
 mod chemical_species_tests {
     use crate::component::{Chemical, ChemicalIdentifier};
+    use std::{thread,time::Duration};
 
     #[test]
     fn test_create_chemical_from_pubchem_id() {
         // Using a known PubChem ID, e.g., 7732 (water)
         let identifier = ChemicalIdentifier::PubchemID(7732);
 
         let chemical = Chemical::new(identifier);
-
+        thread::sleep(Duration::from_secs(10));
+        
         assert!(
             chemical.is_ok(),
             "Failed to create chemical from PubChem ID"
@@ -119,6 +155,8 @@ mod chemical_species_tests {
         let identifier = ChemicalIdentifier::CompoundName(String::from("Water"));
 
         let chemical = Chemical::new(identifier);
+        thread::sleep(Duration::from_secs(10));
+
 
         assert!(chemical.is_ok(), "Failed to create chemical from name");
         let chemical = chemical.unwrap();

oscps-lib/src/thermodynamics.rs

@@ -132,6 +132,7 @@ mod thermo_tests {
     use uom::si::mass::kilogram;
     use uom::si::pressure::pascal;
     use uom::si::thermodynamic_temperature::kelvin;
+    use std::{thread,time::Duration};
 
     #[test]
     ///Test case generates an instance of the 'ThermoState' struct
@@ -146,6 +147,7 @@ mod thermo_tests {
                 acentric_factor: 0.344,    // example
             },
         };
+        thread::sleep(Duration::from_secs(10));
         let water_mass = Mass::new::<kilogram>(2.0);
         let water_species_pair = SpeciesListPair {
             chemical_species: water,
@@ -167,14 +169,14 @@ mod thermo_tests {
 
 
         // Check that the mole fraction's chemical is correctly set
-        // assert_eq!(
-        //     thermo_state.mass_list[0]
-        //         .chemical_species
-        //         .get_pubchem_obj()
-        //         .cids()
-        //         .unwrap()[0],
-        //     962
-        // );
+        assert_eq!(
+            thermo_state.mass_list[0]
+                .chemical_species
+                .get_pubchem_obj()
+                .cids()
+                .unwrap()[0],
+            962
+        );
     }
 
     #[test]
@@ -189,6 +191,7 @@ mod thermo_tests {
                 acentric_factor: 0.344,    // example
             },
         };
+        thread::sleep(Duration::from_secs(10));
 
         let anisdine = Chemical {
             pubchem_obj: pubchem::Compound::new(7732),
@@ -199,7 +202,8 @@ mod thermo_tests {
                 acentric_factor: 0.24,    // (approximated)
             },
         };
-
+        thread::sleep(Duration::from_secs(10));
+        
         let water_mass = Mass::new::<kilogram>(2.0);
         let water_species_pair = SpeciesListPair {
             chemical_species: water,
@@ -222,9 +226,9 @@ mod thermo_tests {
             .mass_frac(&therm_obj.mass_list[0].chemical_species)
             .unwrap();
 
-        // assert!(
-        //     (mass_fraction - 0.2).abs() < 1e-6,
-        //     "Mole fraction calculation failed"
-        // ); // Should be 0.2
+        assert!(
+            (mass_fraction - 0.2).abs() < 1e-6,
+            "Mole fraction calculation failed"
+        ); // Should be 0.2
     }
 }