89 lines
3.0 KiB
Go
89 lines
3.0 KiB
Go
package simulation_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
"cosmossdk.io/x/bank/simulation"
|
|
"cosmossdk.io/x/bank/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
|
)
|
|
|
|
// TestRandomizedGenState tests the normal scenario of applying RandomizedGenState.
|
|
// Abonormal scenarios are not tested here.
|
|
func TestRandomizedGenState(t *testing.T) {
|
|
interfaceRegistry := codectypes.NewInterfaceRegistry()
|
|
cdc := codec.NewProtoCodec(interfaceRegistry)
|
|
cdcOpts := testutil.CodecOptions{}
|
|
s := rand.NewSource(1)
|
|
r := rand.New(s)
|
|
|
|
simState := module.SimulationState{
|
|
AppParams: make(simtypes.AppParams),
|
|
Cdc: cdc,
|
|
AddressCodec: cdcOpts.GetAddressCodec(),
|
|
ValidatorCodec: cdcOpts.GetValidatorCodec(),
|
|
Rand: r,
|
|
NumBonded: 3,
|
|
BondDenom: sdk.DefaultBondDenom,
|
|
Accounts: simtypes.RandomAccounts(r, 3),
|
|
InitialStake: sdkmath.NewInt(1000),
|
|
GenState: make(map[string]json.RawMessage),
|
|
}
|
|
|
|
simulation.RandomizedGenState(&simState)
|
|
|
|
var bankGenesis types.GenesisState
|
|
simState.Cdc.MustUnmarshalJSON(simState.GenState[types.ModuleName], &bankGenesis)
|
|
|
|
assert.Equal(t, true, bankGenesis.Params.GetDefaultSendEnabled(), "Params.GetDefaultSendEnabled")
|
|
assert.Len(t, bankGenesis.Params.GetSendEnabled(), 0, "Params.GetSendEnabled") //nolint:staticcheck // we're testing the old way here
|
|
if assert.Len(t, bankGenesis.Balances, 3) {
|
|
assert.Equal(t, "cosmos1ghekyjucln7y67ntx7cf27m9dpuxxemn4c8g4r", bankGenesis.Balances[2].Address, "Balances[2] address")
|
|
assert.Equal(t, "1000stake", bankGenesis.Balances[2].GetCoins().String(), "Balances[2] coins")
|
|
}
|
|
assert.Equal(t, "6000stake", bankGenesis.Supply.String(), "Supply")
|
|
if assert.Len(t, bankGenesis.SendEnabled, 1, "SendEnabled") {
|
|
assert.Equal(t, true, bankGenesis.SendEnabled[0].Enabled, "SendEnabled[0] value")
|
|
}
|
|
}
|
|
|
|
// TestRandomizedGenState1 tests abnormal scenarios of applying RandomizedGenState.
|
|
func TestRandomizedGenState1(t *testing.T) {
|
|
interfaceRegistry := codectypes.NewInterfaceRegistry()
|
|
cdc := codec.NewProtoCodec(interfaceRegistry)
|
|
|
|
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)
|
|
}
|
|
}
|