bls-aggregation integration test

Makefile

@@ -5,7 +5,7 @@ __CONTRACTS__: ##
 start-anvil-chain-with-contracts-deployed: ##
 	./crates/contracts/anvil/start-anvil-chain-with-el-and-avs-deployed.sh
 
-deploy-contracts-to-anvil-and-save-state: ## 
+deploy-contracts-to-anvil-and-save-state: ##
 	./crates/contracts/anvil/deploy-contracts-save-anvil-state.sh
 
 __TESTING__: ##
@@ -14,15 +14,18 @@ reset-anvil:
 	-docker stop anvil
 	-docker rm anvil
 
-pr: reset-anvil ## 
-	$(MAKE) start-anvil-chain-with-contracts-deployed
-	docker start anvil
+pr: reset-anvil ##
+	$(MAKE) start-anvil-chain-with-contracts-deployed > /dev/null &
+	sleep 4 # needed to wait for anvil setup to finish
 	cargo test --workspace
 	cargo clippy --workspace --lib --examples --tests --benches --all-features
 	cargo +nightly fmt -- --check
 	docker stop anvil
 
 fireblocks-tests:
-	$(MAKE) start-anvil-chain-with-contracts-deployed
-	docker start anvil
+	$(MAKE) start-anvil-chain-with-contracts-deployed > /dev/null &
 	cargo test --workspace --features fireblock-tests
+
+start-anvil: reset-anvil ##
+			 $(MAKE) start-anvil-chain-with-contracts-deployed
+			 docker start anvil

crates/chainio/clients/avsregistry/Cargo.toml

@@ -9,10 +9,15 @@ repository.workspace = true
 license-file.workspace = true
 
 [dependencies]
+alloy-contract.workspace = true
+alloy-network.workspace = true
 alloy-primitives.workspace = true
 alloy-provider.workspace = true
-alloy-transport-http.workspace = true
 alloy-rpc-types.workspace = true
+alloy-signer.workspace = true
+alloy-signer-local.workspace = true
+async-trait.workspace = true
+alloy-transport-http.workspace = true
 num-bigint = "0.4.4"
 reqwest.workspace = true
 eigen-types.workspace = true
@@ -23,14 +28,13 @@ eigen-utils.workspace = true
 eigen-logging.workspace = true
 thiserror.workspace = true
 tracing.workspace = true
-alloy-network.workspace = true
-alloy-signer.workspace = true
-alloy-signer-local.workspace = true
-alloy-contract.workspace = true
+
 [lints]
 workspace = true
 
 [dev-dependencies]
-tokio = {version = "1.37.0" , features = ["test-util", "full","sync"] }
+alloy-node-bindings.workspace = true
+eigen-testing-utils.workspace = true
 hex = "0.4.3"
-once_cell.workspace = true
+once_cell.workspace = true
+tokio = { version = "1.37.0", features = ["test-util", "full", "sync"] }

crates/chainio/clients/avsregistry/README.md

@@ -1,12 +1,12 @@
 # AvsRegistry
 
-Reader , writer and subscribing methods to interact with avs registry contracts 
+Reader , writer and subscribing methods to interact with avs registry contracts
 
-## Example 
+## Example
 
 - [get_operator_from_id](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/get_operator_from_id.rs)
 - [get_operator_id](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/get_operator_id.rs)
 - [get_operator_stake_in_quorums_of_operator_at_current_block](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/get_operator_stake_in_quorums_of_operator_at_current_block.rs)
 - [get_operators_stake_in_quorums_at_block](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/get_operators_stake_in_quorums_at_block.rs)
 - [get_quorum_count](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/get_quorum_count.rs)
-- [query_existing_registered_operator_pub_keys](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/query_existing_registered_operator_pub_keys.rs)
+- [query_existing_registered_operator_pub_keys](https://github.com/Layr-Labs/eigensdk-rs/blob/main/examples/avsregistry-read/examples/query_existing_registered_operator_pub_keys.rs)

crates/chainio/clients/avsregistry/src/error.rs

@@ -62,6 +62,10 @@ pub enum AvsRegistryError {
     #[error("Get Operator from Id")]
     GetOperatorFromId,
 
+    /// Get Operator from Id
+    #[error("Get Operator Info")]
+    GetOperatorInfo,
+
     /// Get Operator Status
     #[error("Get Operator Status")]
     GetOperatorStatus,

crates/chainio/clients/avsregistry/src/fake_reader.rs

@@ -0,0 +1,67 @@
+use crate::{error::AvsRegistryError, reader::AvsRegistryReader};
+use alloy_primitives::{Address, Bytes, FixedBytes};
+use async_trait::async_trait;
+use eigen_crypto_bls::BlsKeyPair;
+use eigen_types::test::TestOperator;
+use eigen_utils::binding::OperatorStateRetriever;
+
+const OPERATOR_STAKE: u128 = 123;
+
+/// This struct is used to test AvsRegistryServiceChainCaller methods.
+#[derive(Debug)]
+pub struct FakeAvsRegistryReader {
+    operator_address: Address,
+    operator_pubkeys: BlsKeyPair,
+    operator_id: FixedBytes<32>,
+}
+
+impl FakeAvsRegistryReader {
+    /// Creates a FakeAvsRegistryReader
+    ///
+    /// # Arguments
+    ///
+    /// * `operator` - A TestOperator.
+    /// * `operator_address` - The operator address.
+    ///
+    /// # Returns
+    ///
+    /// A FakeAvsRegistryReader
+    pub fn new(operator: TestOperator, operator_address: Address) -> Self {
+        Self {
+            operator_address,
+            operator_id: operator.operator_id,
+            operator_pubkeys: operator.bls_keypair,
+        }
+    }
+}
+
+#[async_trait]
+impl AvsRegistryReader for FakeAvsRegistryReader {
+    async fn get_operators_stake_in_quorums_at_block(
+        &self,
+        _block_number: u32,
+        _quorum_numbers: Bytes,
+    ) -> Result<Vec<Vec<OperatorStateRetriever::Operator>>, AvsRegistryError> {
+        Ok(vec![vec![OperatorStateRetriever::Operator {
+            operator: self.operator_address,
+            operatorId: self.operator_id,
+            stake: OPERATOR_STAKE,
+        }]])
+    }
+
+    async fn get_check_signatures_indices(
+        &self,
+        _reference_block_number: u32,
+        _quorum_numbers: Vec<u8>,
+        _non_signer_operator_ids: Vec<FixedBytes<32>>,
+    ) -> Result<OperatorStateRetriever::CheckSignaturesIndices, AvsRegistryError> {
+        unimplemented!()
+    }
+
+    async fn get_operator_from_id(
+        &self,
+        _operator_id: [u8; 32],
+    ) -> Result<Address, AvsRegistryError> {
+        Ok(self.operator_address)
+    }
+}

crates/chainio/clients/avsregistry/src/lib.rs

@@ -19,3 +19,7 @@ pub mod writer;
 
 /// Avs registry error message
 pub mod error;
+
+#[allow(dead_code)]
+/// Fake avs registry module
+pub mod fake_reader;