Fixes reorg_requesequencing_test

OffchainLabs · Dec 24, 2024 · 824786c · 824786c
1 parent eed9c05
commit 824786c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/system_tests/reorg_resequencing_test.go b/system_tests/reorg_resequencing_test.go
@@ -65,11 +65,11 @@ func TestReorgResequencing(t *testing.T) {
 	err = builder.L2.ConsensusNode.TxStreamer.ReorgTo(startMsgCount)
 	Require(t, err)
 
-	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t).Await(ctx)
+	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t)
 	Require(t, err)
 
 	verifyBalances("after empty reorg")
-	compareAllMsgResultsFromConsensusAndExecution(t, builder.L2, "after empty reorg")
+	compareAllMsgResultsFromConsensusAndExecution(t, ctx, builder.L2, "after empty reorg")
 
 	prevMessage, err := builder.L2.ConsensusNode.TxStreamer.GetMessage(startMsgCount - 1)
 	Require(t, err)
@@ -91,20 +91,20 @@ func TestReorgResequencing(t *testing.T) {
 	}})
 	Require(t, err)
 
-	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t).Await(ctx)
+	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t)
 	Require(t, err)
 
 	accountsWithBalance = append(accountsWithBalance, "User4")
 
 	verifyBalances("after reorg with new deposit")
-	compareAllMsgResultsFromConsensusAndExecution(t, builder.L2, "after reorg with new deposit")
+	compareAllMsgResultsFromConsensusAndExecution(t, ctx, builder.L2, "after reorg with new deposit")
 
 	err = builder.L2.ConsensusNode.TxStreamer.ReorgTo(startMsgCount)
 	Require(t, err)
 
-	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t).Await(ctx)
+	_, err = builder.L2.ExecNode.ExecEngine.HeadMessageNumberSync(t)
 	Require(t, err)
 
 	verifyBalances("after second empty reorg")
-	compareAllMsgResultsFromConsensusAndExecution(t, builder.L2, "after second empty reorg")
+	compareAllMsgResultsFromConsensusAndExecution(t, ctx, builder.L2, "after second empty reorg")
 }