minor fix

OffchainLabs · Oct 25, 2023 · 46287a1 · 46287a1
1 parent 17ce3cc
commit 46287a1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/system_tests/estimation_test.go b/system_tests/estimation_test.go
@@ -141,6 +141,10 @@ func TestDifficultyForLatestArbOS(t *testing.T) {
 	_, _, simple, err := mocksgen.DeploySimple(&auth, client)
 	Require(t, err, "could not deploy contract")
 
+	tx, err := simple.StoreDifficulty(&auth)
+	Require(t, err)
+	_, err = EnsureTxSucceeded(ctx, client, tx)
+	Require(t, err)
 	difficulty, err := simple.GetBlockDifficulty(&bind.CallOpts{})
 	Require(t, err)
 	if !arbmath.BigEquals(difficulty, common.Big1) {
@@ -163,6 +167,10 @@ func TestDifficultyForArbOSTen(t *testing.T) {
 	_, _, simple, err := mocksgen.DeploySimple(&auth, client)
 	Require(t, err, "could not deploy contract")
 
+	tx, err := simple.StoreDifficulty(&auth)
+	Require(t, err)
+	_, err = EnsureTxSucceeded(ctx, client, tx)
+	Require(t, err)
 	difficulty, err := simple.GetBlockDifficulty(&bind.CallOpts{})
 	Require(t, err)
 	if !arbmath.BigEquals(difficulty, common.Big1) {