set up test

This commit is contained in:
Roy Crihfield 2022-01-10 18:59:15 -06:00
parent 1f21a65478
commit 802a5908df
2 changed files with 64 additions and 0 deletions

1
go.mod
View File

@ -26,6 +26,7 @@ require (
github.com/smartystreets/assertions v1.0.0 // indirect github.com/smartystreets/assertions v1.0.0 // indirect
github.com/spf13/cobra v1.0.0 github.com/spf13/cobra v1.0.0
github.com/spf13/viper v1.7.0 github.com/spf13/viper v1.7.0
github.com/vulcanize/go-eth-state-node-iterator v0.0.1-alpha
go.uber.org/atomic v1.9.0 // indirect go.uber.org/atomic v1.9.0 // indirect
go.uber.org/goleak v1.1.11 // indirect go.uber.org/goleak v1.1.11 // indirect
go.uber.org/multierr v1.7.0 // indirect go.uber.org/multierr v1.7.0 // indirect

View File

@ -0,0 +1,63 @@
package snapshot
import (
"path/filepath"
"testing"
ethNode "github.com/ethereum/go-ethereum/statediff/indexer/node"
"github.com/ethereum/go-ethereum/statediff/indexer/postgres"
)
func testConfig(leveldbpath, ancientdbpath string) *Config {
dbParams := postgres.ConnectionParams{}
dbParams.Name = "snapshot_test"
dbParams.Hostname = "localhost"
dbParams.Port = 5432
dbParams.User = "tester"
dbParams.Password = "test_pw"
uri := postgres.DbConnectionString(dbParams)
dbconfig := postgres.ConnectionConfig{
MaxIdle: 0,
MaxLifetime: 0,
MaxOpen: 4,
}
nodeinfo := ethNode.Info{
ID: "eth_node_id",
ClientName: "eth_client",
GenesisBlock: "X",
NetworkID: "eth_network",
ChainID: 0,
}
return &Config{
LevelDBPath: leveldbpath,
AncientDBPath: ancientdbpath,
Node: nodeinfo,
connectionURI: uri,
DBConfig: dbconfig,
}
}
func TestCreateSnapshot(t *testing.T) {
datadir := t.TempDir()
config := testConfig(
filepath.Join(datadir, "leveldb"),
filepath.Join(datadir, "ancient"),
)
service, err := NewSnapshotService(config)
if err != nil {
t.Fatal(err)
}
params := SnapshotParams{Height: 1}
err = service.CreateSnapshot(params)
if err != nil {
t.Fatal(err)
}
// err = service.CreateLatestSnapshot(0)
// if err != nil {
// t.Fatal(err)
// }
}