@@ -14,7 +14,6 @@ import (
1414 "github.com/ethereum/go-ethereum/params"
1515
1616 "github.com/offchainlabs/nitro/arbos"
17- "github.com/offchainlabs/nitro/arbos/burn"
1817 "github.com/offchainlabs/nitro/arbos/l2pricing"
1918 "github.com/offchainlabs/nitro/arbos/storage"
2019 "github.com/offchainlabs/nitro/solgen/go/precompilesgen"
@@ -103,14 +102,6 @@ func testRetryableRedeem(t *testing.T, evm *vm.EVM, precompileCtx *Context) {
103102 }
104103}
105104
106- func overrideStateArbOSVersion (evm * vm.EVM , arbosVersion uint64 ) error {
107- versionSlot := uint64 (0 )
108- version := new (big.Int ).SetUint64 (arbosVersion )
109- burner := burn .NewSystemBurner (nil , false )
110- sto := storage .NewGeth (evm .StateDB , burner )
111- return sto .SetByUint64 (versionSlot , common .BigToHash (version ))
112- }
113-
114105func TestRetryableRedeemLegacy (t * testing.T ) {
115106 evm := newMockEVMForTesting ()
116107 precompileCtx := testContext (common.Address {}, evm )
@@ -126,10 +117,8 @@ func TestRetryableRedeemLegacy(t *testing.T) {
126117}
127118
128119func TestRetryableRedeemLegacyArbOS60 (t * testing.T ) {
129- evm := newMockEVMForTesting ()
130- err := overrideStateArbOSVersion (evm , params .ArbosVersion_MultiGasConstraintsVersion )
131- Require (t , err )
132-
120+ arbosVersion := params .ArbosVersion_MultiGasConstraintsVersion
121+ evm := newMockEVMForTestingWithVersion (& arbosVersion )
133122 precompileCtx := testContext (common.Address {}, evm )
134123
135124 model , err := precompileCtx .State .L2PricingState ().GasModelToUse ()
@@ -165,10 +154,9 @@ func TestRetryableRedeemWithGasConstraints(t *testing.T) {
165154 testRetryableRedeem (t , evm , precompileCtx )
166155}
167156
168- func TestRetryableRedeemWithGasConstraintsArbOS60 (t * testing.T ) {
169- evm := newMockEVMForTesting ()
170- err := overrideStateArbOSVersion (evm , params .ArbosVersion_MultiGasConstraintsVersion )
171- Require (t , err )
157+ func TestRetryableRedeemWithGasConstraintsArbOSMultiGasConstraintsVersion (t * testing.T ) {
158+ arbosVersion := params .ArbosVersion_MultiGasConstraintsVersion
159+ evm := newMockEVMForTestingWithVersion (& arbosVersion )
172160
173161 precompileCtx := testContext (common.Address {}, evm )
174162
@@ -192,9 +180,8 @@ func TestRetryableRedeemWithGasConstraintsArbOS60(t *testing.T) {
192180}
193181
194182func TestRetryableRedeemWithMultiGasConstraints (t * testing.T ) {
195- evm := newMockEVMForTesting ()
196- err := overrideStateArbOSVersion (evm , params .ArbosVersion_MultiGasConstraintsVersion )
197- Require (t , err )
183+ arbosVersion := params .ArbosVersion_MultiGasConstraintsVersion
184+ evm := newMockEVMForTestingWithVersion (& arbosVersion )
198185
199186 precompileCtx := testContext (common.Address {}, evm )
200187
@@ -215,7 +202,7 @@ func TestRetryableRedeemWithMultiGasConstraints(t *testing.T) {
215202 uint8 (multigas .ResourceKindWasmComputation ): 6 ,
216203 }
217204
218- err = precompileCtx .State .L2PricingState ().AddMultiGasConstraint (target , window , backlog , weights )
205+ err : = precompileCtx .State .L2PricingState ().AddMultiGasConstraint (target , window , backlog , weights )
219206 Require (t , err )
220207 }
221208
0 commit comments