diff --git a/mempool/mempool_test.go b/mempool/mempool_test.go index 9ac8fe33d..a1bd188d4 100644 --- a/mempool/mempool_test.go +++ b/mempool/mempool_test.go @@ -4,14 +4,28 @@ import ( "encoding/binary" "testing" + "github.com/tendermint/abci/example/counter" + cfg "github.com/tendermint/tendermint/config" "github.com/tendermint/tendermint/config/tendermint_test" "github.com/tendermint/tendermint/proxy" "github.com/tendermint/tendermint/types" - "github.com/tendermint/abci/example/counter" ) +func ResetConfig(name string) *Config { + viperConfig := tendermint_test.ResetConfig(name) + config := new(struct { + cfg.Config `mapstructure:",squash"` + Mempool *Config `mapstructure:"mempool"` + }) + if err := viperConfig.Unmarshal(config); err != nil { + panic(err) + } + return config.Mempool + +} + func TestSerialReap(t *testing.T) { - config := tendermint_test.ResetConfig("mempool_mempool_test") + config := ResetConfig("mempool_test") app := counter.NewCounterApplication(true) app.SetOption("serial", "on") diff --git a/state/execution_test.go b/state/execution_test.go index 40e4044a4..68c318448 100644 --- a/state/execution_test.go +++ b/state/execution_test.go @@ -7,12 +7,10 @@ import ( "github.com/stretchr/testify/require" "github.com/tendermint/abci/example/dummy" crypto "github.com/tendermint/go-crypto" - dbm "github.com/tendermint/tmlibs/db" - cfg "github.com/tendermint/tendermint/config/tendermint_test" - "github.com/tendermint/tendermint/mempool" "github.com/tendermint/tendermint/proxy" "github.com/tendermint/tendermint/state/txindex" "github.com/tendermint/tendermint/types" + dbm "github.com/tendermint/tmlibs/db" ) var ( @@ -24,12 +22,10 @@ var ( func TestApplyBlock(t *testing.T) { cc := proxy.NewLocalClientCreator(dummy.NewDummyApplication()) - config := cfg.ResetConfig("execution_test_") - proxyApp := proxy.NewAppConns(config, cc, nil) + proxyApp := proxy.NewAppConns(cc, nil) _, err := proxyApp.Start() require.Nil(t, err) defer proxyApp.Stop() - mempool := mempool.NewMempool(config, proxyApp.Mempool()) state := state() indexer := &dummyIndexer{0} @@ -38,7 +34,7 @@ func TestApplyBlock(t *testing.T) { // make block block := makeBlock(1, state) - err = state.ApplyBlock(nil, proxyApp.Consensus(), block, block.MakePartSet(testPartSize).Header(), mempool) + err = state.ApplyBlock(nil, proxyApp.Consensus(), block, block.MakePartSet(testPartSize).Header(), types.MockMempool{}) require.Nil(t, err) assert.Equal(t, nTxsPerBlock, indexer.Indexed) // test indexing works diff --git a/state/state_test.go b/state/state_test.go index f80c08fe5..97df5cab4 100644 --- a/state/state_test.go +++ b/state/state_test.go @@ -7,15 +7,27 @@ import ( "github.com/stretchr/testify/assert" abci "github.com/tendermint/abci/types" "github.com/tendermint/go-crypto" - dbm "github.com/tendermint/tmlibs/db" + cfg "github.com/tendermint/tendermint/config" "github.com/tendermint/tendermint/config/tendermint_test" + dbm "github.com/tendermint/tmlibs/db" ) +func ResetConfig(name string) *cfg.Config { + viperConfig := tendermint_test.ResetConfig(name) + config := new(struct { + cfg.Config `mapstructure:",squash"` + }) + if err := viperConfig.Unmarshal(config); err != nil { + panic(err) + } + return &config.Config +} + func TestStateCopyEquals(t *testing.T) { - config := tendermint_test.ResetConfig("state_") + config := ResetConfig("state_") // Get State db - stateDB := dbm.NewDB("state", config.GetString("db_backend"), config.GetString("db_dir")) - state := GetState(config, stateDB) + stateDB := dbm.NewDB("state", config.DBBackend, config.DBDir) + state := GetState(stateDB, config.GenesisFile) stateCopy := state.Copy() @@ -31,10 +43,10 @@ func TestStateCopyEquals(t *testing.T) { } func TestStateSaveLoad(t *testing.T) { - config := tendermint_test.ResetConfig("state_") + config := ResetConfig("state_") // Get State db - stateDB := dbm.NewDB("state", config.GetString("db_backend"), config.GetString("db_dir")) - state := GetState(config, stateDB) + stateDB := dbm.NewDB("state", config.DBBackend, config.DBDir) + state := GetState(stateDB, config.GenesisFile) state.LastBlockHeight += 1 state.Save() @@ -48,9 +60,9 @@ func TestStateSaveLoad(t *testing.T) { func TestABCIResponsesSaveLoad(t *testing.T) { assert := assert.New(t) - config := tendermint_test.ResetConfig("state_") - stateDB := dbm.NewDB("state", config.GetString("db_backend"), config.GetString("db_dir")) - state := GetState(config, stateDB) + config := ResetConfig("state_") + stateDB := dbm.NewDB("state", config.DBBackend, config.DBDir) + state := GetState(stateDB, config.GenesisFile) state.LastBlockHeight += 1