mirror of
https://github.com/luxfi/cli.git
synced 2026-10-30 17:01:20 +00:00
zeekay
96715881cb
Update CLI for genesis Fix import paths from ethereum/go-ethereum to luxfi/geth - Replace all ethereum/go-ethereum imports with luxfi/geth - Add local rawdb package for unexported constants - Fix build errors in genesis tools - Add comprehensive migration documentation - Create build-all.sh script for building all tools - Clean up duplicate functions and unused imports Fix remaining imports Update CLI Remove symlink Remove local replace directives Update Update old templates
266 lines
8.8 KiB
Go
266 lines
8.8 KiB
Go
// Copyright (C) 2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
package upgradecmd
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEarliestTimestamp(t *testing.T) {
|
|
type testRun struct {
|
|
name string
|
|
upgradesFile []byte
|
|
earliest int64
|
|
expectedErr error
|
|
}
|
|
|
|
targetEarliest := time.Now().Add(1 * time.Minute).Unix()
|
|
tests := []testRun{
|
|
{
|
|
name: "only one",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
targetEarliest,
|
|
)),
|
|
earliest: targetEarliest,
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "there are two",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"contractNativeMinterConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
targetEarliest,
|
|
time.Now().Add(1*time.Minute).Add(2*time.Second).Unix(),
|
|
)),
|
|
earliest: targetEarliest,
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "three with second earliest",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"contractNativeMinterConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"txAllowListConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
time.Now().Add(1*time.Minute).Add(2*time.Second).Unix(),
|
|
targetEarliest,
|
|
time.Now().Add(1*time.Minute).Add(4*time.Second).Unix(),
|
|
)),
|
|
earliest: targetEarliest,
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "three with third earliest",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"txAllowListConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"contractNativeMinterConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
time.Now().Add(1*time.Minute).Add(2*time.Second).Unix(),
|
|
time.Now().Add(1*time.Minute).Add(4*time.Second).Unix(),
|
|
targetEarliest,
|
|
)),
|
|
earliest: targetEarliest,
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "no upcoming",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"txAllowListConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"contractNativeMinterConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
time.Now().Add(10*time.Millisecond).Unix(),
|
|
time.Now().Add(15*time.Millisecond).Unix(),
|
|
time.Now().Add(20*time.Millisecond).Unix(),
|
|
)),
|
|
earliest: targetEarliest,
|
|
expectedErr: errNoUpcomingUpgrades,
|
|
},
|
|
}
|
|
|
|
require := require.New(t)
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(_ *testing.T) {
|
|
upgrades, err := getAllUpgrades(tt.upgradesFile)
|
|
require.NoError(err)
|
|
|
|
// give some time so timestamps are def before now
|
|
time.Sleep(1 * time.Second)
|
|
|
|
earliest, err := getEarliestUpcomingTimestamp(upgrades)
|
|
if tt.expectedErr != nil {
|
|
require.ErrorIs(err, tt.expectedErr)
|
|
} else {
|
|
require.NoError(err)
|
|
require.Equal(earliest, tt.earliest)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUpgradeBytesValidation(t *testing.T) {
|
|
type testRun struct {
|
|
name string
|
|
upgradesFile []byte
|
|
expectedErr error
|
|
}
|
|
|
|
tests := []testRun{
|
|
{
|
|
name: "empty file",
|
|
upgradesFile: []byte{},
|
|
expectedErr: errInvalidPrecompiles,
|
|
},
|
|
{
|
|
name: "empty upgrades",
|
|
upgradesFile: []byte(
|
|
`{"precompileUpgrades":[]}`),
|
|
expectedErr: errNoPrecompiles,
|
|
},
|
|
{
|
|
name: "precompile is not []",
|
|
upgradesFile: []byte(
|
|
`{"precompileUpgrades":{"badPrecompile":1234}}`),
|
|
expectedErr: errInvalidPrecompiles,
|
|
},
|
|
{
|
|
name: "no blockTimestamp",
|
|
upgradesFile: []byte(
|
|
`{"precompileUpgrades":[{"feeManagerConfig":{"initialFeeConfig":{"something":"isset"}}}]}`),
|
|
expectedErr: errNoBlockTimestamp,
|
|
},
|
|
{
|
|
name: "bad blockTimestamp type",
|
|
upgradesFile: []byte(
|
|
`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":"1234","initialFeeConfig":{}}}]}`),
|
|
expectedErr: errInvalidPrecompiles,
|
|
},
|
|
{
|
|
name: "zero blockTimestamp",
|
|
upgradesFile: []byte(
|
|
`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":0,"initialFeeConfig":{}}}]}`),
|
|
expectedErr: errBlockTimestampInvalid,
|
|
},
|
|
{
|
|
name: "blockTimestamp ok",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
|
|
skipPrompting := false
|
|
require := require.New(t)
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(_ *testing.T) {
|
|
_, err := validateUpgradeBytes(tt.upgradesFile, nil, skipPrompting)
|
|
require.ErrorIs(err, tt.expectedErr)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestForceIgnorePastTimestamp(t *testing.T) {
|
|
skipPrompting := true
|
|
upgradesFile := []byte(
|
|
`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":1674496268,"initialFeeConfig":{}}}]}`)
|
|
|
|
require := require.New(t)
|
|
_, err := validateUpgradeBytes(upgradesFile, nil, skipPrompting)
|
|
require.NoError(err)
|
|
}
|
|
|
|
func TestLockFile(t *testing.T) {
|
|
type testRun struct {
|
|
name string
|
|
upgradesFile []byte
|
|
lockFile []byte
|
|
expectedErr error
|
|
}
|
|
|
|
sameActivation := time.Now().Add(1 * time.Minute)
|
|
|
|
tests := []testRun{
|
|
{
|
|
name: "same file",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
lockFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "added precompile",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`
|
|
{"precompileUpgrades":[
|
|
{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"txAllowListConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}},
|
|
{"contractNativeMinterConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}
|
|
]}`,
|
|
sameActivation.Unix(),
|
|
time.Now().Add(20*time.Second).Unix(),
|
|
time.Now().Add(30*time.Second).Unix(),
|
|
)),
|
|
lockFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "empty lock",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xcccccccccccccccccccccccccccBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
sameActivation.Unix()),
|
|
),
|
|
lockFile: []byte{},
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
name: "altered initial",
|
|
upgradesFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xcccccccccccccccccccccccccccBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
lockFile: []byte(
|
|
fmt.Sprintf(`{"precompileUpgrades":[{"feeManagerConfig":{"adminAddresses":["0xb794F5eA0ba39494cE839613fffBA74279579268"],"blockTimestamp":%d,"initialFeeConfig":{}}}]}`,
|
|
time.Now().Add(1*time.Minute).Unix()),
|
|
),
|
|
expectedErr: errNewUpgradesNotContainsLock,
|
|
},
|
|
}
|
|
|
|
skipPrompting := false
|
|
require := require.New(t)
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(_ *testing.T) {
|
|
_, err := validateUpgradeBytes(tt.upgradesFile, tt.lockFile, skipPrompting)
|
|
require.ErrorIs(err, tt.expectedErr)
|
|
})
|
|
}
|
|
}
|