reth_codecs/alloy/
withdrawal.rsuse crate::Compact;
use alloy_eips::eip4895::Withdrawal as AlloyWithdrawal;
use alloy_primitives::Address;
use reth_codecs_derive::add_arbitrary_tests;
#[derive(Debug, Clone, PartialEq, Eq, Default, Compact)]
#[cfg_attr(test, derive(arbitrary::Arbitrary, serde::Serialize, serde::Deserialize))]
#[add_arbitrary_tests(compact)]
pub(crate) struct Withdrawal {
index: u64,
validator_index: u64,
address: Address,
amount: u64,
}
impl Compact for AlloyWithdrawal {
fn to_compact<B>(&self, buf: &mut B) -> usize
where
B: bytes::BufMut + AsMut<[u8]>,
{
let withdrawal = Withdrawal {
index: self.index,
validator_index: self.validator_index,
address: self.address,
amount: self.amount,
};
withdrawal.to_compact(buf)
}
fn from_compact(buf: &[u8], len: usize) -> (Self, &[u8]) {
let (withdrawal, _) = Withdrawal::from_compact(buf, len);
let alloy_withdrawal = Self {
index: withdrawal.index,
validator_index: withdrawal.validator_index,
address: withdrawal.address,
amount: withdrawal.amount,
};
(alloy_withdrawal, buf)
}
}
#[cfg(test)]
mod tests {
use super::*;
use proptest::proptest;
use proptest_arbitrary_interop::arb;
proptest! {
#[test]
fn roundtrip(withdrawal in arb::<AlloyWithdrawal>()) {
let mut compacted_withdrawal = Vec::<u8>::new();
let len = withdrawal.to_compact(&mut compacted_withdrawal);
let (decoded, _) = AlloyWithdrawal::from_compact(&compacted_withdrawal, len);
assert_eq!(withdrawal, decoded)
}
}
#[test]
fn test_ensure_backwards_compatibility() {
assert_eq!(Withdrawal::bitflag_encoded_bytes(), 2);
}
}