cosmos-sdk/x/staking/genesis_test.go

60 lines
1.6 KiB
Go

package staking_test
import (
"testing"
"cosmossdk.io/math"
"github.com/stretchr/testify/assert"
"github.com/cosmos/cosmos-sdk/crypto/keys/ed25519"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/staking"
"github.com/cosmos/cosmos-sdk/x/staking/testutil"
"github.com/cosmos/cosmos-sdk/x/staking/types"
)
func TestValidateGenesis(t *testing.T) {
genValidators1 := make([]types.Validator, 1, 5)
pk := ed25519.GenPrivKey().PubKey()
genValidators1[0] = testutil.NewValidator(t, sdk.ValAddress(pk.Address()), pk)
genValidators1[0].Tokens = math.OneInt()
genValidators1[0].DelegatorShares = math.LegacyOneDec()
tests := []struct {
name string
mutate func(*types.GenesisState)
wantErr bool
}{
{"default", func(*types.GenesisState) {}, false},
// validate genesis validators
{"duplicate validator", func(data *types.GenesisState) {
data.Validators = genValidators1
data.Validators = append(data.Validators, genValidators1[0])
}, true},
{"no delegator shares", func(data *types.GenesisState) {
data.Validators = genValidators1
data.Validators[0].DelegatorShares = math.LegacyZeroDec()
}, true},
{"jailed and bonded validator", func(data *types.GenesisState) {
data.Validators = genValidators1
data.Validators[0].Jailed = true
data.Validators[0].Status = types.Bonded
}, true},
}
for _, tt := range tests {
tt := tt
t.Run(tt.name, func(t *testing.T) {
genesisState := types.DefaultGenesisState()
tt.mutate(genesisState)
if tt.wantErr {
assert.Error(t, staking.ValidateGenesis(genesisState))
} else {
assert.NoError(t, staking.ValidateGenesis(genesisState))
}
})
}
}