synapsecns/sanguine

package testutil_test

import (
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/ethergo/backends/simulated"
    "github.com/synapsecns/sanguine/services/rfq/testutil"

    "math/big"
)

func (s *TestUtilSuite) TestDecimals() {
    simulatedBackend := simulated.NewSimulatedBackend(s.GetTestContext(), s.T())
    deployHelper := testutil.NewDeployManager(s.T())

    // test out the usdt token (non-standard, 6 decimal)
    _, usdtRef := deployHelper.GetUSDT(s.GetTestContext(), simulatedBackend)
    s.adjustAmountTestDecimals(usdtRef, 1e6)

    // test out usdc token (standard, 6 decimal)
    _, usdcRef := deployHelper.GetUSDC(s.GetTestContext(), simulatedBackend)
    s.adjustAmountTestDecimals(usdcRef, 1e6)

    _, daiRef := deployHelper.GetDAI(s.GetTestContext(), simulatedBackend)
    s.adjustAmountTestDecimals(daiRef, 1e18)
}

func (s *TestUtilSuite) adjustAmountTestDecimals(handler interface{}, multiplier uint64) {
    amount, err := testutil.AdjustAmount(s.GetTestContext(), big.NewInt(1), handler)
    Nil(s.T(), err)
    Equal(s.T(), amount.Uint64(), 1*multiplier)

    // make sure amount is accounted for
    amount, err = testutil.AdjustAmount(s.GetTestContext(), big.NewInt(2), handler)
    Nil(s.T(), err)
    Equal(s.T(), amount.Uint64(), 2*multiplier)

    amount, err = testutil.AdjustAmount(s.GetTestContext(), big.NewInt(0), handler)
    Nil(s.T(), err)
    Equal(s.T(), amount.Uint64(), uint64(0))
}