91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package vesting_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/vesting"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
|
|
)
|
|
|
|
type HandlerTestSuite struct {
|
|
suite.Suite
|
|
|
|
handler sdk.Handler
|
|
app *simapp.SimApp
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) SetupTest() {
|
|
checkTx := false
|
|
app := simapp.Setup(checkTx)
|
|
|
|
suite.handler = vesting.NewHandler(app.AccountKeeper, app.BankKeeper)
|
|
suite.app = app
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) TestMsgCreateVestingAccount() {
|
|
ctx := suite.app.BaseApp.NewContext(false, tmproto.Header{Height: suite.app.LastBlockHeight() + 1})
|
|
|
|
balances := sdk.NewCoins(sdk.NewInt64Coin("test", 1000))
|
|
addr1 := sdk.AccAddress([]byte("addr1"))
|
|
addr2 := sdk.AccAddress([]byte("addr2"))
|
|
addr3 := sdk.AccAddress([]byte("addr3"))
|
|
|
|
acc1 := suite.app.AccountKeeper.NewAccountWithAddress(ctx, addr1)
|
|
suite.app.AccountKeeper.SetAccount(ctx, acc1)
|
|
suite.Require().NoError(suite.app.BankKeeper.SetBalances(ctx, addr1, balances))
|
|
|
|
testCases := map[string]struct {
|
|
msg *types.MsgCreateVestingAccount
|
|
expectErr bool
|
|
}{
|
|
"create delayed vesting account": {
|
|
msg: types.NewMsgCreateVestingAccount(addr1, addr2, sdk.NewCoins(sdk.NewInt64Coin("test", 100)), ctx.BlockTime().Unix()+10000, true),
|
|
expectErr: false,
|
|
},
|
|
"create continuous vesting account": {
|
|
msg: types.NewMsgCreateVestingAccount(addr1, addr3, sdk.NewCoins(sdk.NewInt64Coin("test", 100)), ctx.BlockTime().Unix()+10000, false),
|
|
expectErr: false,
|
|
},
|
|
"continuous vesting account already exists": {
|
|
msg: types.NewMsgCreateVestingAccount(addr1, addr3, sdk.NewCoins(sdk.NewInt64Coin("test", 100)), ctx.BlockTime().Unix()+10000, false),
|
|
expectErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
tc := tc
|
|
|
|
suite.Run(name, func() {
|
|
res, err := suite.handler(ctx, tc.msg)
|
|
if tc.expectErr {
|
|
suite.Require().Error(err)
|
|
} else {
|
|
suite.Require().NoError(err)
|
|
suite.Require().NotNil(res)
|
|
|
|
accI := suite.app.AccountKeeper.GetAccount(ctx, tc.msg.ToAddress)
|
|
suite.Require().NotNil(accI)
|
|
|
|
if tc.msg.Delayed {
|
|
acc, ok := accI.(*types.DelayedVestingAccount)
|
|
suite.Require().True(ok)
|
|
suite.Require().Equal(tc.msg.Amount, acc.GetVestingCoins(ctx.BlockTime()))
|
|
} else {
|
|
acc, ok := accI.(*types.ContinuousVestingAccount)
|
|
suite.Require().True(ok)
|
|
suite.Require().Equal(tc.msg.Amount, acc.GetVestingCoins(ctx.BlockTime()))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHandlerTestSuite(t *testing.T) {
|
|
suite.Run(t, new(HandlerTestSuite))
|
|
}
|