feat: adds error handling for out-of-gas panics in grpc query handlers. (#20945)

2024-07-16 09:44:42 +00:00
parent 77dd7329e8
commit 511fb07914
4 changed files with 117 additions and 0 deletions
@@ -0,0 +1,98 @@
+package grpc_test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+
+	_ "cosmossdk.io/x/accounts"
+	_ "cosmossdk.io/x/auth"
+	_ "cosmossdk.io/x/auth/tx/config"
+	_ "cosmossdk.io/x/bank"
+	banktypes "cosmossdk.io/x/bank/types"
+	_ "cosmossdk.io/x/consensus"
+	_ "cosmossdk.io/x/staking"
+
+	"github.com/cosmos/cosmos-sdk/codec"
+	"github.com/cosmos/cosmos-sdk/testutil/configurator"
+	"github.com/cosmos/cosmos-sdk/testutil/network"
+	"github.com/cosmos/cosmos-sdk/testutil/testdata"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
+)
+
+type IntegrationTestOutOfGasSuite struct {
+	suite.Suite
+
+	cfg     network.Config
+	network network.NetworkI
+	conn    *grpc.ClientConn
+}
+
+func (s *IntegrationTestOutOfGasSuite) SetupSuite() {
+	var err error
+	s.T().Log("setting up integration test suite")
+
+	s.cfg, err = network.DefaultConfigWithAppConfigWithQueryGasLimit(configurator.NewAppConfig(
+		configurator.AccountsModule(),
+		configurator.AuthModule(),
+		configurator.BankModule(),
+		configurator.GenutilModule(),
+		configurator.StakingModule(),
+		configurator.ConsensusModule(),
+		configurator.TxModule(),
+	), 10)
+	s.NoError(err)
+	s.cfg.NumValidators = 1
+
+	s.network, err = network.New(s.T(), s.T().TempDir(), s.cfg)
+	s.Require().NoError(err)
+
+	_, err = s.network.WaitForHeight(2)
+	s.Require().NoError(err)
+
+	val0 := s.network.GetValidators()[0]
+	s.conn, err = grpc.NewClient(
+		val0.GetAppConfig().GRPC.Address,
+		grpc.WithInsecure(), //nolint:staticcheck // ignore SA1019, we don't need to use a secure connection for tests
+		grpc.WithDefaultCallOptions(grpc.ForceCodec(codec.NewProtoCodec(s.cfg.InterfaceRegistry).GRPCCodec())),
+	)
+	s.Require().NoError(err)
+}
+
+func (s *IntegrationTestOutOfGasSuite) TearDownSuite() {
+	s.T().Log("tearing down integration test suite")
+	s.conn.Close()
+	s.network.Cleanup()
+}
+
+func (s *IntegrationTestOutOfGasSuite) TestGRPCServer_TestService() {
+	// gRPC query to test service should work
+	testClient := testdata.NewQueryClient(s.conn)
+	testRes, err := testClient.Echo(context.Background(), &testdata.EchoRequest{Message: "hello"})
+	s.Require().NoError(err)
+	s.Require().Equal("hello", testRes.Message)
+}
+
+func (s *IntegrationTestOutOfGasSuite) TestGRPCServer_BankBalance_OutOfGas() {
+	val0 := s.network.GetValidators()[0]
+
+	// gRPC query to bank service should work
+	denom := fmt.Sprintf("%stoken", val0.GetMoniker())
+	bankClient := banktypes.NewQueryClient(s.conn)
+	var header metadata.MD
+	_, err := bankClient.Balance(
+		context.Background(),
+		&banktypes.QueryBalanceRequest{Address: val0.GetAddress().String(), Denom: denom},
+		grpc.Header(&header), // Also fetch grpc header
+	)
+
+	s.Require().ErrorContains(err, sdkerrors.ErrOutOfGas.Error())
+}
+
+func TestIntegrationTestOutOfGasSuite(t *testing.T) {
+	suite.Run(t, new(IntegrationTestOutOfGasSuite))
+}