synapsecns/sanguine

package watcher

import (
    "context"
    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/core/types"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/ethergo/chain/chainwatcher"
    "testing"
    "time"
)

// defaultPollInterval gets the original poll interval so it can ber reset after the test.
var defaultPollInterval time.Duration

func init() {
    defaultPollInterval = chainwatcher.PollInterval
}

// GetDefaultPollInterval gets the default poll interval so it can be set for tests.
func GetDefaultPollInterval() time.Duration {
    return defaultPollInterval
}

// TestContractWatcher is a wrapped contract watcher with exported methods for testing.
type TestContractWatcher interface {
    chainwatcher.ContractWatcher
    AddListener(address common.Address, ch chan interface{})
    AddProducer(ctx context.Context, address common.Address, producedEvents <-chan types.Log) error
}

// NewTestContractWatcher exports contractWatcherImpl methods for testing.
func NewTestContractWatcher(ctx context.Context, tb testing.TB, contractListener ContractFilterer, heightWatcher chainwatcher.BlockHeightWatcher, requiredConfirmations uint) TestContractWatcher {
    tb.Helper()
    contractWatcher := NewContractWatcher(ctx, contractListener, heightWatcher, requiredConfirmations)
    castWatcher, ok := contractWatcher.(*contractWatcherImpl)
    True(tb, ok)
    return castWatcher
}

// AddListener adds a listener for testing.
func (c *contractWatcherImpl) AddListener(address common.Address, ch chan interface{}) {
    c.addListener(address, ch)
}

// AddProducer exports addProducer for testing.
func (c *contractWatcherImpl) AddProducer(ctx context.Context, address common.Address, producedEvents <-chan types.Log) error {
    return c.addProducer(ctx, address, producedEvents)
}