go-ethereum/internal/cli/server/server_test.go

package server

import (
	"fmt"
	"net"
	"sync/atomic"
	"testing"
	"time"

	"github.com/ethereum/go-ethereum/log"

	"github.com/stretchr/testify/assert"
)

var initialPort uint64 = 61000

// nextPort gives the next available port starting from 60000
func nextPort() uint64 {
	log.Info("Checking for new port", "current", initialPort)
	port := atomic.AddUint64(&initialPort, 1)
	addr := fmt.Sprintf("localhost:%d", port)

	lis, err := net.Listen("tcp", addr)
	if err == nil {
		lis.Close()

		return port
	} else {
		return nextPort()
	}
}

func TestServer_DeveloperMode(t *testing.T) {
	t.Parallel()

	// get the default config
	config := DefaultConfig()

	// enable developer mode
	config.Developer.Enabled = true
	config.Developer.Period = 2 // block time

	// start the mock server
	server, err := CreateMockServer(config)
	assert.NoError(t, err)

	defer CloseMockServer(server)

	// record the initial block number
	blockNumber := server.backend.BlockChain().CurrentBlock().Header().Number.Int64()

	var i int64 = 0
	for i = 0; i < 3; i++ {
		// We expect the node to mine blocks every `config.Developer.Period` time period
		time.Sleep(time.Duration(config.Developer.Period) * time.Second)
		currBlock := server.backend.BlockChain().CurrentBlock().Header().Number.Int64()
		expected := blockNumber + i + 1
		if res := assert.Equal(t, currBlock, expected); res == false {
			break
		}
	}
}