git.delta.rocks / unique-network / refs/commits / 7eec33a3cc63

1use evm_coder_procedural::AbiCoder;2use evm_coder::{3	types::*,4	abi::{AbiType, AbiRead, AbiWrite, AbiReader, AbiWriter},5};67// TODO: move to build_failed tests8// #[derive(AbiCoder, PartialEq, Debug)]9// struct TypeStructUnit {}1011#[derive(AbiCoder, PartialEq, Debug)]12struct TypeStruct1SimpleParam {13	_a: u8,14}1516#[derive(AbiCoder, PartialEq, Debug)]17struct TypeStruct1DynamicParam {18	_a: String,19}2021#[derive(AbiCoder, PartialEq, Debug)]22struct TypeStruct2SimpleParam {23	_a: u8,24	_b: u32,25}2627#[derive(AbiCoder, PartialEq, Debug)]28struct TypeStruct2DynamicParam {29	_a: String,30	_b: bytes,31}3233#[derive(AbiCoder, PartialEq, Debug)]34struct TypeStruct2MixedParam {35	_a: u8,36	_b: bytes,37}3839#[derive(AbiCoder, PartialEq, Debug)]40struct TypeStruct1DerivedSimpleParam {41	_a: TypeStruct1SimpleParam,42}4344#[derive(AbiCoder, PartialEq, Debug)]45struct TypeStruct2DerivedSimpleParam {46	_a: TypeStruct1SimpleParam,47	_b: TypeStruct2SimpleParam,48}4950#[derive(AbiCoder, PartialEq, Debug)]51struct TypeStruct1DerivedDynamicParam {52	_a: TypeStruct1DynamicParam,53}5455#[derive(AbiCoder, PartialEq, Debug)]56struct TypeStruct2DerivedDynamicParam {57	_a: TypeStruct1DynamicParam,58	_b: TypeStruct2DynamicParam,59}6061#[derive(AbiCoder, PartialEq, Debug)]62struct TypeStruct3DerivedMixedParam {63	_a: TypeStruct1SimpleParam,64	_b: TypeStruct2DynamicParam,65	_c: TypeStruct2MixedParam,66}6768#[test]69fn empty() {}7071#[test]72fn impl_abi_type_signature() {73	assert_eq!(74		<TypeStruct1SimpleParam as AbiType>::SIGNATURE75			.as_str()76			.unwrap(),77		"(uint8)"78	);79	assert_eq!(80		<TypeStruct1DynamicParam as AbiType>::SIGNATURE81			.as_str()82			.unwrap(),83		"(string)"84	);85	assert_eq!(86		<TypeStruct2SimpleParam as AbiType>::SIGNATURE87			.as_str()88			.unwrap(),89		"(uint8,uint32)"90	);91	assert_eq!(92		<TypeStruct2DynamicParam as AbiType>::SIGNATURE93			.as_str()94			.unwrap(),95		"(string,bytes)"96	);97	assert_eq!(98		<TypeStruct2MixedParam as AbiType>::SIGNATURE99			.as_str()100			.unwrap(),101		"(uint8,bytes)"102	);103	assert_eq!(104		<TypeStruct1DerivedSimpleParam as AbiType>::SIGNATURE105			.as_str()106			.unwrap(),107		"((uint8))"108	);109	assert_eq!(110		<TypeStruct2DerivedSimpleParam as AbiType>::SIGNATURE111			.as_str()112			.unwrap(),113		"((uint8),(uint8,uint32))"114	);115	assert_eq!(116		<TypeStruct1DerivedDynamicParam as AbiType>::SIGNATURE117			.as_str()118			.unwrap(),119		"((string))"120	);121	assert_eq!(122		<TypeStruct2DerivedDynamicParam as AbiType>::SIGNATURE123			.as_str()124			.unwrap(),125		"((string),(string,bytes))"126	);127	assert_eq!(128		<TypeStruct3DerivedMixedParam as AbiType>::SIGNATURE129			.as_str()130			.unwrap(),131		"((uint8),(string,bytes),(uint8,bytes))"132	);133}134135#[test]136fn impl_abi_type_is_dynamic() {137	assert_eq!(<TypeStruct1SimpleParam as AbiType>::is_dynamic(), false);138	assert_eq!(<TypeStruct1DynamicParam as AbiType>::is_dynamic(), true);139	assert_eq!(<TypeStruct2SimpleParam as AbiType>::is_dynamic(), false);140	assert_eq!(<TypeStruct2DynamicParam as AbiType>::is_dynamic(), true);141	assert_eq!(<TypeStruct2MixedParam as AbiType>::is_dynamic(), true);142	assert_eq!(143		<TypeStruct1DerivedSimpleParam as AbiType>::is_dynamic(),144		false145	);146	assert_eq!(147		<TypeStruct2DerivedSimpleParam as AbiType>::is_dynamic(),148		false149	);150	assert_eq!(151		<TypeStruct1DerivedDynamicParam as AbiType>::is_dynamic(),152		true153	);154	assert_eq!(155		<TypeStruct2DerivedDynamicParam as AbiType>::is_dynamic(),156		true157	);158	assert_eq!(159		<TypeStruct3DerivedMixedParam as AbiType>::is_dynamic(),160		true161	);162}163164#[test]165fn impl_abi_type_size() {166	const ABI_ALIGNMENT: usize = 32;167	assert_eq!(<TypeStruct1SimpleParam as AbiType>::size(), ABI_ALIGNMENT);168	assert_eq!(<TypeStruct1DynamicParam as AbiType>::size(), ABI_ALIGNMENT);169	assert_eq!(170		<TypeStruct2SimpleParam as AbiType>::size(),171		ABI_ALIGNMENT * 2172	);173	assert_eq!(174		<TypeStruct2DynamicParam as AbiType>::size(),175		ABI_ALIGNMENT * 2176	);177	assert_eq!(178		<TypeStruct2MixedParam as AbiType>::size(),179		ABI_ALIGNMENT * 2180	);181	assert_eq!(182		<TypeStruct1DerivedSimpleParam as AbiType>::size(),183		ABI_ALIGNMENT184	);185	assert_eq!(186		<TypeStruct2DerivedSimpleParam as AbiType>::size(),187		ABI_ALIGNMENT * 3188	);189	assert_eq!(190		<TypeStruct1DerivedDynamicParam as AbiType>::size(),191		ABI_ALIGNMENT192	);193	assert_eq!(194		<TypeStruct2DerivedDynamicParam as AbiType>::size(),195		ABI_ALIGNMENT * 3196	);197	assert_eq!(198		<TypeStruct3DerivedMixedParam as AbiType>::size(),199		ABI_ALIGNMENT * 5200	);201}202203#[derive(AbiCoder, PartialEq, Debug)]204struct TupleStruct1SimpleParam(u8);205206#[derive(AbiCoder, PartialEq, Debug)]207struct TupleStruct1DynamicParam(String);208209#[derive(AbiCoder, PartialEq, Debug)]210struct TupleStruct2SimpleParam(u8, u32);211212#[derive(AbiCoder, PartialEq, Debug)]213struct TupleStruct2DynamicParam(String, bytes);214215#[derive(AbiCoder, PartialEq, Debug)]216struct TupleStruct2MixedParam(u8, bytes);217218#[derive(AbiCoder, PartialEq, Debug)]219struct TupleStruct1DerivedSimpleParam(TupleStruct1SimpleParam);220221#[derive(AbiCoder, PartialEq, Debug)]222struct TupleStruct2DerivedSimpleParam(TupleStruct1SimpleParam, TupleStruct2SimpleParam);223224#[derive(AbiCoder, PartialEq, Debug)]225struct TupleStruct1DerivedDynamicParam(TupleStruct1DynamicParam);226227#[derive(AbiCoder, PartialEq, Debug)]228struct TupleStruct2DerivedDynamicParam(TupleStruct1DynamicParam, TupleStruct2DynamicParam);229230#[derive(AbiCoder, PartialEq, Debug)]231struct TupleStruct3DerivedMixedParam(232	TupleStruct1SimpleParam,233	TupleStruct2DynamicParam,234	TupleStruct2MixedParam,235);236237#[test]238fn impl_abi_type_signature_same_for_structs() {239	assert_eq!(240		<TypeStruct1SimpleParam as AbiType>::SIGNATURE241			.as_str()242			.unwrap(),243		<TupleStruct1SimpleParam as AbiType>::SIGNATURE244			.as_str()245			.unwrap()246	);247	assert_eq!(248		<TypeStruct1DynamicParam as AbiType>::SIGNATURE249			.as_str()250			.unwrap(),251		<TupleStruct1DynamicParam as AbiType>::SIGNATURE252			.as_str()253			.unwrap()254	);255	assert_eq!(256		<TypeStruct2SimpleParam as AbiType>::SIGNATURE257			.as_str()258			.unwrap(),259		<TupleStruct2SimpleParam as AbiType>::SIGNATURE260			.as_str()261			.unwrap()262	);263	assert_eq!(264		<TypeStruct2DynamicParam as AbiType>::SIGNATURE265			.as_str()266			.unwrap(),267		<TupleStruct2DynamicParam as AbiType>::SIGNATURE268			.as_str()269			.unwrap()270	);271	assert_eq!(272		<TypeStruct2MixedParam as AbiType>::SIGNATURE273			.as_str()274			.unwrap(),275		<TupleStruct2MixedParam as AbiType>::SIGNATURE276			.as_str()277			.unwrap(),278	);279	assert_eq!(280		<TypeStruct1DerivedSimpleParam as AbiType>::SIGNATURE281			.as_str()282			.unwrap(),283		<TupleStruct1DerivedSimpleParam as AbiType>::SIGNATURE284			.as_str()285			.unwrap(),286	);287	assert_eq!(288		<TypeStruct2DerivedSimpleParam as AbiType>::SIGNATURE289			.as_str()290			.unwrap(),291		<TupleStruct2DerivedSimpleParam as AbiType>::SIGNATURE292			.as_str()293			.unwrap(),294	);295	assert_eq!(296		<TypeStruct1DerivedDynamicParam as AbiType>::SIGNATURE297			.as_str()298			.unwrap(),299		<TupleStruct1DerivedDynamicParam as AbiType>::SIGNATURE300			.as_str()301			.unwrap(),302	);303	assert_eq!(304		<TypeStruct2DerivedDynamicParam as AbiType>::SIGNATURE305			.as_str()306			.unwrap(),307		<TupleStruct2DerivedDynamicParam as AbiType>::SIGNATURE308			.as_str()309			.unwrap(),310	);311	assert_eq!(312		<TypeStruct3DerivedMixedParam as AbiType>::SIGNATURE313			.as_str()314			.unwrap(),315		<TupleStruct3DerivedMixedParam as AbiType>::SIGNATURE316			.as_str()317			.unwrap(),318	);319}320321#[test]322fn impl_abi_type_is_dynamic_same_for_structs() {323	assert_eq!(324		<TypeStruct1SimpleParam as AbiType>::is_dynamic(),325		<TupleStruct1SimpleParam as AbiType>::is_dynamic()326	);327	assert_eq!(328		<TypeStruct1DynamicParam as AbiType>::is_dynamic(),329		<TupleStruct1DynamicParam as AbiType>::is_dynamic()330	);331	assert_eq!(332		<TypeStruct2SimpleParam as AbiType>::is_dynamic(),333		<TupleStruct2SimpleParam as AbiType>::is_dynamic()334	);335	assert_eq!(336		<TypeStruct2DynamicParam as AbiType>::is_dynamic(),337		<TupleStruct2DynamicParam as AbiType>::is_dynamic()338	);339	assert_eq!(340		<TypeStruct2MixedParam as AbiType>::is_dynamic(),341		<TupleStruct2MixedParam as AbiType>::is_dynamic()342	);343	assert_eq!(344		<TypeStruct1DerivedSimpleParam as AbiType>::is_dynamic(),345		<TupleStruct1DerivedSimpleParam as AbiType>::is_dynamic()346	);347	assert_eq!(348		<TypeStruct2DerivedSimpleParam as AbiType>::is_dynamic(),349		<TupleStruct2DerivedSimpleParam as AbiType>::is_dynamic()350	);351	assert_eq!(352		<TypeStruct1DerivedDynamicParam as AbiType>::is_dynamic(),353		<TupleStruct1DerivedDynamicParam as AbiType>::is_dynamic()354	);355	assert_eq!(356		<TypeStruct2DerivedDynamicParam as AbiType>::is_dynamic(),357		<TupleStruct2DerivedDynamicParam as AbiType>::is_dynamic()358	);359	assert_eq!(360		<TypeStruct3DerivedMixedParam as AbiType>::is_dynamic(),361		<TupleStruct3DerivedMixedParam as AbiType>::is_dynamic()362	);363}364365#[test]366fn impl_abi_type_size_same_for_structs() {367	assert_eq!(368		<TypeStruct1SimpleParam as AbiType>::size(),369		<TupleStruct1SimpleParam as AbiType>::size()370	);371	assert_eq!(372		<TypeStruct1DynamicParam as AbiType>::size(),373		<TupleStruct1DynamicParam as AbiType>::size()374	);375	assert_eq!(376		<TypeStruct2SimpleParam as AbiType>::size(),377		<TupleStruct2SimpleParam as AbiType>::size()378	);379	assert_eq!(380		<TypeStruct2DynamicParam as AbiType>::size(),381		<TupleStruct2DynamicParam as AbiType>::size()382	);383	assert_eq!(384		<TypeStruct2MixedParam as AbiType>::size(),385		<TupleStruct2MixedParam as AbiType>::size()386	);387	assert_eq!(388		<TypeStruct1DerivedSimpleParam as AbiType>::size(),389		<TupleStruct1DerivedSimpleParam as AbiType>::size()390	);391	assert_eq!(392		<TypeStruct2DerivedSimpleParam as AbiType>::size(),393		<TupleStruct2DerivedSimpleParam as AbiType>::size()394	);395	assert_eq!(396		<TypeStruct1DerivedDynamicParam as AbiType>::size(),397		<TupleStruct1DerivedDynamicParam as AbiType>::size()398	);399	assert_eq!(400		<TypeStruct2DerivedDynamicParam as AbiType>::size(),401		<TupleStruct2DerivedDynamicParam as AbiType>::size()402	);403	assert_eq!(404		<TypeStruct3DerivedMixedParam as AbiType>::size(),405		<TupleStruct3DerivedMixedParam as AbiType>::size()406	);407}408409const FUNCTION_IDENTIFIER: u32 = 0xdeadbeef;410411fn test_impl<Tuple, TupleStruct, TypeStruct>(412	tuple_data: Tuple,413	tuple_struct_data: TupleStruct,414	type_struct_data: TypeStruct,415) where416	TypeStruct: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,417	TupleStruct: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,418	Tuple: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,419{420	let encoded_type_struct = test_abi_write_impl(&type_struct_data);421	let encoded_tuple_struct = test_abi_write_impl(&tuple_struct_data);422	let encoded_tuple = test_abi_write_impl(&tuple_data);423424	similar_asserts::assert_eq!(encoded_tuple, encoded_type_struct);425	similar_asserts::assert_eq!(encoded_tuple, encoded_tuple_struct);426427	// dbg!(&encoded_tuple);428	// dbg!(&encoded_tuple_struct);429	// dbg!(&encoded_type_struct);430431	{432		let (_, mut decoder) = AbiReader::new_call(&encoded_tuple).unwrap();433		let restored_struct_data = <TypeStruct>::abi_read(&mut decoder).unwrap();434		assert_eq!(restored_struct_data, type_struct_data);435	}436	{437		let (_, mut decoder) = AbiReader::new_call(&encoded_tuple).unwrap();438		let restored_struct_data = <TupleStruct>::abi_read(&mut decoder).unwrap();439		assert_eq!(restored_struct_data, tuple_struct_data);440	}441442	{443		let (_, mut decoder) = AbiReader::new_call(&encoded_type_struct).unwrap();444		let restored_tuple_data = <Tuple>::abi_read(&mut decoder).unwrap();445		assert_eq!(restored_tuple_data, tuple_data);446	}447	{448		let (_, mut decoder) = AbiReader::new_call(&encoded_tuple_struct).unwrap();449		let restored_tuple_data = <Tuple>::abi_read(&mut decoder).unwrap();450		assert_eq!(restored_tuple_data, tuple_data);451	}452}453454fn test_abi_write_impl<A>(data: &A) -> Vec<u8>455where456	A: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,457{458	let mut writer = AbiWriter::new_call(FUNCTION_IDENTIFIER);459	data.abi_write(&mut writer);460	let encoded_tuple = writer.finish();461	encoded_tuple462}463464#[test]465fn codec_struct_1_simple() {466	let _a = 0xff;467	test_impl::<(uint8,), TupleStruct1SimpleParam, TypeStruct1SimpleParam>(468		(_a,),469		TupleStruct1SimpleParam(_a),470		TypeStruct1SimpleParam { _a },471	);472}473474#[test]475fn codec_struct_1_dynamic() {476	let _a: String = "some string".into();477	test_impl::<(String,), TupleStruct1DynamicParam, TypeStruct1DynamicParam>(478		(_a.clone(),),479		TupleStruct1DynamicParam(_a.clone()),480		TypeStruct1DynamicParam { _a },481	);482}483484#[test]485fn codec_struct_1_derived_simple() {486	let _a: u8 = 0xff;487	test_impl::<((u8,),), TupleStruct1DerivedSimpleParam, TypeStruct1DerivedSimpleParam>(488		((_a,),),489		TupleStruct1DerivedSimpleParam(TupleStruct1SimpleParam(_a)),490		TypeStruct1DerivedSimpleParam {491			_a: TypeStruct1SimpleParam { _a },492		},493	);494}495496#[test]497fn codec_struct_1_derived_dynamic() {498	let _a: String = "some string".into();499	test_impl::<((String,),), TupleStruct1DerivedDynamicParam, TypeStruct1DerivedDynamicParam>(500		((_a.clone(),),),501		TupleStruct1DerivedDynamicParam(TupleStruct1DynamicParam(_a.clone())),502		TypeStruct1DerivedDynamicParam {503			_a: TypeStruct1DynamicParam { _a },504		},505	);506}507508#[test]509fn codec_struct_2_simple() {510	let _a = 0xff;511	let _b = 0xbeefbaba;512	test_impl::<(u8, u32), TupleStruct2SimpleParam, TypeStruct2SimpleParam>(513		(_a, _b),514		TupleStruct2SimpleParam(_a, _b),515		TypeStruct2SimpleParam { _a, _b },516	);517}518519#[test]520fn codec_struct_2_dynamic() {521	let _a: String = "some string".into();522	let _b: bytes = bytes(vec![0x11, 0x22, 0x33]);523	test_impl::<(String, bytes), TupleStruct2DynamicParam, TypeStruct2DynamicParam>(524		(_a.clone(), _b.clone()),525		TupleStruct2DynamicParam(_a.clone(), _b.clone()),526		TypeStruct2DynamicParam { _a, _b },527	);528}529530#[test]531fn codec_struct_2_mixed() {532	let _a: u8 = 0xff;533	let _b: bytes = bytes(vec![0x11, 0x22, 0x33]);534	test_impl::<(u8, bytes), TupleStruct2MixedParam, TypeStruct2MixedParam>(535		(_a.clone(), _b.clone()),536		TupleStruct2MixedParam(_a.clone(), _b.clone()),537		TypeStruct2MixedParam { _a, _b },538	);539}540541#[test]542fn codec_struct_2_derived_simple() {543	let _a = 0xff;544	let _b = 0xbeefbaba;545	test_impl::<((u8,), (u8, u32)), TupleStruct2DerivedSimpleParam, TypeStruct2DerivedSimpleParam>(546		((_a,), (_a, _b)),547		TupleStruct2DerivedSimpleParam(548			TupleStruct1SimpleParam(_a),549			TupleStruct2SimpleParam(_a, _b),550		),551		TypeStruct2DerivedSimpleParam {552			_a: TypeStruct1SimpleParam { _a },553			_b: TypeStruct2SimpleParam { _a, _b },554		},555	);556}557558#[test]559fn codec_struct_2_derived_dynamic() {560	let _a = "some string".to_string();561	let _b = bytes(vec![0x11, 0x22, 0x33]);562	test_impl::<563		((String,), (String, bytes)),564		TupleStruct2DerivedDynamicParam,565		TypeStruct2DerivedDynamicParam,566	>(567		((_a.clone(),), (_a.clone(), _b.clone())),568		TupleStruct2DerivedDynamicParam(569			TupleStruct1DynamicParam(_a.clone()),570			TupleStruct2DynamicParam(_a.clone(), _b.clone()),571		),572		TypeStruct2DerivedDynamicParam {573			_a: TypeStruct1DynamicParam { _a: _a.clone() },574			_b: TypeStruct2DynamicParam { _a, _b },575		},576	);577}578579#[test]580fn codec_struct_3_derived_mixed() {581	let int = 0xff;582	let by = bytes(vec![0x11, 0x22, 0x33]);583	let string = "some string".to_string();584	test_impl::<585		((u8,), (String, bytes), (u8, bytes)),586		TupleStruct3DerivedMixedParam,587		TypeStruct3DerivedMixedParam,588	>(589		((int,), (string.clone(), by.clone()), (int, by.clone())),590		TupleStruct3DerivedMixedParam(591			TupleStruct1SimpleParam(int),592			TupleStruct2DynamicParam(string.clone(), by.clone()),593			TupleStruct2MixedParam(int, by.clone()),594		),595		TypeStruct3DerivedMixedParam {596			_a: TypeStruct1SimpleParam { _a: int },597			_b: TypeStruct2DynamicParam {598				_a: string.clone(),599				_b: by.clone(),600			},601			_c: TypeStruct2MixedParam { _a: int, _b: by },602		},603	);604}