123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
package rpc_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/rpc"
|
|
clitestutil "github.com/cosmos/cosmos-sdk/testutil/cli"
|
|
"github.com/cosmos/cosmos-sdk/testutil/network"
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
grpctypes "github.com/cosmos/cosmos-sdk/types/grpc"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
)
|
|
|
|
type IntegrationTestSuite struct {
|
|
suite.Suite
|
|
|
|
network *network.Network
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) SetupSuite() {
|
|
s.T().Log("setting up integration test suite")
|
|
|
|
cfg, err := network.DefaultConfigWithAppConfig(network.MinimumAppConfig())
|
|
|
|
s.NoError(err)
|
|
|
|
s.network, err = network.New(s.T(), s.T().TempDir(), cfg)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TearDownSuite() {
|
|
s.T().Log("tearing down integration test suite")
|
|
s.network.Cleanup()
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TestStatusCommand() {
|
|
val0 := s.network.Validators[0]
|
|
cmd := rpc.StatusCommand()
|
|
|
|
out, err := clitestutil.ExecTestCLICmd(val0.ClientCtx, cmd, []string{})
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure the output has the validator moniker.
|
|
s.Require().Contains(out.String(), fmt.Sprintf("\"moniker\":\"%s\"", val0.Moniker))
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TestCLIQueryConn() {
|
|
var header metadata.MD
|
|
|
|
testClient := testdata.NewQueryClient(s.network.Validators[0].ClientCtx)
|
|
res, err := testClient.Echo(context.Background(), &testdata.EchoRequest{Message: "hello"}, grpc.Header(&header))
|
|
s.NoError(err)
|
|
|
|
blockHeight := header.Get(grpctypes.GRPCBlockHeightHeader)
|
|
s.Require().Equal([]string{"1"}, blockHeight)
|
|
|
|
s.Equal("hello", res.Message)
|
|
}
|
|
|
|
func (s *IntegrationTestSuite) TestQueryABCIHeight() {
|
|
testCases := []struct {
|
|
name string
|
|
reqHeight int64
|
|
ctxHeight int64
|
|
expHeight int64
|
|
}{
|
|
{
|
|
name: "non zero request height",
|
|
reqHeight: 3,
|
|
ctxHeight: 1, // query at height 1 or 2 would cause an error
|
|
expHeight: 3,
|
|
},
|
|
{
|
|
name: "empty request height - use context height",
|
|
reqHeight: 0,
|
|
ctxHeight: 3,
|
|
expHeight: 3,
|
|
},
|
|
{
|
|
name: "empty request height and context height - use latest height",
|
|
reqHeight: 0,
|
|
ctxHeight: 0,
|
|
expHeight: 4,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
s.Run(tc.name, func() {
|
|
s.network.WaitForHeight(tc.expHeight)
|
|
|
|
val := s.network.Validators[0]
|
|
|
|
clientCtx := val.ClientCtx
|
|
clientCtx = clientCtx.WithHeight(tc.ctxHeight)
|
|
|
|
req := abci.RequestQuery{
|
|
Path: fmt.Sprintf("store/%s/key", banktypes.StoreKey),
|
|
Height: tc.reqHeight,
|
|
Data: banktypes.CreateAccountBalancesPrefix(val.Address),
|
|
Prove: true,
|
|
}
|
|
|
|
res, err := clientCtx.QueryABCI(req)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(tc.expHeight, res.Height)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIntegrationTestSuite(t *testing.T) {
|
|
suite.Run(t, new(IntegrationTestSuite))
|
|
}
|