package mock_test

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/tendermint/tendermint/libs/bytes"
	"github.com/tendermint/tendermint/rpc/client/mock"
	"github.com/tendermint/tendermint/rpc/coretypes"
)

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

	m := &mock.StatusMock{
		Call: mock.Call{
			Response: &coretypes.ResultStatus{
				SyncInfo: coretypes.SyncInfo{
					LatestBlockHash:     bytes.HexBytes("block"),
					LatestAppHash:       bytes.HexBytes("app"),
					LatestBlockHeight:   10,
					MaxPeerBlockHeight:  20,
					TotalSyncedTime:     time.Second,
					RemainingTime:       time.Minute,
					TotalSnapshots:      10,
					ChunkProcessAvgTime: time.Duration(10),
					SnapshotHeight:      10,
					SnapshotChunksCount: 9,
					SnapshotChunksTotal: 10,
					BackFilledBlocks:    9,
					BackFillBlocksTotal: 10,
				},
			}},
	}

	r := mock.NewStatusRecorder(m)
	require.Equal(t, 0, len(r.Calls))

	// make sure response works proper
	status, err := r.Status(ctx)
	require.NoError(t, err)
	assert.EqualValues(t, "block", status.SyncInfo.LatestBlockHash)
	assert.EqualValues(t, 10, status.SyncInfo.LatestBlockHeight)
	assert.EqualValues(t, 20, status.SyncInfo.MaxPeerBlockHeight)
	assert.EqualValues(t, time.Second, status.SyncInfo.TotalSyncedTime)
	assert.EqualValues(t, time.Minute, status.SyncInfo.RemainingTime)

	// make sure recorder works properly
	require.Equal(t, 1, len(r.Calls))
	rs := r.Calls[0]
	assert.Equal(t, "status", rs.Name)
	assert.Nil(t, rs.Args)
	assert.Nil(t, rs.Error)
	require.NotNil(t, rs.Response)
	st, ok := rs.Response.(*coretypes.ResultStatus)
	require.True(t, ok)
	assert.EqualValues(t, "block", st.SyncInfo.LatestBlockHash)
	assert.EqualValues(t, 10, st.SyncInfo.LatestBlockHeight)
	assert.EqualValues(t, 20, st.SyncInfo.MaxPeerBlockHeight)
	assert.EqualValues(t, time.Second, status.SyncInfo.TotalSyncedTime)
	assert.EqualValues(t, time.Minute, status.SyncInfo.RemainingTime)

	assert.EqualValues(t, 10, st.SyncInfo.TotalSnapshots)
	assert.EqualValues(t, time.Duration(10), st.SyncInfo.ChunkProcessAvgTime)
	assert.EqualValues(t, 10, st.SyncInfo.SnapshotHeight)
	assert.EqualValues(t, 9, status.SyncInfo.SnapshotChunksCount)
	assert.EqualValues(t, 10, status.SyncInfo.SnapshotChunksTotal)
	assert.EqualValues(t, 9, status.SyncInfo.BackFilledBlocks)
	assert.EqualValues(t, 10, status.SyncInfo.BackFillBlocksTotal)
}