feat: Accept any struct that implements Into<ProtocolComponent> in Swap
--- don't change below this line --- ENG-4213 Took 27 minutes
This commit is contained in:
@@ -53,6 +53,17 @@ pub struct Swap {
|
|||||||
pub split: f64,
|
pub split: f64,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Swap {
|
||||||
|
pub fn new<T: Into<ProtocolComponent>>(
|
||||||
|
component: T,
|
||||||
|
token_in: Bytes,
|
||||||
|
token_out: Bytes,
|
||||||
|
split: f64,
|
||||||
|
) -> Self {
|
||||||
|
Self { component: component.into(), token_in, token_out, split }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct Transaction {
|
pub struct Transaction {
|
||||||
// Address of the contract to call with the calldata
|
// Address of the contract to call with the calldata
|
||||||
@@ -68,3 +79,42 @@ pub struct EncodingContext {
|
|||||||
pub exact_out: bool,
|
pub exact_out: bool,
|
||||||
pub router_address: Bytes,
|
pub router_address: Bytes,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
struct MockProtocolComponent {
|
||||||
|
id: String,
|
||||||
|
protocol_system: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<MockProtocolComponent> for ProtocolComponent {
|
||||||
|
fn from(component: MockProtocolComponent) -> Self {
|
||||||
|
ProtocolComponent {
|
||||||
|
id: component.id,
|
||||||
|
protocol_system: component.protocol_system,
|
||||||
|
tokens: vec![],
|
||||||
|
protocol_type_name: "".to_string(),
|
||||||
|
chain: Default::default(),
|
||||||
|
contract_ids: vec![],
|
||||||
|
static_attributes: Default::default(),
|
||||||
|
change: Default::default(),
|
||||||
|
creation_tx: Default::default(),
|
||||||
|
created_at: Default::default(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_swap_new() {
|
||||||
|
let component = MockProtocolComponent {
|
||||||
|
id: "i-am-an-id".to_string(),
|
||||||
|
protocol_system: "uniswap_v2".to_string(),
|
||||||
|
};
|
||||||
|
let swap = Swap::new(component, Bytes::from("0x12"), Bytes::from("34"), 0.5);
|
||||||
|
assert_eq!(swap.token_in, Bytes::from("0x12"));
|
||||||
|
assert_eq!(swap.token_out, Bytes::from("0x34"));
|
||||||
|
assert_eq!(swap.component.protocol_system, "uniswap_v2");
|
||||||
|
assert_eq!(swap.component.id, "i-am-an-id");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user