Return refs to the next middleware

crates/packet-forward/src/lib.rs

@@ -177,8 +177,18 @@ pub struct PacketForwardMiddleware<M> {
 }
 
 impl<M> PacketForwardMiddleware<M> {
+    /// Return an immutable ref to the next middleware.
+    pub fn next(&self) -> &M {
+        &self.next
+    }
+
+    /// Return a mutable ref to the next middleware.
+    pub fn next_mut(&mut self) -> &mut M {
+        &mut self.next
+    }
+
     /// Wrap an existing middleware in the PFM.
-    pub const fn next(next: M) -> Self {
+    pub const fn wrap(next: M) -> Self {
         Self { next }
     }
 }

crates/packet-forward/src/tests/utils.rs

@@ -456,7 +456,7 @@ fn failure_injection_err_msg(point: FailurePoint) -> String {
 pub type DummyPfm = PacketForwardMiddleware<Store<DummyTransferModule>>;
 
 pub fn get_dummy_pfm() -> DummyPfm {
-    PacketForwardMiddleware::next(Store::new(DummyTransferModule::new()))
+    PacketForwardMiddleware::wrap(Store::new(DummyTransferModule::new()))
 }
 
 pub fn get_dummy_coin(amount: u64) -> Coin<PrefixedDenom> {