chore(sdk): Add MaybeArbitrary as super trait

crates/optimism/primitives/src/tx_type.rs

@@ -18,6 +18,14 @@ use reth_primitives_traits::{InMemorySize, TxType};
 #[into(u8)]
 pub struct OpTxType(AlloyOpTxType);
 
+impl<'a> reth_primitives::arbitrary::Arbitrary<'a> for OpTxType {
+    fn arbitrary(
+        u: &mut reth_primitives::arbitrary::Unstructured<'a>,
+    ) -> reth_primitives::arbitrary::Result<Self> {
+        Ok(AlloyOpTxType::arbitrary(u)?.into())
+    }
+}
+
 impl TxType for OpTxType {
     #[inline]
     fn is_legacy(&self) -> bool {

crates/primitives-traits/src/block/body.rs

@@ -4,7 +4,7 @@ use alloc::fmt;
 
 use alloy_consensus::Transaction;
 
-use crate::{FullSignedTx, InMemorySize, MaybeSerde};
+use crate::{FullSignedTx, InMemorySize, MaybeArbitrary, MaybeSerde};
 
 /// Helper trait that unifies all behaviour required by transaction to support full node operations.
 pub trait FullBlockBody: BlockBody<Transaction: FullSignedTx> {}
@@ -26,6 +26,7 @@ pub trait BlockBody:
     + alloy_rlp::Decodable
     + InMemorySize
     + MaybeSerde
+    + MaybeArbitrary
 {
     /// Ordered list of signed transactions as committed in block.
     // todo: requires trait for signed transaction

crates/primitives-traits/src/block/header.rs

@@ -5,7 +5,7 @@ use core::fmt;
 use alloy_primitives::Sealable;
 use reth_codecs::Compact;
 
-use crate::{InMemorySize, MaybeSerde};
+use crate::{InMemorySize, MaybeArbitrary, MaybeSerde};
 
 /// Helper trait that unifies all behaviour required by block header to support full node
 /// operations.
@@ -29,6 +29,7 @@ pub trait BlockHeader:
     + Sealable
     + InMemorySize
     + MaybeSerde
+    + MaybeArbitrary
 {
 }
 
@@ -47,5 +48,6 @@ impl<T> BlockHeader for T where
         + Sealable
         + InMemorySize
         + MaybeSerde
+        + MaybeArbitrary
 {
 }

crates/primitives-traits/src/block/mod.rs

@@ -7,7 +7,9 @@ use alloc::fmt;
 
 use alloy_rlp::{Decodable, Encodable};
 
-use crate::{BlockHeader, FullBlockBody, FullBlockHeader, InMemorySize, MaybeSerde};
+use crate::{
+    BlockHeader, FullBlockBody, FullBlockHeader, InMemorySize, MaybeArbitrary, MaybeSerde,
+};
 
 /// Helper trait that unifies all behaviour required by block to support full node operations.
 pub trait FullBlock:
@@ -26,7 +28,17 @@ impl<T> FullBlock for T where
 // senders
 #[auto_impl::auto_impl(&, Arc)]
 pub trait Block:
-    Send + Sync + Unpin + Clone + Default + fmt::Debug + PartialEq + Eq + InMemorySize + MaybeSerde
+    Send
+    + Sync
+    + Unpin
+    + Clone
+    + Default
+    + fmt::Debug
+    + PartialEq
+    + Eq
+    + InMemorySize
+    + MaybeSerde
+    + MaybeArbitrary
 {
     /// Header part of the block.
     type Header: BlockHeader + 'static;

crates/primitives-traits/src/header/sealed.rs

@@ -1,4 +1,4 @@
-use crate::InMemorySize;
+use crate::{InMemorySize, MaybeArbitrary};
 pub use alloy_consensus::Header;
 use alloy_consensus::Sealed;
 use alloy_eips::BlockNumHash;
@@ -156,9 +156,12 @@ impl<H> From<SealedHeader<H>> for Sealed<H> {
 }
 
 #[cfg(any(test, feature = "arbitrary"))]
-impl<'a> arbitrary::Arbitrary<'a> for SealedHeader {
+impl<'a, H> arbitrary::Arbitrary<'a> for SealedHeader<H>
+where
+    H: MaybeArbitrary + Sealable,
+{
     fn arbitrary(u: &mut arbitrary::Unstructured<'a>) -> arbitrary::Result<Self> {
-        let header = Header::arbitrary(u)?;
+        let header = H::arbitrary(u)?;
 
         Ok(Self::seal(header))
     }

crates/primitives-traits/src/receipt.rs

@@ -1,6 +1,6 @@
 //! Receipt abstraction
 
-use crate::{InMemorySize, MaybeSerde};
+use crate::{InMemorySize, MaybeSerde, MaybeArbitrary};
 use alloc::vec::Vec;
 use alloy_consensus::TxReceipt;
 use alloy_primitives::B256;
@@ -26,6 +26,7 @@ pub trait Receipt:
     + alloy_rlp::Decodable
     + MaybeSerde
     + InMemorySize
+    + MaybeArbitrary
 {
     /// Returns transaction type.
     fn tx_type(&self) -> u8;

crates/primitives-traits/src/tx_type.rs

@@ -3,7 +3,7 @@ use core::fmt;
 use alloy_primitives::{U64, U8};
 use reth_codecs::Compact;
 
-use crate::InMemorySize;
+use crate::{InMemorySize, MaybeArbitrary};
 
 /// Helper trait that unifies all behaviour required by transaction type ID to support full node
 /// operations.
@@ -32,6 +32,7 @@ pub trait TxType:
     + alloy_rlp::Encodable
     + alloy_rlp::Decodable
     + InMemorySize
+    + MaybeArbitrary
 {
     /// Returns `true` if this is a legacy transaction.
     fn is_legacy(&self) -> bool;