add test for key contract send

Author: alxiong

.gitignore

@@ -35,3 +35,7 @@ __pycache__/
 
 # Rust bindings for smart contracts
 timeboost-contract/src/bindings
+
+# env var files that may contains secrets
+.env*
+!.env*.example

Cargo.lock

@@ -196,6 +196,21 @@ contract KeyManager is Initializable, OwnableUpgradeable, UUPSUpgradeable {
         return nextCommitteeId - 1;
     }
 
+    /**
+     * @notice This function is used to get the committee members.
+     * @dev We need this because alloy binding for `.committees(id)` getter doesn't include dynamic array pointer
+     * @dev Reverts if the id is greater than the length of the committees mapping.
+     * @dev Reverts if the id is less than the head committee id.
+     * @param id The id of the committee.
+     * @return members The committee members.
+     */
+    function getMembersById(uint64 id) external view virtual returns (CommitteeMember[] memory members) {
+        if (id < _oldestStoredCommitteeId || committees[id].id != id) {
+            revert CommitteeIdDoesNotExist(id);
+        }
+        return committees[id].members;
+    }
+
     /**
      * @notice This function is used to get the committee by id.
      * @dev Reverts if the id is greater than the length of the committees mapping.

contracts/src/KeyManager.sol

@@ -9,6 +9,7 @@ rust-version.workspace = true
 alloy = { workspace = true }
 anyhow = { workspace = true }
 dotenvy = "0.15"
+rand = { workspace = true }
 tracing = { workspace = true }
 tokio = { workspace = true }
 

timeboost-contract/Cargo.toml

@@ -45,9 +45,10 @@ pub async fn deploy_key_manager_contract(
 
 #[cfg(test)]
 mod tests {
-    use alloy::providers::WalletProvider;
+    use alloy::{providers::WalletProvider, sol_types::SolValue};
+    use rand::prelude::*;
 
-    use crate::KeyManager;
+    use crate::{CommitteeMemberSol, KeyManager};
 
     #[tokio::test]
     async fn test_key_manager_deployment() {
@@ -57,5 +58,42 @@ mod tests {
 
         // try read from the contract storage
         assert_eq!(contract.manager().call().await.unwrap(), manager);
+
+        // try write to the contract storage
+        let rng = &mut rand::rng();
+        let members = (0..5)
+            .map(|_| CommitteeMemberSol::random())
+            .collect::<Vec<_>>();
+        let timestamp = rng.random::<u64>();
+
+        let _tx_receipt = contract
+            .setNextCommittee(timestamp, members.clone())
+            .send()
+            .await
+            .unwrap()
+            .get_receipt()
+            .await
+            .unwrap();
+
+        // make sure next committee is correctly registered
+        assert_eq!(contract.nextCommitteeId().call().await.unwrap(), 1);
+        assert_eq!(
+            contract
+                .committees(0)
+                .call()
+                .await
+                .unwrap()
+                .effectiveTimestamp,
+            timestamp
+        );
+        assert_eq!(
+            contract
+                .getMembersById(0)
+                .call()
+                .await
+                .unwrap()
+                .abi_encode_sequence(),
+            members.abi_encode_sequence()
+        );
     }
 }

timeboost-contract/src/deployer.rs

@@ -11,14 +11,13 @@ use anyhow::Result;
 
 // Include the generated contract bindings
 // The build script auto-detects contracts and generates bindings in src/bindings/
-pub mod bindings;
+mod bindings;
 pub mod deployer;
 pub mod provider;
+mod sol_types;
 
-// We manually re-export the type here carefully due to alloy's lack of shared type:
-// tracking issue: https://github.com/foundry-rs/foundry/issues/10153
-pub use bindings::{erc1967proxy::ERC1967Proxy, keymanager::KeyManager};
 use provider::{HttpProviderWithWallet, TestProviderWithWallet, build_provider};
+pub use sol_types::*;
 
 /// Connect to a real blockchain, deploy the KeyManager contract, set the
 /// `TIMEBOOST_KEY_MANAGER_MNEMONIC` as the manager.

timeboost-contract/src/lib.rs

@@ -0,0 +1,24 @@
+//! Solidity types for contract interaction
+
+use alloy::primitives::Bytes;
+use rand::prelude::*;
+
+// We manually re-export the type here carefully due to alloy's lack of shared type:
+// tracking issue: https://github.com/foundry-rs/foundry/issues/10153
+pub use crate::bindings::{
+    erc1967proxy::ERC1967Proxy,
+    keymanager::KeyManager,
+    keymanager::KeyManager::{Committee as CommitteeSol, CommitteeMember as CommitteeMemberSol},
+};
+
+impl CommitteeMemberSol {
+    pub fn random() -> Self {
+        let mut rng = rand::rng();
+        CommitteeMemberSol {
+            sigKey: Bytes::from(rng.random::<[u8; 32]>()),
+            dhKey: Bytes::from(rng.random::<[u8; 32]>()),
+            dkgKey: Bytes::from(rng.random::<[u8; 32]>()),
+            networkAddress: format!("127.0.0.1:{}", rng.random::<u16>()),
+        }
+    }
+}