github.com/cosmos/cosmos-sdk@v0.50.10/x/slashing/keeper/hooks_test.go (about)

     1  package keeper_test
     2  
     3  import (
     4  	"github.com/cosmos/cosmos-sdk/testutil/testdata"
     5  	sdk "github.com/cosmos/cosmos-sdk/types"
     6  	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
     7  )
     8  
     9  func (s *KeeperTestSuite) TestAfterValidatorBonded() {
    10  	ctx, keeper := s.ctx, s.slashingKeeper
    11  	require := s.Require()
    12  
    13  	valAddr := sdk.ValAddress(consAddr.Bytes())
    14  	keeper.Hooks().AfterValidatorBonded(ctx, consAddr, valAddr)
    15  
    16  	_, err := keeper.GetValidatorSigningInfo(ctx, consAddr)
    17  	require.NoError(err)
    18  }
    19  
    20  func (s *KeeperTestSuite) TestAfterValidatorCreatedOrRemoved() {
    21  	ctx, keeper := s.ctx, s.slashingKeeper
    22  	require := s.Require()
    23  
    24  	_, pubKey, addr := testdata.KeyTestPubAddr()
    25  	valAddr := sdk.ValAddress(addr)
    26  
    27  	validator, err := stakingtypes.NewValidator(sdk.ValAddress(addr).String(), pubKey, stakingtypes.Description{})
    28  	require.NoError(err)
    29  
    30  	s.stakingKeeper.EXPECT().Validator(ctx, valAddr).Return(validator, nil)
    31  	err = keeper.Hooks().AfterValidatorCreated(ctx, valAddr)
    32  	require.NoError(err)
    33  
    34  	ePubKey, err := keeper.GetPubkey(ctx, addr.Bytes())
    35  	require.NoError(err)
    36  	require.Equal(ePubKey, pubKey)
    37  
    38  	err = keeper.Hooks().AfterValidatorRemoved(ctx, sdk.ConsAddress(addr), nil)
    39  	require.NoError(err)
    40  
    41  	_, err = keeper.GetPubkey(ctx, addr.Bytes())
    42  	require.Error(err)
    43  }