prysm/testing/endtoend/components/eth1/helpers.go

package eth1

import (
	"context"
	"math/big"
	"time"

	"github.com/OffchainLabs/prysm/v7/testing/endtoend/params"
	e2etypes "github.com/OffchainLabs/prysm/v7/testing/endtoend/types"
	"github.com/ethereum/go-ethereum/accounts/keystore"
	"github.com/ethereum/go-ethereum/core/types"
	"github.com/ethereum/go-ethereum/ethclient"
)

// NetworkId is the ID of the ETH1 chain.
const NetworkId = 1337

// KeystorePassword is the password used to decrypt ETH1 keystores.
const KeystorePassword = "password"

const minerPasswordFile = "password.txt"
const minerFile = "UTC--2021-12-22T19-14-08.590377700Z--878705ba3f8bc32fcf7f4caa1a35e72af65cf766"
const timeGapPerMiningTX = 250 * time.Millisecond

var _ e2etypes.ComponentRunner = (*NodeSet)(nil)
var _ e2etypes.MultipleComponentRunners = (*NodeSet)(nil)
var _ e2etypes.MultipleComponentRunners = (*ProxySet)(nil)
var _ e2etypes.ComponentRunner = (*Miner)(nil)
var _ e2etypes.ComponentRunner = (*Node)(nil)
var _ e2etypes.EngineProxy = (*Proxy)(nil)

// WaitForBlocks waits for a certain amount of blocks to be mined by the ETH1 chain before returning.
func WaitForBlocks(ctx context.Context, web3 *ethclient.Client, key *keystore.Key, blocksToWait uint64) error {
	nonce, err := web3.PendingNonceAt(ctx, key.Address)
	if err != nil {
		return err
	}
	chainID, err := web3.NetworkID(ctx)
	if err != nil {
		return err
	}
	block, err := web3.BlockByNumber(ctx, nil)
	if err != nil {
		return err
	}
	finishBlock := block.NumberU64() + blocksToWait

	for block.NumberU64() <= finishBlock {
		if ctx.Err() != nil {
			return ctx.Err()
		}
		gasPrice, err := web3.SuggestGasPrice(ctx)
		if err != nil {
			return err
		}
		spamTX := types.NewTransaction(nonce, key.Address, big.NewInt(0), params.SpamTxGasLimit, gasPrice, []byte{})
		signed, err := types.SignTx(spamTX, types.NewEIP155Signer(chainID), key.PrivateKey)
		if err != nil {
			return err
		}
		if err = web3.SendTransaction(ctx, signed); err != nil {
			return err
		}
		nonce++
		time.Sleep(timeGapPerMiningTX)
		block, err = web3.BlockByNumber(ctx, nil)
		if err != nil {
			return err
		}
	}
	return nil
}