feat: adding OnChanUpgradeInit handler implementation to 29-fee

modules/apps/29-fee/ibc_middleware.go

@@ -322,7 +322,32 @@ func (im IBCMiddleware) OnTimeoutPacket(
 
 // OnChanUpgradeInit implements the IBCModule interface
 func (im IBCMiddleware) OnChanUpgradeInit(ctx sdk.Context, portID, channelID string, order channeltypes.Order, connectionHops []string, sequence uint64, version, previousVersion string) (string, error) {
- return im.app.OnChanUpgradeInit(ctx, portID, channelID, order, connectionHops, sequence, version, previousVersion)
+ var versionMetadata types.Metadata
+ if err := types.ModuleCdc.UnmarshalJSON([]byte(version), &versionMetadata); err != nil {
+ // Since it is valid for fee version to not be specified, the above middleware version may be for a middleware
+ // lower down in the stack. Thus, if it is not a fee version we pass the entire version string onto the underlying
+ // application.
+ return im.app.OnChanUpgradeInit(ctx, portID, channelID, order, connectionHops, sequence, version, previousVersion)
+ }
+
+ if versionMetadata.FeeVersion != types.Version {
+ return "", errorsmod.Wrapf(types.ErrInvalidVersion, "expected %s, got %s", types.Version, versionMetadata.FeeVersion)
+ }
+
+ appVersion, err := im.app.OnChanUpgradeInit(ctx, portID, channelID, order, connectionHops, sequence, versionMetadata.AppVersion, previousVersion)
+ if err != nil {
+ return "", err
+ }
+
+ versionMetadata.AppVersion = appVersion
+ versionBz, err := types.ModuleCdc.MarshalJSON(&versionMetadata)
+ if err != nil {
+ return "", err
+ }
+
+ im.keeper.SetFeeEnabled(ctx, portID, channelID)
+
+ return string(versionBz), nil
 }
 
 // OnChanUpgradeTry implement s the IBCModule interface

modules/apps/29-fee/ibc_middleware_test.go

@@ -1004,6 +1004,90 @@ func (suite *FeeTestSuite) TestOnTimeoutPacket() {
  }
 }
 
