From e9c1a39608bba786bbed2b53c4596e94ddcbe03c Mon Sep 17 00:00:00 2001 From: TAMARA LIPOWSKI Date: Thu, 21 Aug 2025 11:03:01 -0400 Subject: [PATCH] fix: Updates to new datatype after merging main --- src/encoding/evm/swap_encoder/swap_encoders.rs | 2 +- tests/optimized_transfers_integration_tests.rs | 2 +- tests/protocol_integration_tests.rs | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/encoding/evm/swap_encoder/swap_encoders.rs b/src/encoding/evm/swap_encoder/swap_encoders.rs index c238617..64dd0d8 100644 --- a/src/encoding/evm/swap_encoder/swap_encoders.rs +++ b/src/encoding/evm/swap_encoder/swap_encoders.rs @@ -2304,7 +2304,7 @@ mod tests { let swap = SwapBuilder::new(hashflow_component, token_in.clone(), token_out.clone()) .estimated_amount_in(BigUint::from_str("3000000000").unwrap()) - .protocol_state(&hashflow_state) + .protocol_state(Arc::new(hashflow_state)) .build(); let encoding_context = EncodingContext { diff --git a/tests/optimized_transfers_integration_tests.rs b/tests/optimized_transfers_integration_tests.rs index 2233352..477ed27 100644 --- a/tests/optimized_transfers_integration_tests.rs +++ b/tests/optimized_transfers_integration_tests.rs @@ -800,7 +800,7 @@ fn test_uniswap_v3_hashflow() { let swap_usdc_wbtc = SwapBuilder::new(hashflow_component, usdc.clone(), wbtc.clone()) .estimated_amount_in(BigUint::from_str("4308094737").unwrap()) - .protocol_state(&hashflow_state) + .protocol_state(Arc::new(hashflow_state)) .build(); let encoder = get_tycho_router_encoder(UserTransferType::TransferFrom); diff --git a/tests/protocol_integration_tests.rs b/tests/protocol_integration_tests.rs index fb13bc7..901a7c5 100644 --- a/tests/protocol_integration_tests.rs +++ b/tests/protocol_integration_tests.rs @@ -1,5 +1,5 @@ mod common; -use std::{collections::HashMap, str::FromStr}; +use std::{collections::HashMap, str::FromStr, sync::Arc}; use alloy::{hex, hex::encode}; use num_bigint::{BigInt, BigUint}; @@ -785,7 +785,7 @@ fn test_single_encoding_strategy_hashflow() { let swap_usdc_wbtc = SwapBuilder::new(hashflow_component, usdc.clone(), wbtc.clone()) .estimated_amount_in(BigUint::from_str("4308094737").unwrap()) - .protocol_state(&hashflow_state) + .protocol_state(Arc::new(hashflow_state)) .build(); let encoder = get_tycho_router_encoder(UserTransferType::TransferFrom);