65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package keeper
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/distribution/types"
|
|
"github.com/cosmos/cosmos-sdk/x/staking"
|
|
)
|
|
|
|
func TestSetWithdrawAddr(t *testing.T) {
|
|
ctx, _, keeper, _, _ := CreateTestInputDefault(t, false, 1000)
|
|
|
|
keeper.SetWithdrawAddrEnabled(ctx, false)
|
|
|
|
err := keeper.SetWithdrawAddr(ctx, delAddr1, delAddr2)
|
|
require.NotNil(t, err)
|
|
|
|
keeper.SetWithdrawAddrEnabled(ctx, true)
|
|
|
|
err = keeper.SetWithdrawAddr(ctx, delAddr1, delAddr2)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestWithdrawValidatorCommission(t *testing.T) {
|
|
ctx, ak, keeper, _, _ := CreateTestInputDefault(t, false, 1000)
|
|
|
|
// set zero outstanding rewards
|
|
keeper.SetOutstandingRewards(ctx, types.OutstandingRewards{})
|
|
|
|
// check initial balance
|
|
balance := ak.GetAccount(ctx, sdk.AccAddress(valOpAddr3)).GetCoins()
|
|
expTokens := staking.TokensFromTendermintPower(1000)
|
|
require.Equal(t, sdk.Coins{
|
|
{"stake", staking.TokensFromTendermintPower(1000)},
|
|
}, balance)
|
|
|
|
// set commission
|
|
keeper.SetValidatorAccumulatedCommission(ctx, valOpAddr3, sdk.DecCoins{
|
|
{"mytoken", sdk.NewDec(5).Quo(sdk.NewDec(4))},
|
|
{"stake", sdk.NewDec(3).Quo(sdk.NewDec(2))},
|
|
})
|
|
|
|
// withdraw commission
|
|
keeper.WithdrawValidatorCommission(ctx, valOpAddr3)
|
|
|
|
// check balance increase
|
|
balance = ak.GetAccount(ctx, sdk.AccAddress(valOpAddr3)).GetCoins()
|
|
require.Equal(t, sdk.Coins{
|
|
{"mytoken", sdk.NewInt(1)},
|
|
{"stake", expTokens.AddRaw(1)},
|
|
}, balance)
|
|
|
|
// check remainder
|
|
remainder := keeper.GetValidatorAccumulatedCommission(ctx, valOpAddr3)
|
|
require.Equal(t, sdk.DecCoins{
|
|
{"mytoken", sdk.NewDec(1).Quo(sdk.NewDec(4))},
|
|
{"stake", sdk.NewDec(1).Quo(sdk.NewDec(2))},
|
|
}, remainder)
|
|
|
|
require.True(t, true)
|
|
}
|