ethers-rs/ethers-contract/tests/it/derive.rs

701 lines
20 KiB
Rust

use ethers_contract::{
abigen, EthAbiCodec, EthAbiType, EthCall, EthDisplay, EthError, EthEvent, EthLogDecode,
};
use ethers_core::{
abi::{AbiDecode, AbiEncode, RawLog, Tokenizable},
types::{Address, Bytes, H160, H256, I256, U128, U256},
};
fn assert_tokenizeable<T: Tokenizable>() {}
fn assert_ethcall<T: EthCall>() {}
fn assert_etherror<T: EthError>() {}
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct ValueChanged {
old_author: Address,
new_author: Address,
old_value: String,
new_value: String,
}
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct ValueChangedWrapper {
inner: ValueChanged,
msg: String,
}
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct ValueChangedTuple(Address, Address, String, String);
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct ValueChangedTupleWrapper(ValueChangedTuple, String);
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct ValueChangedVecWrapper {
inner: Vec<ValueChanged>,
}
#[test]
fn can_detokenize_struct() {
let value = ValueChanged {
old_author: "eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
new_author: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
old_value: "50".to_string(),
new_value: "100".to_string(),
};
let token = value.clone().into_token();
assert_eq!(value, ValueChanged::from_token(token).unwrap());
}
#[test]
fn can_derive_abi_type_empty_struct() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct Call();
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct Call2;
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct Call3;
assert_tokenizeable::<Call>();
assert_tokenizeable::<Call2>();
assert_tokenizeable::<Call3>();
}
#[test]
fn can_detokenize_nested_structs() {
let value = ValueChangedWrapper {
inner: ValueChanged {
old_author: "eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
new_author: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
old_value: "50".to_string(),
new_value: "100".to_string(),
},
msg: "hello world".to_string(),
};
let token = value.clone().into_token();
assert_eq!(value, ValueChangedWrapper::from_token(token).unwrap());
}
#[test]
fn can_detokenize_tuple_struct() {
let value = ValueChangedTuple(
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
"50".to_string(),
"100".to_string(),
);
let token = value.clone().into_token();
assert_eq!(value, ValueChangedTuple::from_token(token).unwrap());
}
#[test]
fn can_detokenize_nested_tuple_struct() {
let value = ValueChangedTupleWrapper(
ValueChangedTuple(
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
"50".to_string(),
"100".to_string(),
),
"hello world".to_string(),
);
let token = value.clone().into_token();
assert_eq!(value, ValueChangedTupleWrapper::from_token(token).unwrap());
}
#[test]
fn can_detokenize_single_field() {
let value = ValueChangedVecWrapper { inner: vec![] };
let token = value.clone().into_token();
assert_eq!(value, ValueChangedVecWrapper::from_token(token).unwrap());
let value = ValueChangedVecWrapper {
inner: vec![ValueChanged {
old_author: "eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
new_author: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
old_value: "50".to_string(),
new_value: "100".to_string(),
}],
};
let token = value.clone().into_token();
assert_eq!(value, ValueChangedVecWrapper::from_token(token).unwrap());
}
#[test]
fn can_derive_eth_event() {
#[derive(Debug, Clone, PartialEq, Eq, EthEvent)]
struct ValueChangedEvent {
old_author: Address,
new_author: Address,
old_value: String,
new_value: String,
}
let value = ValueChangedEvent {
old_author: "eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee".parse().unwrap(),
new_author: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
old_value: "50".to_string(),
new_value: "100".to_string(),
};
assert_eq!("ValueChangedEvent", ValueChangedEvent::name());
assert_eq!(
"ValueChangedEvent(address,address,string,string)",
ValueChangedEvent::abi_signature()
);
let token = value.clone().into_token();
assert_eq!(value, ValueChangedEvent::from_token(token).unwrap());
}
#[test]
fn can_set_eth_event_name_attribute() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
#[ethevent(name = "MyEvent")]
struct ValueChangedEvent {
old_author: Address,
new_author: Address,
old_value: String,
new_value: String,
}
assert_eq!("MyEvent", ValueChangedEvent::name());
assert_eq!("MyEvent(address,address,string,string)", ValueChangedEvent::abi_signature());
}
#[test]
fn can_detect_various_event_abi_types() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
struct ValueChangedEvent {
old_author: Address,
s: String,
h1: H256,
i256: I256,
u256: U256,
b: bool,
v: Vec<Address>,
bs: Vec<bool>,
h160: H160,
u128: U128,
int8: i8,
int16: i16,
int32: i32,
int64: i64,
int128: i128,
uint8: u8,
uint16: u16,
uint32: u32,
uint64: u64,
uint128: u128,
}
assert_eq!(
"ValueChangedEvent(address,string,bytes32,int256,uint256,bool,address[],bool[],bytes20,uint128,int8,int16,int32,int64,int128,uint8,uint16,uint32,uint64,uint128)",
ValueChangedEvent::abi_signature()
);
}
#[test]
fn can_set_eth_abi_attribute() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct SomeType {
inner: Address,
msg: String,
}
#[derive(Debug, PartialEq, Eq, EthEvent)]
#[ethevent(abi = "ValueChangedEvent(address,(address,string),string)")]
struct ValueChangedEvent {
old_author: Address,
inner: SomeType,
new_value: String,
}
assert_eq!(
"ValueChangedEvent(address,(address,string),string)",
ValueChangedEvent::abi_signature()
);
#[derive(Debug, PartialEq, Eq, EthEvent)]
#[ethevent(
name = "ValueChangedEvent",
abi = "ValueChangedEvent(address,(address,string),string)"
)]
struct ValueChangedEvent2 {
old_author: Address,
inner: SomeType,
new_value: String,
}
assert_eq!(
"ValueChangedEvent(address,(address,string),string)",
ValueChangedEvent2::abi_signature()
);
}
#[test]
fn can_derive_indexed_and_anonymous_attribute() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
#[ethevent(anonymous)]
struct ValueChangedEvent {
old_author: Address,
#[ethevent(indexed, name = "newAuthor")]
new_author: Address,
old_value: String,
new_value: String,
}
assert_eq!(
"ValueChangedEvent(address,address,string,string) anonymous",
ValueChangedEvent::abi_signature()
);
}
#[test]
fn can_generate_ethevent_from_json() {
abigen!(
DsProxyFactory,
"./../ethers-middleware/contracts/DSProxyFactory.json",
methods {
build(address) as build_with_owner;
}
);
assert_eq!("Created(address,address,address,address)", CreatedFilter::abi_signature());
assert_eq!(
H256([
37, 155, 48, 202, 57, 136, 92, 109, 128, 26, 11, 93, 188, 152, 134, 64, 243, 194, 94,
47, 55, 83, 31, 225, 56, 197, 197, 175, 137, 85, 212, 27,
]),
CreatedFilter::signature()
);
}
#[test]
fn can_decode_event_with_no_topics() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
pub struct LiquidateBorrow {
liquidator: Address,
borrower: Address,
repay_amount: U256,
c_token_collateral: Address,
seize_tokens: U256,
}
// https://etherscan.io/tx/0xb7ba825294f757f8b8b6303b2aef542bcaebc9cc0217ddfaf822200a00594ed9#eventlog index 141
let log = RawLog {
topics: vec!["298637f684da70674f26509b10f07ec2fbc77a335ab1e7d6215a4b2484d8bb52"
.parse()
.unwrap()],
data: vec![
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 188, 205, 0, 29, 173, 151, 238, 5, 127, 91, 31,
197, 154, 221, 40, 175, 143, 32, 26, 201, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 133, 129,
195, 136, 163, 5, 24, 136, 69, 34, 251, 23, 122, 146, 252, 33, 147, 81, 8, 20, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 9, 18, 195, 162, 210,
38, 128, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 77, 220, 45, 25, 57, 72, 146, 109, 2,
249, 177, 254, 158, 29, 170, 7, 24, 39, 14, 213, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 80, 30, 88,
],
};
let event = <LiquidateBorrow as EthLogDecode>::decode_log(&log).unwrap();
assert_eq!(event.seize_tokens, 5250648u64.into());
assert_eq!(event.repay_amount, 653800000000000000u64.into());
}
#[test]
fn can_decode_event_single_param() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
pub struct OneParam {
#[ethevent(indexed)]
param1: U256,
}
let log = RawLog {
topics: vec![
"bd9bb67345a2fcc8ef3b0857e7e2901f5a0dcfc7fe5e3c10dc984f02842fb7ba".parse().unwrap(),
"000000000000000000000000000000000000000000000000000000000000007b".parse().unwrap(),
],
data: vec![],
};
let event = <OneParam as EthLogDecode>::decode_log(&log).unwrap();
assert_eq!(event.param1, 123u64.into());
}
#[test]
fn can_decode_event_tuple_single_param() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
struct OneParam(#[ethevent(indexed)] U256);
let log = RawLog {
topics: vec![
"bd9bb67345a2fcc8ef3b0857e7e2901f5a0dcfc7fe5e3c10dc984f02842fb7ba".parse().unwrap(),
"000000000000000000000000000000000000000000000000000000000000007b".parse().unwrap(),
],
data: vec![],
};
let event = <OneParam as EthLogDecode>::decode_log(&log).unwrap();
assert_eq!(event.0, 123u64.into());
}
#[test]
fn can_decode_event_with_no_params() {
#[derive(Debug, PartialEq, Eq, EthEvent)]
pub struct NoParam {}
let log = RawLog {
topics: vec!["59a6f900daaeb7581ff830f3a97097fa6372db29b0b50c6d1818ede9d1daaa0c"
.parse()
.unwrap()],
data: vec![],
};
let _ = <NoParam as EthLogDecode>::decode_log(&log).unwrap();
}
#[test]
fn eth_display_works() {
#[derive(Debug, Clone, EthAbiType, EthDisplay)]
struct MyStruct {
addr: Address,
old_value: String,
new_value: String,
h: H256,
i: I256,
arr_u8: [u8; 32],
arr_u16: [u16; 32],
v: Vec<u8>,
}
let item = MyStruct {
addr: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa".parse().unwrap(),
old_value: "50".to_string(),
new_value: "100".to_string(),
h: H256::random(),
i: I256::zero(),
arr_u8: [0; 32],
arr_u16: [1; 32],
v: vec![0; 32],
};
let val = format!(
"0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa, 50, 100, 0x{}, {}, 0x{}, {:?}, 0x{}",
hex::encode(item.h),
item.i,
hex::encode(item.arr_u8),
item.arr_u16,
hex::encode(&item.v),
);
assert_eq!(val, format!("{item}"));
}
#[test]
fn eth_display_works_for_human_readable() {
abigen!(
HevmConsole,
r#"[
event log(string)
event log2(string x)
]"#,
);
let log = LogFilter("abc".to_string());
assert_eq!("abc".to_string(), format!("{log}"));
let log = Log2Filter { x: "abc".to_string() };
assert_eq!("abc".to_string(), format!("{log}"));
}
#[test]
fn can_derive_ethcall() {
#[derive(Debug, Clone, EthCall, EthDisplay)]
struct MyStruct {
addr: Address,
old_value: String,
new_value: String,
h: H256,
i: I256,
arr_u8: [u8; 32],
arr_u16: [u16; 32],
nested_arr: [[u8; 32]; 2],
double_nested: [[[u8; 32]; 2]; 3],
v: Vec<u8>,
}
assert_tokenizeable::<MyStruct>();
assert_ethcall::<MyStruct>();
#[derive(Debug, Clone, EthCall, EthDisplay)]
#[ethcall(name = "my_call")]
struct MyCall {
addr: Address,
old_value: String,
new_value: String,
}
assert_eq!(MyCall::abi_signature().as_ref(), "my_call(address,string,string)");
assert_tokenizeable::<MyCall>();
assert_ethcall::<MyCall>();
}
#[test]
fn can_derive_ethcall_with_nested_structs() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
struct SomeType {
inner: Address,
msg: String,
}
#[derive(Debug, PartialEq, Eq, EthCall)]
#[ethcall(name = "foo", abi = "foo(address,(address,string),string)")]
struct FooCall {
old_author: Address,
inner: SomeType,
new_value: String,
}
assert_eq!(FooCall::abi_signature().as_ref(), "foo(address,(address,string),string)");
assert_tokenizeable::<FooCall>();
assert_ethcall::<FooCall>();
}
#[test]
fn can_derive_for_enum() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType)]
enum ActionChoices {
GoLeft,
GoRight,
GoStraight,
SitStill,
}
assert_tokenizeable::<ActionChoices>();
let token = ActionChoices::GoLeft.into_token();
assert_eq!(ActionChoices::GoLeft, ActionChoices::from_token(token).unwrap());
}
#[test]
fn can_derive_abi_codec() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType, EthAbiCodec)]
pub struct SomeType {
inner: Address,
msg: String,
}
let val = SomeType { inner: Default::default(), msg: "hello".to_string() };
let encoded = val.clone().encode();
let other = SomeType::decode(encoded).unwrap();
assert_eq!(val, other);
}
#[test]
fn can_derive_abi_codec_single_field() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType, EthAbiCodec)]
pub struct SomeType {
inner: Vec<U256>,
}
let val = SomeType { inner: Default::default() };
let encoded = val.clone().encode();
let decoded = SomeType::decode(&encoded).unwrap();
assert_eq!(val, decoded);
let encoded_tuple = (Vec::<U256>::default(),).encode();
assert_eq!(encoded_tuple, encoded);
let decoded_tuple = SomeType::decode(&encoded_tuple).unwrap();
assert_eq!(decoded_tuple, decoded);
let tuple = (val,);
let encoded = tuple.clone().encode();
let decoded = <(SomeType,)>::decode(&encoded).unwrap();
assert_eq!(tuple, decoded);
let wrapped =
ethers_core::abi::encode(&ethers_core::abi::Tokenize::into_tokens(tuple.clone())).to_vec();
assert_eq!(wrapped, encoded);
let decoded_wrapped = <(SomeType,)>::decode(&wrapped).unwrap();
assert_eq!(decoded_wrapped, tuple);
}
#[test]
fn can_derive_abi_codec_two_field() {
#[derive(Debug, Clone, PartialEq, Eq, EthAbiType, EthAbiCodec)]
pub struct SomeType {
inner: Vec<U256>,
addr: Address,
}
let val = SomeType { inner: Default::default(), addr: Default::default() };
let encoded = val.clone().encode();
let decoded = SomeType::decode(&encoded).unwrap();
assert_eq!(val, decoded);
let encoded_tuple = (Vec::<U256>::default(), Address::default()).encode();
assert_eq!(encoded_tuple, encoded);
let decoded_tuple = SomeType::decode(&encoded_tuple).unwrap();
assert_eq!(decoded_tuple, decoded);
let tuple = (val,);
let encoded = tuple.clone().encode();
let decoded = <(SomeType,)>::decode(&encoded).unwrap();
assert_eq!(tuple, decoded);
let wrapped =
ethers_core::abi::encode(&ethers_core::abi::Tokenize::into_tokens(tuple.clone())).to_vec();
assert_eq!(wrapped, encoded);
let decoded_wrapped = <(SomeType,)>::decode(&wrapped).unwrap();
assert_eq!(decoded_wrapped, tuple);
}
#[test]
fn can_derive_ethcall_for_bytes() {
#[derive(Clone, Debug, Default, Eq, PartialEq, EthCall, EthDisplay)]
#[ethcall(name = "batch", abi = "batch(bytes[],bool)")]
pub struct BatchCall {
pub calls: Vec<Bytes>,
pub revert_on_fail: bool,
}
assert_ethcall::<BatchCall>();
}
#[test]
fn can_derive_array_tuples() {
#[derive(Clone, Debug, Default, Eq, PartialEq, EthEvent, EthDisplay)]
#[ethevent(name = "DiamondCut", abi = "DiamondCut((address,uint8,bytes4[])[],address,bytes)")]
pub struct DiamondCutFilter {
pub diamond_cut: Vec<(Address, u8, Vec<[u8; 4]>)>,
pub init: Address,
pub calldata: Bytes,
}
}
#[test]
fn can_handle_abigen_tuples() {
#[derive(Clone, Debug, Default, Eq, PartialEq, EthCall, EthDisplay)]
#[ethcall(name = "swap", abi = "swap((uint8,uint8)[])")]
pub struct SwapCall {
pub pairs_to_swap: ::std::vec::Vec<(u8, u8)>,
}
}
#[test]
fn eth_display_works_on_ethers_bytes() {
#[derive(Clone, Debug, Default, Eq, PartialEq, EthCall, EthDisplay)]
#[ethcall(name = "logBytes", abi = "logBytes(bytes)")]
pub struct LogBytesCall {
pub p_0: ethers_core::types::Bytes,
}
let call = LogBytesCall { p_0: hex::decode(b"aaaaaa").unwrap().into() };
let s = format!("{call}");
assert_eq!(s, "0xaaaaaa");
}
#[test]
fn can_use_result_name() {
abigen!(
ResultContract,
r#"[
struct Result {uint256 result;}
result(Result result) (uint256)
]"#,
);
let _call = ResultCall { result: Result { result: U256::zero() } };
}
#[test]
fn can_derive_etherror() {
#[derive(Debug, PartialEq, Eq, EthError)]
#[etherror(name = "MyError", abi = "MyError(address,address,string)")]
struct MyError {
old_author: Address,
addr: Address,
new_value: String,
}
assert_eq!(MyError::abi_signature().as_ref(), "MyError(address,address,string)");
assert_tokenizeable::<MyError>();
assert_etherror::<MyError>();
}
#[test]
fn can_use_human_readable_error() {
abigen!(
ErrContract,
r#"[
error MyError(address,address,string)
]"#,
);
assert_etherror::<MyError>();
}
// <https://github.com/gakonst/ethers-rs/issues/2142>
#[test]
fn derives_abi_name() {
#[derive(Debug, EthEvent)]
#[ethevent(abi = "Transfer(address,address,uint256)")]
struct Erc20TransferEvent {
#[ethevent(indexed, name = "_from")]
from: Address,
#[ethevent(indexed, name = "_to")]
to: Address,
#[ethevent(name = "_value")]
value: U256,
}
assert_eq!(Erc20TransferEvent::abi_signature(), "Transfer(address,address,uint256)");
assert_eq!(
Erc20TransferEvent::signature(),
"0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef".parse().unwrap()
);
}
// <https://github.com/gakonst/ethers-rs/issues/2261>
#[test]
fn derive_empty_events() {
#[derive(Debug, EthEvent)]
#[ethevent(abi = "EmptyEvent()")]
struct EmptyEvent;
let log = RawLog { topics: vec![EmptyEvent::signature()], data: vec![] };
let _event = <EmptyEvent as EthLogDecode>::decode_log(&log).unwrap();
let log = RawLog { topics: vec![EmptyEvent::signature()], data: vec![0] };
assert!(<EmptyEvent as EthLogDecode>::decode_log(&log).is_err());
let log = RawLog { topics: vec![EmptyEvent::signature(), H256::random()], data: vec![0] };
assert!(<EmptyEvent as EthLogDecode>::decode_log(&log).is_err());
assert_eq!(EmptyEvent::abi_signature(), "EmptyEvent()");
abigen!(
DummyContract,
r#"[
event EmptyEvent2()
]"#,
);
assert_eq!(EmptyEvent2Filter::abi_signature(), "EmptyEvent2()");
}