From 30d046f39b4a5d87bc26e348dcd651af8dba4767 Mon Sep 17 00:00:00 2001
From: Ivan <ivanverch@gmail.com>
Date: Sat, 14 Dec 2024 11:21:11 +0800
Subject: [PATCH] Add TestMsgRedelegate

---
 x/tier/keeper/msg_redelegate_test.go | 158 +++++++++++++++++++++++++++
 1 file changed, 158 insertions(+)
 create mode 100644 x/tier/keeper/msg_redelegate_test.go

diff --git a/x/tier/keeper/msg_redelegate_test.go b/x/tier/keeper/msg_redelegate_test.go
new file mode 100644
index 0000000..97fab65
--- /dev/null
+++ b/x/tier/keeper/msg_redelegate_test.go
@@ -0,0 +1,158 @@
+package keeper_test
+
+import (
+	"testing"
+	"time"
+
+	"cosmossdk.io/math"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/stretchr/testify/require"
+
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
+	appparams "github.com/sourcenetwork/sourcehub/app/params"
+	"github.com/sourcenetwork/sourcehub/x/tier/types"
+)
+
+func TestMsgRedelegate(t *testing.T) {
+	k, ms, ctx := setupMsgServer(t)
+	sdkCtx := sdk.UnwrapSDKContext(ctx)
+
+	validCoin := sdk.NewCoin(appparams.DefaultBondDenom, math.NewInt(100))
+	zeroCoin := sdk.NewCoin(appparams.DefaultBondDenom, math.ZeroInt())
+	negativeAmount := math.NewInt(-100)
+
+	delAddr := "source1m4f5a896t7fzd9vc7pfgmc3fxkj8n24s68fcw9"
+	srcValAddr := "sourcevaloper1cy0p47z24ejzvq55pu3lesxwf73xnrnd0pzkqm"
+	dstValAddr := "sourcevaloper13fj7t2yptf9k6ad6fv38434znzay4s4pjk0r4f"
+
+	delAddress, err := sdk.AccAddressFromBech32(delAddr)
+	require.NoError(t, err)
+	srcValAddress, err := sdk.ValAddressFromBech32(srcValAddr)
+	require.NoError(t, err)
+	dstValAddress, err := sdk.ValAddressFromBech32(dstValAddr)
+	require.NoError(t, err)
+
+	initialDelegatorBalance := math.NewInt(2000)
+	initializeDelegator(t, &k, sdkCtx, delAddress, initialDelegatorBalance)
+	initialSrcValidatorBalance := math.NewInt(1000)
+	initialDstValidatorBalance := math.NewInt(500)
+	initializeValidator(t, k.GetStakingKeeper().(*stakingkeeper.Keeper), sdkCtx, srcValAddress, initialSrcValidatorBalance)
+	initializeValidator(t, k.GetStakingKeeper().(*stakingkeeper.Keeper), sdkCtx, dstValAddress, initialDstValidatorBalance)
+
+	// lock some tokens to test redelegate logic
+	err = k.Lock(ctx, delAddress, srcValAddress, validCoin.Amount)
+	require.NoError(t, err)
+
+	stakingParams, err := k.GetStakingKeeper().(*stakingkeeper.Keeper).GetParams(ctx)
+	require.NoError(t, err)
+
+	// expectedCompletionTime should match the default staking unbonding time (e.g. 21 days)
+	expectedCompletionTime := sdkCtx.BlockTime().Add(stakingParams.UnbondingTime)
+
+	testCases := []struct {
+		name      string
+		input     *types.MsgRedelegate
+		expErr    bool
+		expErrMsg string
+	}{
+		{
+			name: "valid redelegate",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    delAddr,
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: dstValAddr,
+				Stake:               validCoin,
+			},
+			expErr: false,
+		},
+		{
+			name: "insufficient lockup",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    delAddr,
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: dstValAddr,
+				Stake:               sdk.NewCoin(appparams.DefaultBondDenom, math.NewInt(500)),
+			},
+			expErr:    true,
+			expErrMsg: "subtract locked stake from source validator",
+		},
+		{
+			name: "invalid stake amount (zero)",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    delAddr,
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: dstValAddr,
+				Stake:               zeroCoin,
+			},
+			expErr:    true,
+			expErrMsg: "invalid amount",
+		},
+		{
+			name: "invalid stake amount (negative)",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    delAddr,
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: dstValAddr,
+				Stake: sdk.Coin{
+					Denom:  appparams.DefaultBondDenom,
+					Amount: negativeAmount,
+				},
+			},
+			expErr:    true,
+			expErrMsg: "invalid amount",
+		},
+		{
+			name: "non-existent lockup",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    "source1wjj5v5rlf57kayyeskncpu4hwev25ty645p2et",
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: dstValAddr,
+				Stake:               validCoin,
+			},
+			expErr:    true,
+			expErrMsg: "subtract locked stake from source validator",
+		},
+		{
+			name: "source and destination validator are the same",
+			input: &types.MsgRedelegate{
+				DelegatorAddress:    delAddr,
+				SrcValidatorAddress: srcValAddr,
+				DstValidatorAddress: srcValAddr,
+				Stake:               validCoin,
+			},
+			expErr:    true,
+			expErrMsg: "src and dst validator addresses are the same",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			err := tc.input.ValidateBasic()
+			if err != nil {
+				if tc.expErr {
+					require.Contains(t, err.Error(), tc.expErrMsg)
+					return
+				}
+				t.Fatalf("unexpected error in ValidateBasic: %v", err)
+			}
+
+			resp, err := ms.Redelegate(sdkCtx, tc.input)
+
+			if tc.expErr {
+				require.Error(t, err)
+				require.Contains(t, err.Error(), tc.expErrMsg)
+			} else {
+				require.NoError(t, err)
+				require.NotNil(t, resp)
+
+				srcLockup := k.GetLockupAmount(sdkCtx, delAddress, srcValAddress)
+				require.Equal(t, math.ZeroInt(), srcLockup, "Source validator lockup should be zero after valid redelegate")
+
+				dstLockup := k.GetLockupAmount(sdkCtx, delAddress, dstValAddress)
+				require.Equal(t, validCoin.Amount, dstLockup, "Destination validator lockup should equal redelegated amount")
+
+				require.WithinDuration(t, expectedCompletionTime, resp.CompletionTime, time.Second)
+			}
+		})
+	}
+}