reset chain when skipping fuzz loop

delvtech · Dec 17, 2024 · ee792ad · ee792ad
1 parent 119f54c
commit ee792ad
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/crates/hyperdrive-math/src/short/max.rs b/crates/hyperdrive-math/src/short/max.rs
@@ -903,7 +903,7 @@ mod tests {
 
             let min_share_reserves = state.calculate_min_share_reserves(checkpoint_exposure)?;
 
-            // Make sure a short is possible.
+            // Check that a short is possible.
             if state
                 .effective_share_reserves()?
                 .min(state.share_reserves())
@@ -992,6 +992,9 @@ mod tests {
             if state.effective_share_reserves()?
                 < state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                // Don't need to reset bob because he hasn't done anything.
                 continue;
             }
 
@@ -1110,6 +1113,10 @@ mod tests {
                     if state.effective_share_reserves()?
                         < state.calculate_min_share_reserves(checkpoint_exposure)?
                     {
+                        chain.revert(id).await?;
+                        alice.reset(Default::default()).await?;
+                        bob.reset(Default::default()).await?;
+                        celine.reset(Default::default()).await?;
                         continue;
                     }
 

diff --git a/crates/hyperdrive-math/src/short/open.rs b/crates/hyperdrive-math/src/short/open.rs
@@ -541,6 +541,10 @@ mod tests {
             if original_state.effective_share_reserves()?
                 < original_state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                bob.reset(Default::default()).await?;
+                celine.reset(Default::default()).await?;
                 continue;
             }
 
@@ -880,6 +884,9 @@ mod tests {
             if state.effective_share_reserves()?
                 < state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                bob.reset(Default::default()).await?;
                 continue;
             }
 
@@ -1022,6 +1029,9 @@ mod tests {
             if state.effective_share_reserves()?
                 < state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                bob.reset(Default::default()).await?;
                 continue;
             }
 
@@ -1193,6 +1203,10 @@ mod tests {
             if state.effective_share_reserves()?
                 < state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                bob.reset(Default::default()).await?;
+                celine.reset(Default::default()).await?;
                 continue;
             }
 

diff --git a/crates/hyperdrive-math/src/test_utils/preamble.rs b/crates/hyperdrive-math/src/test_utils/preamble.rs
@@ -287,6 +287,10 @@ mod tests {
             if state.effective_share_reserves()?
                 < state.calculate_min_share_reserves(checkpoint_exposure)?
             {
+                chain.revert(id).await?;
+                alice.reset(Default::default()).await?;
+                bob.reset(Default::default()).await?;
+                celine.reset(Default::default()).await?;
                 continue;
             }