feat: ParseKeyCounterpartyRelayer function

modules/apps/29-fee/keeper/keeper.go

@@ -1,8 +1,6 @@
 package keeper
 
 import (
-	"strings"
-
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	capabilitytypes "github.com/cosmos/cosmos-sdk/x/capability/types"
@@ -162,12 +160,9 @@ func (k Keeper) GetAllRelayerAddresses(ctx sdk.Context) []types.RegisteredRelaye
 
 	var registeredAddrArr []types.RegisteredRelayerAddress
 	for ; iterator.Valid(); iterator.Next() {
-		keySplit := strings.Split(string(iterator.Key()), "/")
-
-		addr := types.RegisteredRelayerAddress{
-			Address:             keySplit[1],
-			CounterpartyAddress: string(iterator.Value()),
-			ChannelId:           keySplit[2],
+		addr, err := types.ParseKeyCounterpartyRelayer(string(iterator.Key()), string(iterator.Value()))
+		if err != nil {
+			panic(err)
 		}
 
 		registeredAddrArr = append(registeredAddrArr, addr)
@@ -265,7 +260,7 @@ func (k Keeper) HasFeesInEscrow(ctx sdk.Context, packetID channeltypes.PacketId)
 	return store.Has(key)
 }
 
-// SetFeesInEscrow sets the given packet fees in escrow keyed by the packetID 
+// SetFeesInEscrow sets the given packet fees in escrow keyed by the packetID
 func (k Keeper) SetFeesInEscrow(ctx sdk.Context, packetID channeltypes.PacketId, fees types.PacketFees) {
 	store := ctx.KVStore(k.storeKey)
 	bz := k.MustMarshalFees(fees)

modules/apps/29-fee/types/keys.go

@@ -71,11 +71,27 @@ func ParseKeyFeeEnabled(key string) (portID, channelID string, err error) {
 	return portID, channelID, nil
 }
 
-// KeyCounterpartyRelayer returns the key for relayer address -> counteryparty address mapping
+// KeyCounterpartyRelayer returns the key for relayer address -> counterparty address mapping
 func KeyCounterpartyRelayer(address, channelID string) []byte {
 	return []byte(fmt.Sprintf("%s/%s/%s", CounterpartyRelayerAddressKeyPrefix, address, channelID))
 }
 
+// ParseKeyCounterpartyRelayer returns the registered relayer address -> counterparty address mapping
+func ParseKeyCounterpartyRelayer(key string, counterpartyAddress string) (RegisteredRelayerAddress, error) {
+	keySplit := strings.Split(key, "/")
+	if len(keySplit) != 3 {
+		return RegisteredRelayerAddress{}, sdkerrors.Wrapf(
+			sdkerrors.ErrLogic, "key provided is incorrect: the key split has incorrect length, expected %d, got %d", 3, len(keySplit),
+		)
+	}
+
+	return RegisteredRelayerAddress{
+		Address:             keySplit[1],
+		CounterpartyAddress: counterpartyAddress,
+		ChannelId:           keySplit[2],
+	}, nil
+}
+
 // KeyForwardRelayerAddress returns the key for packetID -> forwardAddress mapping
 func KeyForwardRelayerAddress(packetId channeltypes.PacketId) []byte {
 	return []byte(fmt.Sprintf("%s/%s/%s/%d", ForwardRelayerPrefix, packetId.PortId, packetId.ChannelId, packetId.Sequence))

modules/apps/29-fee/types/keys_test.go

@@ -134,3 +134,39 @@ func TestParseKeyForwardRelayerAddress(t *testing.T) {
 		}
 	}
 }
+
+func TestParseKeyCounterpartyRelayer(t *testing.T) {
+	var (
+		relayerAddress             = "relayer_address"
+		counterpartyRelayerAddress = "counterparty_address"
+	)
+
+	testCases := []struct {
+		name    string
+		key     string
+		expPass bool
+	}{
+		{
+			"success",
+			string(types.KeyCounterpartyRelayer(relayerAddress, ibctesting.FirstChannelID)),
+			true,
+		},
+		{
+			"incorrect key - key split has incorrect length",
+			"relayerAddress/relayer_address/transfer/channel-0",
+			false,
+		},
+	}
+
+	for _, tc := range testCases {
+		registeredAddr, err := types.ParseKeyCounterpartyRelayer(tc.key, counterpartyRelayerAddress)
+		validRegisteredAddr := types.RegisteredRelayerAddress{relayerAddress, counterpartyRelayerAddress, ibctesting.FirstChannelID}
+
+		if tc.expPass {
+			require.NoError(t, err)
+			require.Equal(t, validRegisteredAddr, registeredAddr)
+		} else {
+			require.Error(t, err)
+		}
+	}
+}