Add tests for liquidation staked tokens (#299)

* add tests for liquidation staked tokens

* change test description

---------

Co-authored-by: Petr Ivanov <[email protected]>

Author: ivanovpetr

Diff for: x/liquidvesting/keeper/msg_server_test.go

@@ -8,6 +8,7 @@ import (
 	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
 	sdkvesting "github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
 	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/haqq-network/haqq/contracts"
@@ -76,6 +77,22 @@ func (suite *KeeperTestSuite) TestLiquidate() {
 			amount:     sdk.NewCoin("aISLM", amount.AmountOf("aISLM")),
 			expectPass: true,
 		},
+		{
+			name: "ok - liquidate partially staked tokens",
+			malleate: func() {
+				funder := sdk.AccAddress(types.ModuleName)
+				baseAccount := authtypes.NewBaseAccountWithAddress(addr1)
+				startTime := suite.ctx.BlockTime().Add(-10 * time.Second)
+				clawbackAccount := vestingtypes.NewClawbackVestingAccount(baseAccount, funder, amount, startTime, lockupPeriods, vestingPeriods, nil)
+				testutil.FundAccount(s.ctx, s.app.BankKeeper, addr1, amount) //nolint:errcheck
+				s.app.AccountKeeper.SetAccount(s.ctx, clawbackAccount)
+				s.app.StakingKeeper.Delegate(s.ctx, addr1, third.AmountOf("aISLM"), stakingtypes.Unbonded, s.validator, true) //nolint:errcheck
+			},
+			from:       addr1,
+			to:         addr2,
+			amount:     sdk.NewCoin("aISLM", third.AmountOf("aISLM").Add(third.AmountOf("aISLM"))),
+			expectPass: true,
+		},
 		{
 			name: "fail - liquidate amount bigger than locked but less than total",
 			malleate: func() {
@@ -92,6 +109,22 @@ func (suite *KeeperTestSuite) TestLiquidate() {
 			amount:     sdk.NewCoin("aISLM", amount.AmountOf("aISLM").Add(math.NewInt(1_500_000))),
 			expectPass: false,
 		},
+		{
+			name: "fail - liquidate staked tokens",
+			malleate: func() {
+				funder := sdk.AccAddress(types.ModuleName)
+				baseAccount := authtypes.NewBaseAccountWithAddress(addr1)
+				startTime := suite.ctx.BlockTime().Add(-10 * time.Second)
+				clawbackAccount := vestingtypes.NewClawbackVestingAccount(baseAccount, funder, amount, startTime, lockupPeriods, vestingPeriods, nil)
+				testutil.FundAccount(s.ctx, s.app.BankKeeper, addr1, amount) //nolint:errcheck
+				s.app.AccountKeeper.SetAccount(s.ctx, clawbackAccount)
+				s.app.StakingKeeper.Delegate(s.ctx, addr1, amount.AmountOf("aISLM"), stakingtypes.Unbonded, s.validator, true) //nolint:errcheck
+			},
+			from:       addr1,
+			to:         addr2,
+			amount:     sdk.NewCoin("aISLM", amount.AmountOf("aISLM")),
+			expectPass: false,
+		},
 		{
 			name: "fail - liquidate tokens partially unlocked",
 			malleate: func() {