mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-28 07:36:44 +00:00
64 lines
1.3 KiB
Go
64 lines
1.3 KiB
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().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().Number.Int64()
|
|
expected := blockNumber + i + 1
|
|
|
|
if res := assert.Equal(t, currBlock, expected); res == false {
|
|
break
|
|
}
|
|
}
|
|
}
|