erigon-pulse/cmd/devnet/tests/context.go

package tests

import (
	"fmt"
	"os"
	"runtime"
	"testing"

	"github.com/ledgerwatch/erigon-lib/chain/networkname"
	"github.com/ledgerwatch/erigon/cmd/devnet/devnet"
	"github.com/ledgerwatch/erigon/cmd/devnet/services"
	"github.com/ledgerwatch/erigon/cmd/devnet/services/polygon"
	"github.com/ledgerwatch/erigon/turbo/debug"
	"github.com/ledgerwatch/log/v3"
)

func initDevnet(chainName string, dataDir string, logger log.Logger) (devnet.Devnet, error) {
	const baseRpcHost = "localhost"
	const baseRpcPort = 8545

	switch chainName {
	case networkname.BorDevnetChainName:
		heimdallGrpcAddr := polygon.HeimdallGrpcAddressDefault
		const sprintSize uint64 = 0
		return NewBorDevnetWithLocalHeimdall(dataDir, baseRpcHost, baseRpcPort, heimdallGrpcAddr, sprintSize, logger), nil

	case networkname.DevChainName:
		return NewDevDevnet(dataDir, baseRpcHost, baseRpcPort, logger), nil

	case "":
		envChainName, _ := os.LookupEnv("DEVNET_CHAIN")
		if envChainName == "" {
			envChainName = networkname.DevChainName
		}
		return initDevnet(envChainName, dataDir, logger)

	default:
		return nil, fmt.Errorf("unknown network: '%s'", chainName)
	}
}

func ContextStart(t *testing.T, chainName string) (devnet.Context, error) {
	if runtime.GOOS == "windows" {
		t.Skip("FIXME: TempDir RemoveAll cleanup error: remove dev-0\\clique\\db\\clique\\mdbx.dat: The process cannot access the file because it is being used by another process")
	}

	debug.RaiseFdLimit()
	logger := log.New()
	dataDir := t.TempDir()

	var network devnet.Devnet
	network, err := initDevnet(chainName, dataDir, logger)
	if err != nil {
		return nil, fmt.Errorf("ContextStart initDevnet failed: %w", err)
	}

	runCtx, err := network.Start(logger)
	if err != nil {
		return nil, fmt.Errorf("ContextStart devnet start failed: %w", err)
	}

	t.Cleanup(services.UnsubscribeAll)
	t.Cleanup(network.Stop)

	return runCtx, nil
}