synapsecns/sanguine

package nonce

import (
    "context"
    "math/big"
    "testing"

    "github.com/ethereum/go-ethereum/common"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/ethergo/chain"
)

// TestManager exports the nonce manager with additional methods required only for testing.
type TestManager interface {
    Manager
    AssertNoncesEqual(address common.Address, equalTo int64)
    GetChainID() *big.Int
}

// testManagerImpl implements the nonce manager.
type testManagerImpl struct {
    // the underlying nonce manager object
    *nonceManagerImp
    // tb is used for test assertions
    tb      testing.TB
    chainID *big.Int
}

func (t *testManagerImpl) GetChainID() *big.Int {
    return t.chainID
}

// NewTestNonceManger wraps NewNonceManager w/ newly exported methods for testing.
//
//nolint:staticcheck
func NewTestNonceManger(ctx context.Context, tb testing.TB, chain chain.Chain) TestManager {
    tb.Helper()
    manager := NewNonceManager(ctx, chain, chain.GetBigChainID())
    castManager, ok := manager.(*nonceManagerImp)
    True(tb, ok)
    return &testManagerImpl{
        nonceManagerImp: castManager,
        tb:              tb,
        chainID:         chain.GetBigChainID(),
    }
}

// AssertNoncesEqual asserts the nonce for an address is equal to the passed in nonce.
func (t *testManagerImpl) AssertNoncesEqual(address common.Address, equalTo int64) {
    nonce, err := t.GetNextNonce(address)
    Nil(t.tb, err)

    // should be 0
    equalityCheck := nonce.Cmp(big.NewInt(equalTo))
    Equal(t.tb, equalityCheck, 0)
}