synapsecns/sanguine

package anvil_test

import (
    "github.com/ethereum/go-ethereum/accounts/abi/bind"
    "github.com/ethereum/go-ethereum/core/types"
    "github.com/ethereum/go-ethereum/params"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/ethergo/backends/base"
    "github.com/synapsecns/sanguine/ethergo/util"
    "math/big"
)

// Make sure a hardforked rpc will have a balance.
func (a *AnvilSuite) TestFundAccount() {
    fundedAccount := base.MockAccount(a.T())
    ether := big.NewInt(params.Ether)
    a.backend.FundAccount(a.GetTestContext(), fundedAccount.Address, *ether)

    realBalance, err := a.backend.BalanceAt(a.GetTestContext(), fundedAccount.Address, nil)
    a.Require().NoError(err)

    Equal(a.T(), ether, realBalance)
}

func (a *AnvilSuite) TestGetTxContext() {
    res := a.backend.GetTxContext(a.GetTestContext(), nil)

    prevCount, err := a.counter.GetCount(&bind.CallOpts{Context: a.GetTestContext()})
    a.Require().NoError(err)

    res.TransactOpts.NoSend = true
    tx, err := a.counter.IncrementCounter(res.TransactOpts)
    a.Require().NoError(err)

    sender, err := util.TxToCall(tx)
    a.Require().NoError(err)

    Equal(a.T(), res.TransactOpts.From, sender.From)

    a.backend.WaitForConfirmation(a.GetTestContext(), tx)

    newCount, err := a.counter.GetCount(&bind.CallOpts{Context: a.GetTestContext()})
    a.Require().NoError(err)

    Equal(a.T(), prevCount.Uint64()+1, newCount.Uint64())
}

func (a *AnvilSuite) TestImpersonateAccount() {
    ogCount, err := a.counter.GetVitalikCount(&bind.CallOpts{Context: a.GetTestContext()})
    a.Require().NoError(err)

    // impersonate vitalik, and send the fren some eth
    err = a.backend.ImpersonateAccount(a.GetTestContext(), vitalik, func(transactOpts *bind.TransactOpts) *types.Transaction {
        tx, err := a.counter.VitalikIncrement(transactOpts)
        a.Require().NoError(err)

        return tx
    })
    a.Require().NoError(err)

    vitalikCount, err := a.counter.GetVitalikCount(&bind.CallOpts{Context: a.GetTestContext()})
    a.Require().NoError(err)

    Equal(a.T(), ogCount.Uint64()+10, vitalikCount.Uint64())
}