test framework

Author: ayushboss

target_chains/stylus/contracts/pyth-receiver/src/integration_tests.rs

@@ -0,0 +1,17 @@
+#[cfg(test)]
+mod test {
+    use super::*;
+    use alloy_primitives::address;
+    use stylus_sdk::testing::*;
+
+    #[test]
+    fn test_initialize() {
+        // Set up test environment
+        let vm = TestVM::default();
+        // Initialize your contract
+        let mut contract = PythReceiver::from(&vm);
+
+        let wormhole_address = address!("0x3F38404A2e3Cb949bcDfA19a5C3bDf3fE375fEb0");
+
+    }
+}

target_chains/stylus/contracts/pyth-receiver/src/lib.rs

@@ -7,6 +7,7 @@ extern crate alloc;
 
 mod structs;
 mod error;
+mod integration_tests;
 
 use alloc::vec::Vec;
 use stylus_sdk::{alloy_primitives::{U16, U32, U256, U64, I32, I64, FixedBytes, Address},
@@ -163,8 +164,6 @@ impl PythReceiver {
                 let parsed_vaa = wormhole.parse_and_verify_vm(config, Vec::from(vaa)).map_err(|_| PythReceiverError::InvalidWormholeMessage).unwrap();
                 let vaa = Vaa::read(&mut parsed_vaa.as_slice()).unwrap();
 
-                // TODO: CHECK IF THE VAA IS FROM A VALID DATA SOURCE
-
                 let cur_emitter_address: &[u8; 32] = vaa.body.emitter_address.as_slice().try_into().expect("emitter address must be 32 bytes");
 
                 let cur_data_source = DataSource {
@@ -283,3 +282,7 @@ fn parse_wormhole_proof(vaa: Vaa) -> Result<MerkleRoot<Keccak160>, PythReceiverE
     });
     Ok(root)
 }
+
+
+#[cfg(test)]
+pub use integration_tests::*;