package base_test

import (
	"github.com/ethereum/go-ethereum/core/types"
	. "github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"github.com/synapsecns/sanguine/ethergo/backends/base"
	confirmationMocks "github.com/synapsecns/sanguine/ethergo/backends/base/mocks"
	"github.com/synapsecns/sanguine/ethergo/mocks"
	"time"
)

func (b *BaseSuite) TestWaitForConfirmation() {
	mockClient := new(confirmationMocks.ConfirmationClient)

	mockTx := mocks.GetMockTxes(b.GetTestContext(), b.T(), 1, types.LegacyTxType)[0]

	const minConfirmTime = 60 * time.Millisecond
	confirmStart := time.Now()
	timer := time.NewTimer(minConfirmTime)

	mockClient.On("TransactionByHash", mock.Anything, mock.Anything).Once().Return(nil, true, nil)
	mockClient.On("TransactionReceipt", mock.Anything, mock.Anything).Return(nil, nil)
	mockClient.On("TransactionByHash", mock.Anything, mock.Anything).WaitUntil(timer.C).Return(mockTx, false, nil)
	mockClient.On("TransactionReceipt", mock.Anything, mock.Anything).Return(&types.Receipt{Status: types.ReceiptStatusSuccessful}, nil)

	base.WaitForConfirmation(b.GetTestContext(), mockClient, mockTx, time.Millisecond*5)
	False(b.T(), time.Since(confirmStart) < minConfirmTime, "tx could not have been confirmed yet")
}