chore: After rebase with master fixes

Took 2 minutes
This commit is contained in:
Diana Carvalho
2025-02-04 17:32:14 +00:00
parent c89f46a2e6
commit 79f513c76b
2 changed files with 13 additions and 10 deletions

View File

@@ -498,8 +498,10 @@ mod tests {
token_out: dai.clone(), token_out: dai.clone(),
split: 0f64, split: 0f64,
}; };
let swap_encoder_registry = get_swap_encoder_registry();
let encoder = SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum).unwrap(); let encoder =
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
.unwrap();
let solution = Solution { let solution = Solution {
exact_out: false, exact_out: false,
given_token: NATIVE_ADDRESS.clone(), given_token: NATIVE_ADDRESS.clone(),
@@ -545,8 +547,10 @@ mod tests {
token_out: WETH_ADDRESS.clone(), token_out: WETH_ADDRESS.clone(),
split: 0f64, split: 0f64,
}; };
let swap_encoder_registry = get_swap_encoder_registry();
let encoder = SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum).unwrap(); let encoder =
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
.unwrap();
let solution = Solution { let solution = Solution {
exact_out: false, exact_out: false,
given_token: dai, given_token: dai,

View File

@@ -24,7 +24,7 @@ impl<S: StrategyEncoderRegistry> EVMTychoEncoder<S> {
} }
} }
impl<S: StrategySelector> EVMTychoEncoder<S> { impl<S: StrategyEncoderRegistry> EVMTychoEncoder<S> {
fn validate_solution(&self, solution: &Solution) -> Result<(), EncodingError> { fn validate_solution(&self, solution: &Solution) -> Result<(), EncodingError> {
if solution.exact_out { if solution.exact_out {
return Err(EncodingError::FatalError( return Err(EncodingError::FatalError(
@@ -104,13 +104,12 @@ impl<S: StrategyEncoderRegistry> TychoEncoder<S> for EVMTychoEncoder<S> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use tycho_core::models::Chain; use tycho_core::{dto::ProtocolComponent, models::Chain};
use tycho_core::dto::ProtocolComponent;
use super::*; use super::*;
use crate::encoding::{strategy_encoder::StrategyEncoder, swap_encoder::SwapEncoder}; use crate::encoding::{
use crate::encoding::{models::Swap, strategy_encoder::StrategyEncoder}; models::Swap, strategy_encoder::StrategyEncoder, swap_encoder::SwapEncoder,
};
struct MockStrategyRegistry { struct MockStrategyRegistry {
strategy: Box<dyn StrategyEncoder>, strategy: Box<dyn StrategyEncoder>,