+func (suite *FeeTestSuite) TestOnChanUpgradeInit() {
+ var path *ibctesting.Path
+
+ testCases := []struct {
+ name string
+ malleate func()
+ expError error
+ }{
+ {
+ "success",
+ func() {},
+ nil,
+ },
+ {
+ "invalid upgrade version",
+ func() {
+ path.EndpointA.ChannelConfig.ProposedUpgrade.Fields.Version = "invalid-version"
+ path.EndpointB.ChannelConfig.ProposedUpgrade.Fields.Version = "invalid-version"
+
+ suite.chainA.GetSimApp().FeeMockModule.IBCApp.OnChanUpgradeInit = func(_ sdk.Context, _, _ string, _ channeltypes.Order, _ []string, _ uint64, _, _ string) (string, error) {
+ return "", ibcmock.MockApplicationCallbackError
+ }
+ },
+ ibcmock.MockApplicationCallbackError,
+ },
+ {
+ "invalid fee version",
+ func() {
+ upgradeVersion := string(types.ModuleCdc.MustMarshalJSON(&types.Metadata{FeeVersion: "invalid-version", AppVersion: ibcmock.Version}))
+ path.EndpointA.ChannelConfig.ProposedUpgrade.Fields.Version = upgradeVersion
+ path.EndpointB.ChannelConfig.ProposedUpgrade.Fields.Version = upgradeVersion
+ },
+ types.ErrInvalidVersion,
+ },
+ {
+ "underlying app callback returns error",
+ func() {
+ suite.chainA.GetSimApp().FeeMockModule.IBCApp.OnChanUpgradeInit = func(_ sdk.Context, _, _ string, _ channeltypes.Order, _ []string, _ uint64, _, _ string) (string, error) {
+ return "", ibcmock.MockApplicationCallbackError
+ }
+ },
+ ibcmock.MockApplicationCallbackError,
+ },
+ }
+
+ for _, tc := range testCases {
+ tc := tc
+ suite.Run(tc.name, func() {
+ suite.SetupTest()
+
+ path = ibctesting.NewPath(suite.chainA, suite.chainB)
+
+ // configure the initial path to create an unincentivized mock channel
+ path.EndpointA.ChannelConfig.PortID = ibctesting.MockFeePort
+ path.EndpointB.ChannelConfig.PortID = ibctesting.MockFeePort
+ path.EndpointA.ChannelConfig.Version = ibcmock.Version
+ path.EndpointB.ChannelConfig.Version = ibcmock.Version
+
+ suite.coordinator.Setup(path)
+
+ // configure the channel upgrade version to enabled ics29 fee middleware
+ upgradeVersion := string(types.ModuleCdc.MustMarshalJSON(&types.Metadata{FeeVersion: types.Version, AppVersion: ibcmock.Version}))
+ path.EndpointA.ChannelConfig.ProposedUpgrade.Fields.Version = upgradeVersion
+ path.EndpointB.ChannelConfig.ProposedUpgrade.Fields.Version = upgradeVersion
+
+ tc.malleate()
+
+ err := path.EndpointA.ChanUpgradeInit()
+
+ isFeeEnabled := suite.chainA.GetSimApp().IBCFeeKeeper.IsFeeEnabled(suite.chainA.GetContext(), path.EndpointA.ChannelConfig.PortID, path.EndpointA.ChannelID)
+
+ expPass := tc.expError == nil
+ if expPass {
+ suite.Require().True(isFeeEnabled)
+ suite.Require().NoError(err)
+ } else {
+ suite.Require().False(isFeeEnabled)
+ suite.Require().Error(err)
+ suite.Require().ErrorIs(err, tc.expError)
+ }
+ })
+ }
+}
+
 func (suite *FeeTestSuite) TestGetAppVersion() {
  var (
  portID string

testing/chain.go

@@ -318,14 +318,13 @@ func (chain *TestChain) SendMsgs(msgs ...sdk.Msg) (*sdk.Result, error) {
  chain.Coordinator.UpdateTimeForChain(chain)
 
  _, r, err := simapp.SignAndDeliver(
- chain.TB,
  chain.TxConfig,
  chain.App.GetBaseApp(),
  msgs,
  chain.ChainID,
  []uint64{chain.SenderAccount.GetAccountNumber()},
  []uint64{chain.SenderAccount.GetSequence()},
- true, chain.SenderPrivKey,
+ chain.SenderPrivKey,
  )
  if err != nil {
  return nil, err

testing/simapp/test_helpers.go

@@ -3,7 +3,6 @@ package simapp
 import (
  "encoding/json"
  "math/rand"
- "testing"
  "time"
 
  dbm "github.com/cometbft/cometbft-db"
@@ -22,7 +21,6 @@ import (
  authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
  banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
  stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
- "github.com/stretchr/testify/require"
 
  "github.com/cosmos/ibc-go/v7/testing/mock"
 )
@@ -209,8 +207,8 @@ func SetupWithGenesisAccounts(genAccs []authtypes.GenesisAccount, balances ...ba
 //
 // CONTRACT: BeginBlock must be called before this function.
 func SignAndDeliver(
- tb testing.TB, txCfg client.TxConfig, app *bam.BaseApp, msgs []sdk.Msg,
- chainID string, accNums, accSeqs []uint64, expPass bool, priv ...cryptotypes.PrivKey,
+ txCfg client.TxConfig, app *bam.BaseApp, msgs []sdk.Msg,
+ chainID string, accNums, accSeqs []uint64, priv ...cryptotypes.PrivKey,
 ) (sdk.GasInfo, *sdk.Result, error) {
  tx, err := simtestutil.GenSignedMockTx(
  rand.New(rand.NewSource(time.Now().UnixNano())),
@@ -223,18 +221,9 @@ func SignAndDeliver(
  accSeqs,
  priv...,
  )
- require.NoError(tb, err)
-
- // Simulate a sending a transaction
- gInfo, res, err := app.SimDeliver(txCfg.TxEncoder(), tx)
-
- if expPass {
- require.NoError(tb, err)
- require.NotNil(tb, res)
- } else {
- require.Error(tb, err)
- require.Nil(tb, res)
+ if err != nil {
+ return sdk.GasInfo{}, nil, err
  }
 
- return gInfo, res, err
+ return app.SimDeliver(txCfg.TxEncoder(), tx)
 }