test: fix failing tests after protocompat changes (#22889)

Co-authored-by: Marko <[email protected]>

baseapp/internal/protocompat/protocompat.go

@@ -123,14 +123,13 @@ func makeGogoHybridHandler(prefMethod protoreflect.MethodDescriptor, cdc codec.B
 				return fmt.Errorf("invalid request type %T, method %s does not accept protov2 messages", inReq, prefMethod.FullName())
 			}
 			resp, err := method.Handler(handler, ctx, func(msg any) error {
-				setPointer(msg, inReq)
-				return nil
+				return setPointer(msg, inReq)
 			}, nil)
 			if err != nil {
 				return err
 			}
-			setPointer(outResp, resp)
-			return nil
+
+			return setPointer(outResp, resp)
 		}, nil
 	}
 	// this is a gogo handler, and we have a protov2 counterparty.
@@ -161,14 +160,13 @@ func makeGogoHybridHandler(prefMethod protoreflect.MethodDescriptor, cdc codec.B
 		case gogoproto.Message:
 			// we can just call the handler after making a copy of the message, for safety reasons.
 			resp, err := method.Handler(handler, ctx, func(msg any) error {
-				setPointer(msg, m)
-				return nil
+				return setPointer(msg, m)
 			}, nil)
 			if err != nil {
 				return err
 			}
-			setPointer(outResp, resp)
-			return nil
+
+			return setPointer(outResp, resp)
 		default:
 			panic("unreachable")
 		}
@@ -235,6 +233,20 @@ func ResponseFullNameFromMethodDesc(sd *grpc.ServiceDesc, method grpc.MethodDesc
 
 // since proto.Merge breaks due to the custom cosmos sdk any, we are forced to do this ugly setPointer hack.
 // ref: https://github.com/cosmos/cosmos-sdk/issues/22779
-func setPointer(dst, src any) {
-	reflect.ValueOf(dst).Elem().Set(reflect.ValueOf(src).Elem())
+func setPointer(dst, src any) error {
+	dstValue := reflect.ValueOf(dst)
+	srcValue := reflect.ValueOf(src)
+	if !dstValue.IsValid() || !srcValue.IsValid() {
+		return fmt.Errorf("dst and src must be valid")
+	}
+	if dstValue.IsNil() || srcValue.IsNil() {
+		return fmt.Errorf("dst and src must be non-nil")
+	}
+	dstElem := dstValue.Elem()
+	srcElem := srcValue.Elem()
+	if dstElem.Type() != srcElem.Type() {
+		return fmt.Errorf("dst and src must have the same type")
+	}
+	dstElem.Set(srcElem)
+	return nil
 }

scripts/mockgen.sh

@@ -15,7 +15,7 @@ $mockgen_cmd -source=x/mint/types/expected_keepers.go -package testutil -destina
 $mockgen_cmd -source=x/auth/tx/config/expected_keepers.go -package testutil -destination x/auth/tx/testutil/expected_keepers_mocks.go
 $mockgen_cmd -source=x/auth/types/expected_keepers.go -package testutil -destination x/auth/testutil/expected_keepers_mocks.go
 $mockgen_cmd -source=x/auth/ante/expected_keepers.go -package testutil -destination x/auth/ante/testutil/expected_keepers_mocks.go
-$mockgen_cmd -source=x/authz/expected_keepers.go -package testutil -destination x/authz/testutil/expected_keepers_mocks.go
+$mockgen_cmd -source=x/authz/testutil/expected_keepers.go -package testutil -destination x/authz/testutil/expected_keepers_mocks.go
 $mockgen_cmd -source=x/bank/types/expected_keepers.go -package testutil -destination x/bank/testutil/expected_keepers_mocks.go
 $mockgen_cmd -source=x/group/testutil/expected_keepers.go -package testutil -destination x/group/testutil/expected_keepers_mocks.go
 $mockgen_cmd -source=x/evidence/types/expected_keepers.go -package testutil -destination x/evidence/testutil/expected_keepers_mocks.go

x/authz/keeper/keeper_test.go

@@ -72,6 +72,7 @@ func (s *TestSuite) SetupTest() {
 	// gomock initializations
 	ctrl := gomock.NewController(s.T())
 	s.bankKeeper = authztestutil.NewMockBankKeeper(ctrl)
+	s.bankKeeper.EXPECT().Send(gomock.Any(), gomock.Any()).Return(&banktypes.MsgSendResponse{}, nil).AnyTimes()
 	banktypes.RegisterInterfaces(s.encCfg.InterfaceRegistry)
 	banktypes.RegisterMsgServer(s.baseApp.MsgServiceRouter(), s.bankKeeper)
 

x/authz/testutil/expected_keepers.go

@@ -0,0 +1,19 @@
+package testutil
+
+import (
+	"context"
+
+	banktypes "cosmossdk.io/x/bank/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+)
+
+// BankKeeper extends bank `MsgServer` to mock `Send` and to register handlers in MsgServiceRouter
+type BankKeeper interface {
+	banktypes.MsgServer
+
+	SpendableCoins(ctx context.Context, addr sdk.AccAddress) sdk.Coins
+	MintCoins(ctx context.Context, moduleName string, amt sdk.Coins) error
+	SendCoinsFromModuleToAccount(ctx context.Context, senderModule string, recipientAddr sdk.AccAddress, amt sdk.Coins) error
+	GetAllBalances(ctx context.Context, addr sdk.AccAddress) sdk.Coins
+}