test(evmstaking/keeper): add test cases for set address (#67)

test(evmstaking/keeper): add test cases for set_address

client/x/evmstaking/keeper/set_address_test.go

@@ -0,0 +1,96 @@
+package keeper_test
+
+import (
+	"github.com/cometbft/cometbft/crypto"
+	"github.com/ethereum/go-ethereum/common"
+
+	"github.com/piplabs/story/contracts/bindings"
+	"github.com/piplabs/story/lib/k1util"
+)
+
+func (s *TestSuite) TestProcessSetWithdrawalAddress() {
+	require := s.Require()
+	ctx, keeper := s.Ctx, s.EVMStakingKeeper
+
+	pubKeys, accAddrs, _ := createAddresses(2)
+	delAddr := accAddrs[0]
+	delPubKey := pubKeys[0]
+	invalidPubKey := delPubKey.Bytes()[0:20]
+
+	execAddr, err := k1util.CosmosPubkeyToEVMAddress(delPubKey.Bytes())
+	require.NoError(err)
+	anotherExecAddr, err := k1util.CosmosPubkeyToEVMAddress(pubKeys[1].Bytes())
+	require.NoError(err)
+
+	tcs := []struct {
+		name          string
+		sameAddr      bool
+		input         func(execAddr common.Address, pubKey crypto.PubKey) *bindings.IPTokenStakingSetWithdrawalAddress
+		expectedError string
+	}{
+		{
+			name:     "pass: delegator and execution address are the same",
+			sameAddr: true,
+			input: func(execAddr common.Address, pubKey crypto.PubKey) *bindings.IPTokenStakingSetWithdrawalAddress {
+				paddedExecAddr := common.LeftPadBytes(execAddr.Bytes(), 32)
+				return &bindings.IPTokenStakingSetWithdrawalAddress{
+					DelegatorCmpPubkey: pubKey.Bytes(),
+					ExecutionAddress:   [32]byte(paddedExecAddr),
+				}
+			},
+			expectedError: "",
+		},
+		{
+			name:     "pass: delegator and execution address are different",
+			sameAddr: false,
+			input: func(execAddr common.Address, pubKey crypto.PubKey) *bindings.IPTokenStakingSetWithdrawalAddress {
+				paddedExecAddr := common.LeftPadBytes(execAddr.Bytes(), 32)
+				return &bindings.IPTokenStakingSetWithdrawalAddress{
+					DelegatorCmpPubkey: pubKey.Bytes(),
+					ExecutionAddress:   [32]byte(paddedExecAddr),
+				}
+			},
+			expectedError: "",
+		},
+		{
+			name:     "fail: invalid delegator public key",
+			sameAddr: false,
+			input: func(execAddr common.Address, pubKey crypto.PubKey) *bindings.IPTokenStakingSetWithdrawalAddress {
+				paddedExecAddr := common.LeftPadBytes(execAddr.Bytes(), 32)
+				return &bindings.IPTokenStakingSetWithdrawalAddress{
+					DelegatorCmpPubkey: invalidPubKey,
+					ExecutionAddress:   [32]byte(paddedExecAddr),
+				}
+			},
+			expectedError: "depositor pubkey to cosmos",
+		},
+	}
+
+	for _, tc := range tcs {
+		s.Run(tc.name, func() {
+			evmAddr := execAddr
+			if !tc.sameAddr {
+				evmAddr = anotherExecAddr
+			}
+			ev := tc.input(evmAddr, pubKeys[0])
+
+			cachedCtx, _ := ctx.CacheContext()
+			err := keeper.ProcessSetWithdrawalAddress(cachedCtx, ev)
+			if tc.expectedError != "" {
+				require.Error(err)
+				require.Contains(err.Error(), tc.expectedError)
+
+				// Ensure no state change occurred
+				addr, _ := keeper.DelegatorMap.Get(cachedCtx, delAddr.String())
+				require.Empty(addr)
+			} else {
+				require.NoError(err)
+
+				// check result
+				addr, err := keeper.DelegatorMap.Get(cachedCtx, delAddr.String())
+				require.NoError(err)
+				require.Equal(evmAddr.String(), addr)
+			}
+		})
+	}
+}