avs registry service chain caller tests (Layr-Labs#90)

This PR adds unit tests for `AvsRegistryServiceChainCaller` methods

---------

Co-authored-by: tomasarrachea <tomas.arrachea@lambdaclass.com>
Co-authored-by: Pablo Deymonnaz <deymonnaz@gmail.com>
Co-authored-by: Pablo Deymonnaz <pdeymon@fi.uba.ar>

crates/chainio/clients/avsregistry/Cargo.toml

@@ -15,20 +15,19 @@ alloy-primitives.workspace = true
 alloy-provider.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
+eigen-crypto-bls.workspace = true
 ark-ff.workspace = true
 eigen-client-elcontracts.workspace = true
-eigen-crypto-bls.workspace = true
-eigen-logging.workspace = true
-eigen-types.workspace = true
 eigen-utils.workspace = true
-num-bigint = "0.4.4"
-reqwest.workspace = true
+eigen-logging.workspace = true
 thiserror.workspace = true
 tracing.workspace = true
-[lints]
-workspace = true
+alloy-signer-local.workspace = true
 
 [dev-dependencies]
 hex = "0.4.3"

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;