keep deprecatedContract behavior for previously deployed contracts

ava-labs · Sep 26, 2024 · c4ea40c · c4ea40c
1 parent 311dbd9
commit c4ea40c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/core/state_transition_test.go b/core/state_transition_test.go
@@ -142,7 +142,7 @@ func TestNativeAssetContractCall(t *testing.T) {
 		makeTx(1, contractAddr, common.Big0, 100_000, big.NewInt(params.LaunchMinGasPrice), nil), // No input data is necessary, since this will hit the contract's fallback function.
 	}
 
-	phase6Tests := map[string]stateTransitionTest{
+	tests := map[string]stateTransitionTest{
 		"phase5": {
 			config:  params.TestApricotPhase5Config,
 			txs:     txs,
@@ -167,10 +167,23 @@ func TestNativeAssetContractCall(t *testing.T) {
 			gasUsed: []uint64{132091, 21618},
 			want:    "",
 		},
+		"durango": {
+			config:  params.TestDurangoChainConfig,
+			txs:     txs,
+			gasUsed: []uint64{132117, 21618},
+			want:    "",
+		},
+		"etna": {
+			config:  params.TestEtnaChainConfig,
+			txs:     txs,
+			gasUsed: []uint64{132117, 21618},
+			want:    "",
+		},
 	}
 
-	for name, stTest := range phase6Tests {
+	for name, stTest := range tests {
 		t.Run(name, func(t *testing.T) {
+			stTest.config.SetEthUpgrades()
 			executeStateTransitionTest(t, stTest)
 		})
 	}

diff --git a/core/vm/contracts.go b/core/vm/contracts.go
@@ -172,6 +172,9 @@ var PrecompiledContractsCancun = map[common.Address]contract.StatefulPrecompiled
 	common.BytesToAddress([]byte{8}):    newWrappedPrecompiledContract(&bn256PairingIstanbul{}),
 	common.BytesToAddress([]byte{9}):    newWrappedPrecompiledContract(&blake2F{}),
 	common.BytesToAddress([]byte{0x0a}): newWrappedPrecompiledContract(&kzgPointEvaluation{}),
+	genesisContractAddr:                 &deprecatedContract{},
+	NativeAssetBalanceAddr:              &deprecatedContract{},
+	NativeAssetCallAddr:                 &deprecatedContract{},
 }
 
 // PrecompiledContractsBLS contains the set of pre-compiled Ethereum