WIP

AztecProtocol · Mar 6, 2024 · 0201509 · 0201509
1 parent 8600c07
commit 0201509
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/l1-contracts/test/NewInbox.t.sol b/l1-contracts/test/NewInbox.t.sol
@@ -196,8 +196,13 @@ contract NewInboxTest is Test {
 
     // Now we consume the trees
     for (uint256 i = 0; i < numTreesToConsume; i++) {
+      uint256 expectedNumTrees =
+        (inbox.getToConsume() + 1 == inbox.getInProgress()) ? numTrees + 1 : numTrees;
       bytes32 root = inbox.consume();
 
+      // We perform this check to verify that new tree initialization works as expected
+      assertEq(inbox.getNumTrees(), expectedNumTrees, "Unexptected number of trees");
+
       // We perform empty roots check only after first batch because after second one the following simple accounting
       // does not work
       if (isFirstRun) {