fix lint add export evm state to cmd

onflow · Dec 5, 2024 · 5495113 · 5495113
1 parent 96c4472
commit 5495113
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 72 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"os"
 
+	"github.com/onflow/flow-evm-gateway/cmd/export-evm-state"
 	"github.com/onflow/flow-evm-gateway/cmd/run"
 	"github.com/onflow/flow-evm-gateway/cmd/version"
 	"github.com/rs/zerolog/log"
@@ -23,6 +24,7 @@ func Execute() {
 
 func main() {
 	rootCmd.AddCommand(version.Cmd)
+	rootCmd.AddCommand(export.Cmd)
 	rootCmd.AddCommand(run.Cmd)
 
 	Execute()

diff --git a/cmd/util/main.go b/cmd/util/main.go
diff --git a/services/evm/extract_test.go b/services/evm/extract_test.go
@@ -30,8 +30,9 @@ func extractEVMState(
 	t *testing.T, chainID flowGo.ChainID,
 	registerStoreDir string, evmHeight uint64) *state.EVMState {
 
-	store, err := pebble.New(registerStoreDir, log.Logger)
+	pebbleDB, err := pebble.OpenDB(registerStoreDir)
 	require.NoError(t, err)
+	store := pebble.New(pebbleDB, log.Logger)
 
 	evmState, err := evmState.ExtractEVMState(chainID, evmHeight, store)
 	require.NoError(t, err)