f65215ad92
StoreApp just the queries, BaseApp with handler/ticker Ticker now defined top level, as an interface, with context Name for info taken as parameter, start cmd uses commandline name Cleaner compisition of apps.
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
abci "github.com/tendermint/abci/types"
|
|
wire "github.com/tendermint/go-wire"
|
|
cmn "github.com/tendermint/tmlibs/common"
|
|
"github.com/tendermint/tmlibs/log"
|
|
|
|
"github.com/cosmos/cosmos-sdk/modules/base"
|
|
)
|
|
|
|
//-----------------------------------
|
|
// Test cases start here
|
|
|
|
func randPower() uint64 {
|
|
return uint64(cmn.RandInt()%50 + 60)
|
|
}
|
|
|
|
func makeVal() *abci.Validator {
|
|
return &abci.Validator{
|
|
PubKey: cmn.RandBytes(10),
|
|
Power: randPower(),
|
|
}
|
|
}
|
|
|
|
// withNewPower returns a copy of the validator with a different power
|
|
func withNewPower(val *abci.Validator) *abci.Validator {
|
|
res := *val
|
|
res.Power = randPower()
|
|
return &res
|
|
}
|
|
|
|
func TestEndBlock(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
|
|
logger := log.NewNopLogger()
|
|
handler := base.ValSetHandler{}
|
|
store, err := MockStoreApp("vals", logger)
|
|
require.Nil(err, "%+v", err)
|
|
app := NewBaseApp(store, handler, nil)
|
|
|
|
val1 := makeVal()
|
|
val2 := makeVal()
|
|
val3 := makeVal()
|
|
val1a := withNewPower(val1)
|
|
val2a := withNewPower(val2)
|
|
|
|
cases := [...]struct {
|
|
changes [][]*abci.Validator
|
|
expected []*abci.Validator
|
|
}{
|
|
// Nothing in, nothing out, no crash
|
|
0: {},
|
|
// One in, one out, no problem
|
|
1: {
|
|
changes: [][]*abci.Validator{{val1}},
|
|
expected: []*abci.Validator{val1},
|
|
},
|
|
// Combine a few ones
|
|
2: {
|
|
changes: [][]*abci.Validator{{val1}, {val2, val3}},
|
|
expected: []*abci.Validator{val1, val2, val3},
|
|
},
|
|
// Make sure changes all to one validators are squished into one diff
|
|
3: {
|
|
changes: [][]*abci.Validator{{val1}, {val2, val1a}, {val2a, val3}},
|
|
expected: []*abci.Validator{val1a, val2a, val3},
|
|
},
|
|
}
|
|
|
|
for i, tc := range cases {
|
|
app.BeginBlock(abci.RequestBeginBlock{})
|
|
for _, c := range tc.changes {
|
|
tx := base.ValChangeTx{c}.Wrap()
|
|
txBytes := wire.BinaryBytes(tx)
|
|
res := app.DeliverTx(txBytes)
|
|
require.True(res.IsOK(), "%#v", res)
|
|
}
|
|
diff := app.EndBlock(app.height)
|
|
// TODO: don't care about order here...
|
|
assert.Equal(tc.expected, diff.Diffs, "%d", i)
|
|
}
|
|
}
|