update tests

Layr-Labs · Oct 15, 2024 · d38693f · d38693f
1 parent 3a2095f
commit d38693f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/crates/services/operatorsinfo/src/operatorsinfo_inmemory.rs b/crates/services/operatorsinfo/src/operatorsinfo_inmemory.rs
@@ -297,7 +297,7 @@ mod tests {
     use eigen_testing_utils::anvil_constants::{
         get_avs_directory_address, get_delegation_manager_address,
         get_operator_state_retriever_address, get_registry_coordinator_address,
-        get_strategy_manager_address,
+        get_rewards_coordinator_address, get_strategy_manager_address,
     };
     use eigen_types::operator::Operator;
     use eigen_utils::get_provider;
@@ -473,6 +473,8 @@ mod tests {
             get_delegation_manager_address(http_endpoint.clone()).await;
         let avs_directory_address = get_avs_directory_address(http_endpoint.clone()).await;
         let strategy_manager_address = get_strategy_manager_address(http_endpoint.clone()).await;
+        let rewards_coordinator_address =
+            get_rewards_coordinator_address(http_endpoint.clone()).await;
         let el_chain_reader = ELChainReader::new(
             get_test_logger(),
             Address::ZERO,
@@ -485,6 +487,7 @@ mod tests {
         let el_chain_writer = ELChainWriter::new(
             delegation_manager_address,
             strategy_manager_address,
+            rewards_coordinator_address,
             el_chain_reader,
             http_endpoint.to_string(),
             pvt_key.to_string(),

diff --git a/examples/info-operator-service/examples/get_operator_info.rs b/examples/info-operator-service/examples/get_operator_info.rs
@@ -15,7 +15,7 @@ use eigen_testing_utils::{
     anvil_constants::{
         get_avs_directory_address, get_delegation_manager_address,
         get_operator_state_retriever_address, get_registry_coordinator_address,
-        get_strategy_manager_address,
+        get_rewards_coordinator_address, get_strategy_manager_address,
     },
 };
 use std::{
@@ -73,6 +73,7 @@ pub async fn register_operator(pvt_key: &str, bls_key: &str) {
     let delegation_manager_address = get_delegation_manager_address(http_endpoint.clone()).await;
     let avs_directory_address = get_avs_directory_address(http_endpoint.clone()).await;
     let strategy_manager_address = get_strategy_manager_address(http_endpoint.clone()).await;
+    let rewards_coordinator_address = get_rewards_coordinator_address(http_endpoint.clone()).await;
     let el_chain_reader = ELChainReader::new(
         get_test_logger(),
         Address::ZERO,
@@ -85,6 +86,7 @@ pub async fn register_operator(pvt_key: &str, bls_key: &str) {
     let el_chain_writer = ELChainWriter::new(
         delegation_manager_address,
         strategy_manager_address,
+        rewards_coordinator_address,
         el_chain_reader,
         http_endpoint.to_string(),
         pvt_key.to_string(),