e9534b0935
* Add gRPC proxy * Make GRPC disabled by default * WIP on integration tests * WIP on integration tests * Start setting up in process tests * Start setting up in process tests * Make it compile * Add start server to network util * Add Println * Use go routine * Fix scopelint * Move to proxy_test * Add response type cache * Remove proxy * Tweaks * Use channel to handle error * Use error chan * Update server/start.go Co-authored-by: Federico Kunze <31522760+fedekunze@users.noreply.github.com> * Use %w * Add sdk.Context * Add comments * Fix lint * Add header and tests * Address comments * Factorize some code * Fix lint * Add height and prove in req metadata * Add reflection test * Fix lint * Put grpc test in server/grpc * Update baseapp/grpcserver.go * Update baseapp/grpcserver.go * Remove proof header Co-authored-by: Amaury Martiny <amaury.martiny@protonmail.com> Co-authored-by: Federico Kunze <31522760+fedekunze@users.noreply.github.com> Co-authored-by: Alexander Bezobchuk <alexanderbez@users.noreply.github.com> Co-authored-by: SaReN <sahithnarahari@gmail.com>
102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package grpc_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
servergrpc "github.com/cosmos/cosmos-sdk/server/grpc"
|
|
"github.com/cosmos/cosmos-sdk/testutil/network"
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
rpb "google.golang.org/grpc/reflection/grpc_reflection_v1alpha"
|
|
)
|
|
|
|
type IntegrationTestSuite struct {
|
|
suite.Suite
|
|
|
|
network *network.Network
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) SetupSuite() {
|
|
s.T().Log("setting up integration test suite")
|
|
|
|
s.network = network.New(s.T(), network.DefaultConfig())
|
|
s.Require().NotNil(s.network)
|
|
|
|
_, err := s.network.WaitForHeight(2)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TearDownSuite() {
|
|
s.T().Log("tearing down integration test suite")
|
|
s.network.Cleanup()
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TestGRPC() {
|
|
val0 := s.network.Validators[0]
|
|
conn, err := grpc.Dial(
|
|
val0.AppConfig.GRPC.Address,
|
|
grpc.WithInsecure(), // Or else we get "no transport security set"
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// gRPC query to test service should work
|
|
testClient := testdata.NewTestServiceClient(conn)
|
|
testRes, err := testClient.Echo(context.Background(), &testdata.EchoRequest{Message: "hello"})
|
|
s.Require().NoError(err)
|
|
s.Require().Equal("hello", testRes.Message)
|
|
|
|
// gRPC query to bank service should work
|
|
denom := fmt.Sprintf("%stoken", val0.Moniker)
|
|
bankClient := banktypes.NewQueryClient(conn)
|
|
var header metadata.MD
|
|
bankRes, err := bankClient.Balance(
|
|
context.Background(),
|
|
&banktypes.QueryBalanceRequest{Address: val0.Address, Denom: denom},
|
|
grpc.Header(&header), // Also fetch grpc header
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(
|
|
sdk.NewCoin(denom, s.network.Config.AccountTokens),
|
|
*bankRes.GetBalance(),
|
|
)
|
|
blockHeight := header.Get(servergrpc.GRPCBlockHeightHeader)
|
|
s.Require().Equal([]string{"2"}, blockHeight)
|
|
|
|
// Request metadata should work
|
|
bankRes, err = bankClient.Balance(
|
|
metadata.AppendToOutgoingContext(context.Background(), servergrpc.GRPCBlockHeightHeader, "1"), // Add metadata to request
|
|
&banktypes.QueryBalanceRequest{Address: val0.Address, Denom: denom},
|
|
grpc.Header(&header),
|
|
)
|
|
blockHeight = header.Get(servergrpc.GRPCBlockHeightHeader)
|
|
s.Require().Equal([]string{"1"}, blockHeight)
|
|
|
|
// Test server reflection
|
|
reflectClient := rpb.NewServerReflectionClient(conn)
|
|
stream, err := reflectClient.ServerReflectionInfo(context.Background(), grpc.WaitForReady(true))
|
|
s.Require().NoError(err)
|
|
s.Require().NoError(stream.Send(&rpb.ServerReflectionRequest{
|
|
MessageRequest: &rpb.ServerReflectionRequest_ListServices{},
|
|
}))
|
|
res, err := stream.Recv()
|
|
s.Require().NoError(err)
|
|
services := res.GetListServicesResponse().Service
|
|
servicesMap := make(map[string]bool)
|
|
for _, s := range services {
|
|
servicesMap[s.Name] = true
|
|
}
|
|
// Make sure the following services are present
|
|
s.Require().True(servicesMap["cosmos.bank.Query"])
|
|
}
|
|
|
|
func TestIntegrationTestSuite(t *testing.T) {
|
|
suite.Run(t, new(IntegrationTestSuite))
|
|
}
|