wormhole/wormchain/x/wormhole/keeper/consensus_guardian_set_inde...

39 lines
1.2 KiB
Go

package keeper_test
import (
"testing"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/stretchr/testify/require"
keepertest "github.com/wormhole-foundation/wormchain/testutil/keeper"
"github.com/wormhole-foundation/wormchain/testutil/nullify"
"github.com/wormhole-foundation/wormchain/x/wormhole/keeper"
"github.com/wormhole-foundation/wormchain/x/wormhole/types"
)
func createTestConsensusGuardianSetIndex(keeper *keeper.Keeper, ctx sdk.Context) types.ConsensusGuardianSetIndex {
item := types.ConsensusGuardianSetIndex{}
keeper.SetConsensusGuardianSetIndex(ctx, item)
return item
}
func TestConsensusGuardianSetIndexGet(t *testing.T) {
keeper, ctx := keepertest.WormholeKeeper(t)
item := createTestConsensusGuardianSetIndex(keeper, ctx)
rst, found := keeper.GetConsensusGuardianSetIndex(ctx)
require.True(t, found)
require.Equal(t,
nullify.Fill(&item),
nullify.Fill(&rst),
)
}
func TestConsensusGuardianSetIndexRemove(t *testing.T) {
keeper, ctx := keepertest.WormholeKeeper(t)
createTestConsensusGuardianSetIndex(keeper, ctx)
keeper.RemoveConsensusGuardianSetIndex(ctx)
_, found := keeper.GetConsensusGuardianSetIndex(ctx)
require.False(t, found)
}