synapsecns/sanguine

package etherscan_test

import (
    "github.com/Flaque/filet"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/tools/abigen/internal/etherscan"
    "time"
)

var waitTime = time.Second

func (s EtherscanSuite) TestRateLimiter() {
    lockPath := filet.TmpDir(s.T(), "")

    rateLimiter, err := etherscan.NewFileRateLimiter(s.GetTestContext(), lockPath, waitTime)
    Nil(s.T(), err)

    for lockCount := 0; lockCount < 2; lockCount++ {
        expectedEndTime := time.Now().Add(waitTime)
        // obtain lock obtains the lcok
        ok, err := rateLimiter.ObtainLock(s.GetTestContext())
        True(s.T(), ok)
        Nil(s.T(), err)

        // release lock releases the lcok
        ok, err = rateLimiter.ReleaseLock()
        True(s.T(), ok)
        Nil(s.T(), err)

        GreaterOrEqual(s.T(), expectedEndTime.UnixNano(), time.Now().UnixNano())
    }
}