2018-10-04 18:37:40 -07:00
|
|
|
package keeper
|
|
|
|
|
2018-10-09 10:58:59 -07:00
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/stake"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
2018-10-04 18:37:40 -07:00
|
|
|
|
|
|
|
func TestWithdrawDelegationReward(t *testing.T) {
|
2018-10-09 10:58:59 -07:00
|
|
|
ctx, accMapper, keeper, sk := CreateTestInput(t, false, 100)
|
|
|
|
stakeHandler := stake.NewHandler(sk)
|
|
|
|
denom := sk.GetParams(ctx).BondDenom
|
|
|
|
|
|
|
|
//first make a validator
|
|
|
|
msgCreateValidator := stake.NewTestMsgCreateValidator(valAddr1, valPk1, 10)
|
|
|
|
got := stakeHandler(ctx, msgCreateValidator)
|
|
|
|
require.True(t, got.IsOK(), "expected msg to be ok, got %v", got)
|
|
|
|
|
|
|
|
// delegate
|
|
|
|
msgDelegate := stake.NewTestMsgDelegate(delAddr1, valAddr1, 10)
|
|
|
|
got = stakeHandler(ctx, msgDelegate)
|
|
|
|
require.True(t, got.IsOK())
|
2018-10-04 18:37:40 -07:00
|
|
|
|
2018-10-09 10:58:59 -07:00
|
|
|
amt1 := accMapper.GetAccount(ctx, delAddr1).GetCoins().AmountOf(denom)
|
|
|
|
require.Equal(t, 90, amt1.Int64())
|
2018-10-08 17:14:03 -07:00
|
|
|
keeper.WithdrawDelegationReward(ctx, delAddr1, valAddr1)
|
2018-10-04 18:37:40 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestWithdrawDelegationRewardsAll(t *testing.T) {
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetDelegatorRewardsAll(t *testing.T) {
|
|
|
|
|
|
|
|
}
|