[solana/cosmwasm/near] Rust tests refactor

.github/workflows/ci-near-contract.yml

@@ -0,0 +1,29 @@
+name: Test Near Contract
+
+on:
+  pull_request:
+    paths:
+      - target_chains/near/**
+      - wormhole_attester/sdk/rust/**
+  push:
+    branches:
+      - main
+    paths:
+      - target_chains/near/**
+      - wormhole_attester/sdk/rust/**
+
+env:
+  CARGO_TERM_COLOR: always
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    defaults:
+      run:
+        working-directory: target_chains/near/receiver
+    steps:
+      - uses: actions/checkout@v2
+      - name: Install nextest
+        run: curl -LsSf https://get.nexte.st/latest/linux | tar zxf - -C ${CARGO_HOME:-~/.cargo}/bin
+      - name: Test
+        run: ./workspace-test.sh

pythnet/pythnet_sdk/Cargo.toml

@@ -10,6 +10,9 @@ edition = "2021"
 crate-type = ["lib"]
 name = "pythnet_sdk"
 
+[features]
+test-utils = ["dep:wormhole-sdk", "dep:serde_wormhole"]
+
 [dependencies]
 bincode = "1.3.1"
 borsh = "0.10.3"
@@ -23,6 +26,12 @@ quickcheck = { version = "1", optional = true}
 sha3 = "0.10.4"
 slow_primes = "0.1.14"
 thiserror = "1.0.40"
+serde_wormhole = { git = "https://github.com/wormhole-foundation/wormhole", optional = true, tag="v2.23.37"}
+wormhole-sdk = { git = "https://github.com/wormhole-foundation/wormhole", optional = true, tag="v2.23.37"}
+
+[patch.crates-io]
+serde_wormhole = { git = "https://github.com/wormhole-foundation/wormhole", tag="v2.23.37"}
+
 
 [dev-dependencies]
 base64 = "0.21.0"

pythnet/pythnet_sdk/src/lib.rs

@@ -5,6 +5,9 @@ pub mod messages;
 pub mod wire;
 pub mod wormhole;
 
+#[cfg(feature = "test-utils")]
+pub mod test_utils;
+
 pub(crate) type Pubkey = [u8; 32];
 
 /// Official Message Buffer Program Id

pythnet/pythnet_sdk/src/test_utils/mod.rs

@@ -0,0 +1,133 @@
+use {
+    crate::{
+        accumulators::{
+            merkle::MerkleTree,
+            Accumulator,
+        },
+        hashers::keccak256_160::Keccak160,
+        messages::Message,
+        wire::{
+            to_vec,
+            v1::{
+                AccumulatorUpdateData,
+                MerklePriceUpdate,
+                Proof,
+                WormholeMerkleRoot,
+                WormholeMessage,
+                WormholePayload,
+            },
+            PrefixedVec,
+        },
+    },
+    byteorder::BigEndian,
+    serde_wormhole::RawMessage,
+    std::io::{
+        Cursor,
+        Write,
+    },
+    wormhole_sdk::{
+        Address,
+        Chain,
+        Vaa,
+    },
+};
+
+pub const EMITTER_CHAIN: u16 = 0;
+pub const DEFAULT_SEQUENCE: u64 = 2;
+
+pub fn default_emitter_addr() -> [u8; 32] {
+    [1u8; 32]
+}
+
+pub fn default_governance_addr() -> [u8; 32] {
+    [2u8; 32]
+}
+
+pub fn default_new_governance_addr() -> [u8; 32] {
+    [3u8; 32]
+}
+
+pub fn default_wrong_emitter_addr() -> [u8; 32] {
+    [4u8; 32]
+}
+
+pub fn create_accumulator_message(
+    all_feeds: &[Message],
+    updates: &[Message],
+    corrupt_wormhole_message: bool,
+) -> Vec<u8> {
+    let all_feeds_bytes: Vec<_> = all_feeds
+        .iter()
+        .map(|f| to_vec::<_, BigEndian>(f).unwrap())
+        .collect();
+    let all_feeds_bytes_refs: Vec<_> = all_feeds_bytes.iter().map(|f| f.as_ref()).collect();
+    let tree = MerkleTree::<Keccak160>::new(all_feeds_bytes_refs.as_slice()).unwrap();
+    let mut price_updates: Vec<MerklePriceUpdate> = vec![];
+    for update in updates {
+        let proof = tree
+            .prove(&to_vec::<_, BigEndian>(update).unwrap())
+            .unwrap();
+        price_updates.push(MerklePriceUpdate {
+            message: PrefixedVec::from(to_vec::<_, BigEndian>(update).unwrap()),
+            proof,
+        });
+    }
+    create_accumulator_message_from_updates(
+        price_updates,
+        tree,
+        corrupt_wormhole_message,
+        default_emitter_addr(),
+        EMITTER_CHAIN,
+    )
+}
+
+pub fn create_accumulator_message_from_updates(
+    price_updates: Vec<MerklePriceUpdate>,
+    tree: MerkleTree<Keccak160>,
+    corrupt_wormhole_message: bool,
+    emitter_address: [u8; 32],
+    emitter_chain: u16,
+) -> Vec<u8> {
+    let mut root_hash = [0u8; 20];
+    root_hash.copy_from_slice(&to_vec::<_, BigEndian>(&tree.root).unwrap()[..20]);
+    let wormhole_message = WormholeMessage::new(WormholePayload::Merkle(WormholeMerkleRoot {
+        slot:      0,
+        ring_size: 0,
+        root:      root_hash,
+    }));
+
+    let mut vaa_payload = to_vec::<_, BigEndian>(&wormhole_message).unwrap();
+    if corrupt_wormhole_message {
+        vaa_payload[0] = 0;
+    }
+
+    let vaa = create_vaa_from_payload(
+        &vaa_payload,
+        emitter_address,
+        emitter_chain,
+        DEFAULT_SEQUENCE,
+    );
+
+    let accumulator_update_data = AccumulatorUpdateData::new(Proof::WormholeMerkle {
+        vaa:     PrefixedVec::from(serde_wormhole::to_vec(&vaa).unwrap()),
+        updates: price_updates,
+    });
+
+    to_vec::<_, BigEndian>(&accumulator_update_data).unwrap()
+}
+
+pub fn create_vaa_from_payload(
+    payload: &[u8],
+    emitter_address: [u8; 32],
+    emitter_chain: u16,
+    sequence: u64,
+) -> Vaa<Box<RawMessage>> {
+    let vaa: Vaa<Box<RawMessage>> = Vaa {
+        emitter_chain: Chain::from(emitter_chain),
+        emitter_address: Address(emitter_address),
+        sequence,
+        payload: <Box<RawMessage>>::from(payload.to_vec()),
+        ..Default::default()
+    };
+    vaa
+}

target_chains/cosmwasm/Cargo.toml

@@ -14,3 +14,6 @@ codegen-units = 1
 panic = 'abort'
 incremental = false
 overflow-checks = true
+
+[patch.crates-io]
+serde_wormhole = { git = "https://github.com/wormhole-foundation/wormhole", tag="v2.23.37"}

target_chains/cosmwasm/contracts/pyth/Cargo.toml

@@ -45,3 +45,6 @@ wormhole-cosmwasm =  {git = "https://github.com/wormhole-foundation/wormhole", t
 [dev-dependencies]
 cosmwasm-vm = { version = "1.0.0", default-features = false }
 serde_json = "1.0"
+pythnet-sdk = { path = "../../../../pythnet/pythnet_sdk", features = ["test-utils"] }
+serde_wormhole = { git = "https://github.com/wormhole-foundation/wormhole", tag="v2.23.37"}
+wormhole-sdk = { git = "https://github.com/wormhole-foundation/wormhole", tag="v2.23.37"}