Merge PR #6772: x/bank/simulation/genesis.go: add unit tests
This commit is contained in:
parent
4b07cb6cc4
commit
9262319caf
|
@ -0,0 +1,72 @@
|
|||
package simulation_test
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"math/rand"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/cosmos/cosmos-sdk/codec"
|
||||
"github.com/cosmos/cosmos-sdk/types/module"
|
||||
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
||||
"github.com/cosmos/cosmos-sdk/x/bank/simulation"
|
||||
"github.com/cosmos/cosmos-sdk/x/bank/types"
|
||||
)
|
||||
|
||||
// TestRandomizedGenState tests the normal scenario of applying RandomizedGenState.
|
||||
// Abonormal scenarios are not tested here.
|
||||
func TestRandomizedGenState(t *testing.T) {
|
||||
cdc := codec.New()
|
||||
s := rand.NewSource(1)
|
||||
r := rand.New(s)
|
||||
|
||||
simState := module.SimulationState{
|
||||
AppParams: make(simtypes.AppParams),
|
||||
Cdc: cdc,
|
||||
Rand: r,
|
||||
NumBonded: 3,
|
||||
Accounts: simtypes.RandomAccounts(r, 3),
|
||||
InitialStake: 1000,
|
||||
GenState: make(map[string]json.RawMessage),
|
||||
}
|
||||
|
||||
simulation.RandomizedGenState(&simState)
|
||||
|
||||
var bankGenesis types.GenesisState
|
||||
simState.Cdc.MustUnmarshalJSON(simState.GenState[types.ModuleName], &bankGenesis)
|
||||
|
||||
require.Equal(t, true, bankGenesis.Params.GetDefaultSendEnabled())
|
||||
require.Len(t, bankGenesis.Params.GetSendEnabled(), 1)
|
||||
require.Len(t, bankGenesis.Balances, 3)
|
||||
require.Equal(t, "cosmos1ghekyjucln7y67ntx7cf27m9dpuxxemn4c8g4r", bankGenesis.Balances[2].GetAddress().String())
|
||||
require.Equal(t, "1000stake", bankGenesis.Balances[2].GetCoins().String())
|
||||
require.Equal(t, "6000stake", bankGenesis.Supply.String())
|
||||
}
|
||||
|
||||
// TestRandomizedGenState tests abnormal scenarios of applying RandomizedGenState.
|
||||
func TestRandomizedGenState1(t *testing.T) {
|
||||
cdc := codec.New()
|
||||
|
||||
s := rand.NewSource(1)
|
||||
r := rand.New(s)
|
||||
|
||||
// all these tests will panic
|
||||
tests := []struct {
|
||||
simState module.SimulationState
|
||||
panicMsg string
|
||||
}{
|
||||
{ // panic => reason: incomplete initialization of the simState
|
||||
module.SimulationState{}, "invalid memory address or nil pointer dereference"},
|
||||
{ // panic => reason: incomplete initialization of the simState
|
||||
module.SimulationState{
|
||||
AppParams: make(simtypes.AppParams),
|
||||
Cdc: cdc,
|
||||
Rand: r,
|
||||
}, "assignment to entry in nil map"},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
require.Panicsf(t, func() { simulation.RandomizedGenState(&tt.simState) }, tt.panicMsg)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue