tendermint/rpc/client/helpers_test.go

package client_test

import (
	"errors"
	"strings"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/tendermint/tendermint/rpc/client"
	"github.com/tendermint/tendermint/rpc/client/mock"
	ctypes "github.com/tendermint/tendermint/rpc/core/types"
)

func TestWaitForHeight(t *testing.T) {
	assert, require := assert.New(t), require.New(t)

	// test with error result - immediate failure
	m := &mock.StatusMock{
		Call: mock.Call{
			Error: errors.New("bye"),
		},
	}
	r := mock.NewStatusRecorder(m)

	// connection failure always leads to error
	err := client.WaitForHeight(r, 8)
	require.NotNil(err)
	require.Equal("bye", err.Error())
	// we called status once to check
	require.Equal(1, len(r.Calls))

	// now set current block height to 10
	m.Call = mock.Call{
		Response: &ctypes.ResultStatus{LatestBlockHeight: 10},
	}

	// we will not wait for more than 10 blocks
	err = client.WaitForHeight(r, 40)
	require.NotNil(err)
	require.True(strings.Contains(err.Error(), "aborting"))
	// we called status once more to check
	require.Equal(2, len(r.Calls))

	// waiting for the past returns immediately
	err = client.WaitForHeight(r, 5)
	require.Nil(err)
	// we called status once more to check
	require.Equal(3, len(r.Calls))

	// next tick in background while we wait
	go func() {
		time.Sleep(500 * time.Millisecond)
		m.Call.Response = &ctypes.ResultStatus{LatestBlockHeight: 15}
	}()

	// we wait for a few blocks
	err = client.WaitForHeight(r, 12)
	require.Nil(err)
	// we called status once to check
	require.Equal(5, len(r.Calls))

	pre := r.Calls[3]
	require.Nil(pre.Error)
	prer, ok := pre.Response.(*ctypes.ResultStatus)
	require.True(ok)
	assert.Equal(10, prer.LatestBlockHeight)

	post := r.Calls[4]
	require.Nil(post.Error)
	postr, ok := post.Response.(*ctypes.ResultStatus)
	require.True(ok)
	assert.Equal(15, postr.LatestBlockHeight)
}