certusone
/
cosmos-sdk
mirror of https://github.com/certusone/cosmos-sdk.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Cleanup testcase a bit

This commit is contained in:
Christopher Goes 2018-05-26 00:36:12 +02:00
parent 97b084b842
commit 8aaff4b96e
No known key found for this signature in database
GPG Key ID: E828D98232D328D3
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
x/slashing/keeper_test.go
View File

@ -29,7 +29,7 @@ var (
newPubKey("0B485CFC0EECC619440448436F8FC9DF40566F2369E72400281454CB552AFB50"), newPubKey("0B485CFC0EECC619440448436F8FC9DF40566F2369E72400281454CB552AFB50"),
newPubKey("0B485CFC0EECC619440448436F8FC9DF40566F2369E72400281454CB552AFB51"), newPubKey("0B485CFC0EECC619440448436F8FC9DF40566F2369E72400281454CB552AFB51"),
} }
initCoins int64 = 100 initCoins int64 = 200
) )
func createTestCodec() *wire.Codec { func createTestCodec() *wire.Codec {
@ -59,7 +59,7 @@ func createTestInput(t *testing.T) (sdk.Context, bank.Keeper, stake.Keeper, Keep
ck := bank.NewKeeper(accountMapper) ck := bank.NewKeeper(accountMapper)
sk := stake.NewKeeper(cdc, keyStake, ck, stake.DefaultCodespace) sk := stake.NewKeeper(cdc, keyStake, ck, stake.DefaultCodespace)
genesis := stake.DefaultGenesisState() genesis := stake.DefaultGenesisState()
genesis.Pool.BondedTokens = initCoins * int64(len(addrs)) genesis.Pool.LooseUnbondedTokens = initCoins * int64(len(addrs))
stake.InitGenesis(ctx, sk, genesis) stake.InitGenesis(ctx, sk, genesis)
for _, addr := range addrs { for _, addr := range addrs {
ck.AddCoins(ctx, addr, sdk.Coins{ ck.AddCoins(ctx, addr, sdk.Coins{
@ -72,7 +72,7 @@ func createTestInput(t *testing.T) (sdk.Context, bank.Keeper, stake.Keeper, Keep
func TestHandleDoubleSign(t *testing.T) { func TestHandleDoubleSign(t *testing.T) {
ctx, ck, sk, keeper := createTestInput(t) ctx, ck, sk, keeper := createTestInput(t)
addr, val, amt := addrs[0], pks[0], int64(10) addr, val, amt := addrs[0], pks[0], int64(100)
got := stake.NewHandler(sk)(ctx, newTestMsgDeclareCandidacy(addr, val, amt)) got := stake.NewHandler(sk)(ctx, newTestMsgDeclareCandidacy(addr, val, amt))
require.True(t, got.IsOK()) require.True(t, got.IsOK())
_ = sk.Tick(ctx) _ = sk.Tick(ctx)
@ -87,7 +87,7 @@ func TestHandleDoubleSign(t *testing.T) {
func TestHandleAbsentValidator(t *testing.T) { func TestHandleAbsentValidator(t *testing.T) {
ctx, ck, sk, keeper := createTestInput(t) ctx, ck, sk, keeper := createTestInput(t)
addr, val, amt := addrs[0], pks[0], int64(10) addr, val, amt := addrs[0], pks[0], int64(100)
sh := stake.NewHandler(sk) sh := stake.NewHandler(sk)
got := sh(ctx, newTestMsgDeclareCandidacy(addr, val, amt)) got := sh(ctx, newTestMsgDeclareCandidacy(addr, val, amt))
require.True(t, got.IsOK()) require.True(t, got.IsOK())
@ -121,7 +121,7 @@ func TestHandleAbsentValidator(t *testing.T) {
validator := sk.ValidatorByPubKey(ctx, val) validator := sk.ValidatorByPubKey(ctx, val)
require.Equal(t, sdk.Bonded, validator.GetStatus()) require.Equal(t, sdk.Bonded, validator.GetStatus())
pool := sk.GetPool(ctx) pool := sk.GetPool(ctx)
require.Equal(t, int64(210), pool.BondedTokens) require.Equal(t, int64(100), pool.BondedTokens)
// 51st block missed // 51st block missed
ctx = ctx.WithBlockHeight(height) ctx = ctx.WithBlockHeight(height)
keeper.handleValidatorSignature(ctx, val, false) keeper.handleValidatorSignature(ctx, val, false)
@ -140,9 +140,9 @@ func TestHandleAbsentValidator(t *testing.T) {
require.True(t, got.IsOK()) // should succeed after jail expiration require.True(t, got.IsOK()) // should succeed after jail expiration
validator = sk.ValidatorByPubKey(ctx, val) validator = sk.ValidatorByPubKey(ctx, val)
require.Equal(t, sdk.Bonded, validator.GetStatus()) require.Equal(t, sdk.Bonded, validator.GetStatus())
pool = sk.GetPool(ctx)
// should have been slashed // should have been slashed
require.Equal(t, int64(208), pool.BondedTokens) pool = sk.GetPool(ctx)
require.Equal(t, int64(99), pool.BondedTokens)
} }
func newPubKey(pk string) (res crypto.PubKey) { func newPubKey(pk string) (res crypto.PubKey) {