tendermint/internal/proxy/app_conn_test.go

package proxy

import (
	"context"
	"fmt"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
	abciclient "github.com/tendermint/tendermint/abci/client"
	"github.com/tendermint/tendermint/abci/example/kvstore"
	"github.com/tendermint/tendermint/abci/server"
	"github.com/tendermint/tendermint/abci/types"
	"github.com/tendermint/tendermint/libs/log"
	tmrand "github.com/tendermint/tendermint/libs/rand"
)

//----------------------------------------

type appConnTestI interface {
	EchoAsync(ctx context.Context, msg string) (*abciclient.ReqRes, error)
	FlushSync(context.Context) error
	InfoSync(context.Context, types.RequestInfo) (*types.ResponseInfo, error)
}

type appConnTest struct {
	appConn abciclient.Client
}

func newAppConnTest(appConn abciclient.Client) appConnTestI {
	return &appConnTest{appConn}
}

func (app *appConnTest) EchoAsync(ctx context.Context, msg string) (*abciclient.ReqRes, error) {
	return app.appConn.EchoAsync(ctx, msg)
}

func (app *appConnTest) FlushSync(ctx context.Context) error {
	return app.appConn.FlushSync(ctx)
}

func (app *appConnTest) InfoSync(ctx context.Context, req types.RequestInfo) (*types.ResponseInfo, error) {
	return app.appConn.InfoSync(ctx, req)
}

//----------------------------------------

var SOCKET = "socket"

func TestEcho(t *testing.T) {
	sockPath := fmt.Sprintf("unix:///tmp/echo_%v.sock", tmrand.Str(6))
	logger := log.TestingLogger()
	clientCreator := abciclient.NewRemoteCreator(logger, sockPath, SOCKET, true)

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	// Start server
	s := server.NewSocketServer(logger.With("module", "abci-server"), sockPath, kvstore.NewApplication())
	require.NoError(t, s.Start(ctx), "error starting socket server")
	t.Cleanup(func() { cancel(); s.Wait() })

	// Start client
	cli, err := clientCreator(logger.With("module", "abci-client"))
	require.NoError(t, err, "Error creating ABCI client:")

	require.NoError(t, cli.Start(ctx), "Error starting ABCI client")

	proxy := newAppConnTest(cli)
	t.Log("Connected")

	for i := 0; i < 1000; i++ {
		_, err = proxy.EchoAsync(ctx, fmt.Sprintf("echo-%v", i))
		if err != nil {
			t.Error(err)
		}
		// flush sometimes
		if i%128 == 0 {
			if err := proxy.FlushSync(ctx); err != nil {
				t.Error(err)
			}
		}
	}
	if err := proxy.FlushSync(ctx); err != nil {
		t.Error(err)
	}
}

func BenchmarkEcho(b *testing.B) {
	b.StopTimer() // Initialize
	sockPath := fmt.Sprintf("unix:///tmp/echo_%v.sock", tmrand.Str(6))
	logger := log.TestingLogger()
	clientCreator := abciclient.NewRemoteCreator(logger, sockPath, SOCKET, true)

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	// Start server
	s := server.NewSocketServer(logger.With("module", "abci-server"), sockPath, kvstore.NewApplication())
	require.NoError(b, s.Start(ctx), "Error starting socket server")
	b.Cleanup(func() { cancel(); s.Wait() })

	// Start client
	cli, err := clientCreator(logger.With("module", "abci-client"))
	require.NoError(b, err, "Error creating ABCI client")

	require.NoError(b, cli.Start(ctx), "Error starting ABCI client")

	proxy := newAppConnTest(cli)
	b.Log("Connected")
	echoString := strings.Repeat(" ", 200)
	b.StartTimer() // Start benchmarking tests

	for i := 0; i < b.N; i++ {
		_, err = proxy.EchoAsync(ctx, echoString)
		if err != nil {
			b.Error(err)
		}
		// flush sometimes
		if i%128 == 0 {
			if err := proxy.FlushSync(ctx); err != nil {
				b.Error(err)
			}
		}
	}
	if err := proxy.FlushSync(ctx); err != nil {
		b.Error(err)
	}

	b.StopTimer()
	// info := proxy.InfoSync(types.RequestInfo{""})
	// b.Log("N: ", b.N, info)
}

func TestInfo(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	sockPath := fmt.Sprintf("unix:///tmp/echo_%v.sock", tmrand.Str(6))
	logger := log.TestingLogger()
	clientCreator := abciclient.NewRemoteCreator(logger, sockPath, SOCKET, true)

	// Start server
	s := server.NewSocketServer(logger.With("module", "abci-server"), sockPath, kvstore.NewApplication())
	require.NoError(t, s.Start(ctx), "Error starting socket server")
	t.Cleanup(func() { cancel(); s.Wait() })

	// Start client
	cli, err := clientCreator(logger.With("module", "abci-client"))
	require.NoError(t, err, "Error creating ABCI client")

	require.NoError(t, cli.Start(ctx), "Error starting ABCI client")

	proxy := newAppConnTest(cli)
	t.Log("Connected")

	resInfo, err := proxy.InfoSync(ctx, RequestInfo)
	require.NoError(t, err)

	if resInfo.Data != "{\"size\":0}" {
		t.Error("Expected ResponseInfo with one element '{\"size\":0}' but got something else")
	}
}