From bd6c16b4628cf3383689893090d1f9ea741eac65 Mon Sep 17 00:00:00 2001
From: Marie Gauthier <marie.gauthier63@gmail.com>
Date: Fri, 30 Oct 2020 16:27:12 +0100
Subject: [PATCH] Handle ServiceMsg in StdTxBuilder.SetMsgs (#7731)

* Handle ServiceMsg in StdTxBuilder.SetMsgs

* Add test
---
 x/auth/client/rest/rest_test.go         | 50 +++++++++++++++++++++++--
 x/auth/legacy/legacytx/stdtx_builder.go | 20 +++++++++-
 2 files changed, 66 insertions(+), 4 deletions(-)

diff --git a/x/auth/client/rest/rest_test.go b/x/auth/client/rest/rest_test.go
index 1c8d492426..e06778e4e4 100644
--- a/x/auth/client/rest/rest_test.go
+++ b/x/auth/client/rest/rest_test.go
@@ -2,19 +2,22 @@ package rest_test
 
 import (
 	"fmt"
-	"testing"
-
 	"strings"
+	"testing"
 
 	"github.com/stretchr/testify/suite"
 
+	"github.com/cosmos/cosmos-sdk/client/flags"
 	"github.com/cosmos/cosmos-sdk/client/tx"
+	"github.com/cosmos/cosmos-sdk/crypto/hd"
+	"github.com/cosmos/cosmos-sdk/crypto/keyring"
 	"github.com/cosmos/cosmos-sdk/testutil/network"
 	"github.com/cosmos/cosmos-sdk/testutil/testdata"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/rest"
 	"github.com/cosmos/cosmos-sdk/types/tx/signing"
 	authclient "github.com/cosmos/cosmos-sdk/x/auth/client"
+	bankcli "github.com/cosmos/cosmos-sdk/x/bank/client/testutil"
 
 	rest2 "github.com/cosmos/cosmos-sdk/x/auth/client/rest"
 	"github.com/cosmos/cosmos-sdk/x/auth/legacy/legacytx"
@@ -37,7 +40,11 @@ func (s *IntegrationTestSuite) SetupSuite() {
 	s.cfg = cfg
 	s.network = network.New(s.T(), cfg)
 
-	_, err := s.network.WaitForHeight(1)
+	kb := s.network.Validators[0].ClientCtx.Keyring
+	_, _, err := kb.NewMnemonic("newAccount", keyring.English, sdk.FullFundraiserPath, hd.Secp256k1)
+	s.Require().NoError(err)
+
+	_, err = s.network.WaitForHeight(1)
 	s.Require().NoError(err)
 }
 
@@ -128,6 +135,43 @@ func (s *IntegrationTestSuite) TestQueryTxByHash() {
 	s.Require().True(strings.Contains(string(txJSON), stdTx.Memo))
 }
 
+func (s *IntegrationTestSuite) TestQueryTxByHashWithServiceMessage() {
+	val := s.network.Validators[0]
+
+	account, err := val.ClientCtx.Keyring.Key("newAccount")
+	s.Require().NoError(err)
+
+	sendTokens := sdk.NewInt64Coin(s.cfg.BondDenom, 10)
+
+	out, err := bankcli.ServiceMsgSendExec(
+		val.ClientCtx,
+		val.Address,
+		account.GetAddress(),
+		sdk.NewCoins(sendTokens),
+		fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
+		fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastBlock),
+		fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdk.NewInt(10))).String()),
+		fmt.Sprintf("--gas=%d", flags.DefaultGasLimit),
+	)
+
+	s.Require().NoError(err)
+	var txRes sdk.TxResponse
+	s.Require().NoError(val.ClientCtx.JSONMarshaler.UnmarshalJSON(out.Bytes(), &txRes))
+
+	s.Require().NoError(s.network.WaitForNextBlock())
+
+	txJSON, err := rest.GetRequest(fmt.Sprintf("%s/txs/%s", val.APIAddress, txRes.TxHash))
+	s.Require().NoError(err)
+
+	var result legacytx.StdTx
+	s.Require().NoError(val.ClientCtx.LegacyAmino.UnmarshalJSON(txJSON, &result))
+	s.Require().NotNil(result)
+	msgs := result.GetMsgs()
+	s.Require().Equal(len(msgs), 1)
+	_, ok := msgs[0].(*types.MsgSend)
+	s.Require().True(ok)
+}
+
 func (s *IntegrationTestSuite) TestMultipleSyncBroadcastTxRequests() {
 	// First test transaction from validator should have sequence=1 (non-genesis tx)
 	testCases := []struct {
diff --git a/x/auth/legacy/legacytx/stdtx_builder.go b/x/auth/legacy/legacytx/stdtx_builder.go
index 1b1a55c042..153c4b2bbb 100644
--- a/x/auth/legacy/legacytx/stdtx_builder.go
+++ b/x/auth/legacy/legacytx/stdtx_builder.go
@@ -29,7 +29,25 @@ func (s *StdTxBuilder) GetTx() authsigning.Tx {
 
 // SetMsgs implements TxBuilder.SetMsgs
 func (s *StdTxBuilder) SetMsgs(msgs ...sdk.Msg) error {
-	s.Msgs = msgs
+	stdTxMsgs := make([]sdk.Msg, len(msgs))
+
+	for i, msg := range msgs {
+		switch msg := msg.(type) {
+		case sdk.ServiceMsg:
+			// Since ServiceMsg isn't registered with amino, we unpack msg.Request
+			// into a Msg so that it's handled gracefully for the legacy
+			// GET /txs/{hash} and /txs endpoints.
+			sdkMsg, ok := msg.Request.(sdk.Msg)
+			if !ok {
+				return sdkerrors.Wrapf(sdkerrors.ErrInvalidType, "Expecting %T at %d to implement sdk.Msg", msg.Request, i)
+			}
+			stdTxMsgs[i] = sdkMsg
+		default:
+			// legacy sdk.Msg
+			stdTxMsgs[i] = msg
+		}
+	}
+	s.Msgs = stdTxMsgs
 	return nil
 }