You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
1.9 KiB
85 lines
1.9 KiB
package abciclient_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"math/rand"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
abciclient "github.com/tendermint/tendermint/abci/client"
|
|
"github.com/tendermint/tendermint/abci/server"
|
|
"github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
"github.com/tendermint/tendermint/libs/service"
|
|
)
|
|
|
|
func TestProperSyncCalls(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
app := slowApp{}
|
|
logger := log.TestingLogger()
|
|
|
|
_, c := setupClientServer(ctx, t, logger, app)
|
|
|
|
resp := make(chan error, 1)
|
|
go func() {
|
|
// This is BeginBlockSync unrolled....
|
|
reqres, err := c.BeginBlockAsync(ctx, types.RequestBeginBlock{})
|
|
assert.NoError(t, err)
|
|
err = c.FlushSync(ctx)
|
|
assert.NoError(t, err)
|
|
res := reqres.Response.GetBeginBlock()
|
|
assert.NotNil(t, res)
|
|
resp <- c.Error()
|
|
}()
|
|
|
|
select {
|
|
case <-time.After(time.Second):
|
|
require.Fail(t, "No response arrived")
|
|
case err, ok := <-resp:
|
|
require.True(t, ok, "Must not close channel")
|
|
assert.NoError(t, err, "This should return success")
|
|
}
|
|
}
|
|
|
|
func setupClientServer(
|
|
ctx context.Context,
|
|
t *testing.T,
|
|
logger log.Logger,
|
|
app types.Application,
|
|
) (service.Service, abciclient.Client) {
|
|
t.Helper()
|
|
|
|
// some port between 20k and 30k
|
|
port := 20000 + rand.Int31()%10000
|
|
addr := fmt.Sprintf("localhost:%d", port)
|
|
|
|
s, err := server.NewServer(logger, addr, "socket", app)
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.Start(ctx))
|
|
t.Cleanup(s.Wait)
|
|
|
|
c := abciclient.NewSocketClient(logger, addr, true)
|
|
require.NoError(t, c.Start(ctx))
|
|
t.Cleanup(c.Wait)
|
|
|
|
require.True(t, s.IsRunning())
|
|
require.True(t, c.IsRunning())
|
|
|
|
return s, c
|
|
}
|
|
|
|
type slowApp struct {
|
|
types.BaseApplication
|
|
}
|
|
|
|
func (slowApp) BeginBlock(req types.RequestBeginBlock) types.ResponseBeginBlock {
|
|
time.Sleep(200 * time.Millisecond)
|
|
return types.ResponseBeginBlock{}
|
|
}
|