fix test and lint

Author: RogerKSI

x/restake/keeper/hooks.go

@@ -1,7 +1,7 @@
 package keeper
 
 import (
-	"cosmossdk.io/math"
+	sdkmath "cosmossdk.io/math"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
 
@@ -84,7 +84,7 @@ func (h Hooks) AfterUnbondingInitiated(_ sdk.Context, _ uint64) error {
 	return nil
 }
 
-func (h Hooks) isAbleToUnbond(ctx sdk.Context, addr sdk.AccAddress, delegated math.Int) error {
+func (h Hooks) isAbleToUnbond(ctx sdk.Context, addr sdk.AccAddress, delegated sdkmath.Int) error {
 	iterator := sdk.KVStoreReversePrefixIterator(ctx.KVStore(h.k.storeKey), types.LocksByAmountIndexKey(addr))
 	defer iterator.Close()
 

x/restake/keeper/keeper_lock.go

@@ -1,7 +1,6 @@
 package keeper
 
 import (
-	"cosmossdk.io/math"
 	sdkmath "cosmossdk.io/math"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
@@ -10,7 +9,7 @@ import (
 
 // SetLockedPower sets the new locked power amount of the address to the key
 // This function will override the previous locked amount.
-func (k Keeper) SetLockedPower(ctx sdk.Context, lockerAddr sdk.AccAddress, keyName string, amount math.Int) error {
+func (k Keeper) SetLockedPower(ctx sdk.Context, lockerAddr sdk.AccAddress, keyName string, amount sdkmath.Int) error {
 	if !amount.IsUint64() {
 		return types.ErrInvalidAmount
 	}
@@ -69,19 +68,19 @@ func (k Keeper) SetLockedPower(ctx sdk.Context, lockerAddr sdk.AccAddress, keyNa
 }
 
 // GetLockedPower returns locked power of the address to the key.
-func (k Keeper) GetLockedPower(ctx sdk.Context, lockerAddr sdk.AccAddress, keyName string) (math.Int, error) {
+func (k Keeper) GetLockedPower(ctx sdk.Context, lockerAddr sdk.AccAddress, keyName string) (sdkmath.Int, error) {
 	key, err := k.GetKey(ctx, keyName)
 	if err != nil {
-		return math.Int{}, types.ErrKeyNotFound
+		return sdkmath.Int{}, types.ErrKeyNotFound
 	}
 
 	if !key.IsActive {
-		return math.Int{}, types.ErrKeyNotActive
+		return sdkmath.Int{}, types.ErrKeyNotActive
 	}
 
 	lock, err := k.GetLock(ctx, lockerAddr, keyName)
 	if err != nil {
-		return math.Int{}, types.ErrLockNotFound
+		return sdkmath.Int{}, types.ErrLockNotFound
 	}
 
 	return lock.Amount, nil

x/restake/keeper/msg_server.go

@@ -46,7 +46,7 @@ func (k msgServer) ClaimRewards(
 	finalRewards, remainders := reward.Rewards.TruncateDecimal()
 
 	if !finalRewards.IsZero() {
-		lock.PosRewardDebts = sdk.NewDecCoins()
+		lock.PosRewardDebts = k.getTotalRewards(ctx, lock)
 		lock.NegRewardDebts = remainders
 		k.SetLock(ctx, lock)
 

x/restake/keeper/msg_server_test.go

@@ -50,6 +50,7 @@ func (suite *KeeperTestSuite) TestMsgClaimRewards() {
 				lock, err := suite.restakeKeeper.GetLock(ctx, ValidAddress1, ValidKey1)
 				suite.Require().NoError(err)
 				suite.Require().Equal(sdk.NewDecCoins(sdk.NewDecCoin("uband", sdkmath.NewInt(1))), lock.PosRewardDebts)
+				suite.Require().Equal(sdk.DecCoins(nil), lock.NegRewardDebts)
 			},
 		},
 	}