2020-10-29 08:32:47 -07:00
|
|
|
package baseapp_test
|
2020-06-03 06:33:51 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2020-10-29 08:32:47 -07:00
|
|
|
"os"
|
2021-10-16 01:21:12 -07:00
|
|
|
"sync"
|
2020-06-03 06:33:51 -07:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/require"
|
2020-10-29 08:32:47 -07:00
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
dbm "github.com/tendermint/tm-db"
|
2020-06-03 06:33:51 -07:00
|
|
|
|
2020-10-29 08:32:47 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
2020-09-21 09:48:28 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/codec/types"
|
2020-10-29 08:32:47 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
2020-07-20 05:30:12 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
2020-06-03 06:33:51 -07:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
)
|
|
|
|
|
2020-12-04 07:06:50 -08:00
|
|
|
func TestGRPCGatewayRouter(t *testing.T) {
|
2020-10-29 08:32:47 -07:00
|
|
|
qr := baseapp.NewGRPCQueryRouter()
|
2020-07-03 09:42:12 -07:00
|
|
|
interfaceRegistry := testdata.NewTestInterfaceRegistry()
|
2020-08-17 10:02:13 -07:00
|
|
|
qr.SetInterfaceRegistry(interfaceRegistry)
|
2020-10-15 06:07:59 -07:00
|
|
|
testdata.RegisterQueryServer(qr, testdata.QueryImpl{})
|
2020-10-29 08:32:47 -07:00
|
|
|
helper := &baseapp.QueryServiceTestHelper{
|
2020-06-04 11:48:06 -07:00
|
|
|
GRPCQueryRouter: qr,
|
2020-10-29 08:32:47 -07:00
|
|
|
Ctx: sdk.Context{}.WithContext(context.Background()),
|
2020-06-03 06:33:51 -07:00
|
|
|
}
|
2020-10-15 06:07:59 -07:00
|
|
|
client := testdata.NewQueryClient(helper)
|
2020-06-03 06:33:51 -07:00
|
|
|
|
|
|
|
res, err := client.Echo(context.Background(), &testdata.EchoRequest{Message: "hello"})
|
|
|
|
require.Nil(t, err)
|
|
|
|
require.NotNil(t, res)
|
|
|
|
require.Equal(t, "hello", res.Message)
|
|
|
|
|
|
|
|
require.Panics(t, func() {
|
|
|
|
_, _ = client.Echo(context.Background(), nil)
|
|
|
|
})
|
|
|
|
|
|
|
|
res2, err := client.SayHello(context.Background(), &testdata.SayHelloRequest{Name: "Foo"})
|
|
|
|
require.Nil(t, err)
|
|
|
|
require.NotNil(t, res)
|
|
|
|
require.Equal(t, "Hello Foo!", res2.Greeting)
|
2020-07-03 09:42:12 -07:00
|
|
|
|
|
|
|
spot := &testdata.Dog{Name: "Spot", Size_: "big"}
|
|
|
|
any, err := types.NewAnyWithValue(spot)
|
|
|
|
require.NoError(t, err)
|
|
|
|
res3, err := client.TestAny(context.Background(), &testdata.TestAnyRequest{AnyAnimal: any})
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, res3)
|
|
|
|
require.Equal(t, spot, res3.HasAnimal.Animal.GetCachedValue())
|
2020-06-03 06:33:51 -07:00
|
|
|
}
|
2020-10-29 08:32:47 -07:00
|
|
|
|
|
|
|
func TestRegisterQueryServiceTwice(t *testing.T) {
|
|
|
|
// Setup baseapp.
|
|
|
|
db := dbm.NewMemDB()
|
|
|
|
encCfg := simapp.MakeTestEncodingConfig()
|
|
|
|
app := baseapp.NewBaseApp("test", log.NewTMLogger(log.NewSyncWriter(os.Stdout)), db, encCfg.TxConfig.TxDecoder())
|
|
|
|
app.SetInterfaceRegistry(encCfg.InterfaceRegistry)
|
|
|
|
testdata.RegisterInterfaces(encCfg.InterfaceRegistry)
|
|
|
|
|
|
|
|
// First time registering service shouldn't panic.
|
|
|
|
require.NotPanics(t, func() {
|
|
|
|
testdata.RegisterQueryServer(
|
|
|
|
app.GRPCQueryRouter(),
|
|
|
|
testdata.QueryImpl{},
|
|
|
|
)
|
|
|
|
})
|
|
|
|
|
|
|
|
// Second time should panic.
|
|
|
|
require.Panics(t, func() {
|
|
|
|
testdata.RegisterQueryServer(
|
|
|
|
app.GRPCQueryRouter(),
|
|
|
|
testdata.QueryImpl{},
|
|
|
|
)
|
|
|
|
})
|
|
|
|
}
|
2021-10-16 01:21:12 -07:00
|
|
|
|
|
|
|
// Tests that we don't have data races per
|
|
|
|
// https://github.com/cosmos/cosmos-sdk/issues/10324
|
|
|
|
// but with the same client connection being used concurrently.
|
|
|
|
func TestQueryDataRaces_sameConnectionToSameHandler(t *testing.T) {
|
|
|
|
var mu sync.Mutex
|
|
|
|
var helper *baseapp.QueryServiceTestHelper
|
|
|
|
makeClientConn := func(qr *baseapp.GRPCQueryRouter) *baseapp.QueryServiceTestHelper {
|
|
|
|
mu.Lock()
|
|
|
|
defer mu.Unlock()
|
|
|
|
|
|
|
|
if helper == nil {
|
|
|
|
helper = &baseapp.QueryServiceTestHelper{
|
|
|
|
GRPCQueryRouter: qr,
|
|
|
|
Ctx: sdk.Context{}.WithContext(context.Background()),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return helper
|
|
|
|
}
|
|
|
|
testQueryDataRacesSameHandler(t, makeClientConn)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Tests that we don't have data races per
|
|
|
|
// https://github.com/cosmos/cosmos-sdk/issues/10324
|
|
|
|
// but with unique client connections requesting from the same handler concurrently.
|
|
|
|
func TestQueryDataRaces_uniqueConnectionsToSameHandler(t *testing.T) {
|
|
|
|
// Return a new handler for every single call.
|
|
|
|
testQueryDataRacesSameHandler(t, func(qr *baseapp.GRPCQueryRouter) *baseapp.QueryServiceTestHelper {
|
|
|
|
return &baseapp.QueryServiceTestHelper{
|
|
|
|
GRPCQueryRouter: qr,
|
|
|
|
Ctx: sdk.Context{}.WithContext(context.Background()),
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func testQueryDataRacesSameHandler(t *testing.T, makeClientConn func(*baseapp.GRPCQueryRouter) *baseapp.QueryServiceTestHelper) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
qr := baseapp.NewGRPCQueryRouter()
|
|
|
|
interfaceRegistry := testdata.NewTestInterfaceRegistry()
|
|
|
|
qr.SetInterfaceRegistry(interfaceRegistry)
|
|
|
|
testdata.RegisterQueryServer(qr, testdata.QueryImpl{})
|
|
|
|
|
|
|
|
// The goal is to invoke the router concurrently and check for any data races.
|
|
|
|
// 0. Run with: go test -race
|
|
|
|
// 1. Synchronize every one of the 1,000 goroutines waiting to all query at the
|
|
|
|
// same time.
|
|
|
|
// 2. Once the greenlight is given, perform a query through the router.
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
defer wg.Wait()
|
|
|
|
|
|
|
|
greenlight := make(chan bool)
|
|
|
|
n := 1000
|
|
|
|
ready := make(chan bool, n)
|
|
|
|
go func() {
|
|
|
|
for i := 0; i < n; i++ {
|
|
|
|
<-ready
|
|
|
|
}
|
|
|
|
close(greenlight)
|
|
|
|
}()
|
|
|
|
|
|
|
|
for i := 0; i < n; i++ {
|
|
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
|
|
defer wg.Done()
|
|
|
|
|
|
|
|
// Wait until we get the green light to start.
|
|
|
|
ready <- true
|
|
|
|
<-greenlight
|
|
|
|
|
|
|
|
client := testdata.NewQueryClient(makeClientConn(qr))
|
|
|
|
res, err := client.Echo(context.Background(), &testdata.EchoRequest{Message: "hello"})
|
|
|
|
require.Nil(t, err)
|
|
|
|
require.NotNil(t, res)
|
|
|
|
require.Equal(t, "hello", res.Message)
|
|
|
|
|
|
|
|
require.Panics(t, func() {
|
|
|
|
_, _ = client.Echo(context.Background(), nil)
|
|
|
|
})
|
|
|
|
|
|
|
|
res2, err := client.SayHello(context.Background(), &testdata.SayHelloRequest{Name: "Foo"})
|
|
|
|
require.Nil(t, err)
|
|
|
|
require.NotNil(t, res)
|
|
|
|
require.Equal(t, "Hello Foo!", res2.Greeting)
|
|
|
|
|
|
|
|
spot := &testdata.Dog{Name: "Spot", Size_: "big"}
|
|
|
|
any, err := types.NewAnyWithValue(spot)
|
|
|
|
require.NoError(t, err)
|
|
|
|
res3, err := client.TestAny(context.Background(), &testdata.TestAnyRequest{AnyAnimal: any})
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, res3)
|
|
|
|
require.Equal(t, spot, res3.HasAnimal.Animal.GetCachedValue())
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
}
|