124 lines
4.1 KiB
Nim
124 lines
4.1 KiB
Nim
|
import std/os
|
||
|
import std/macros
|
||
|
import std/json
|
||
|
import std/httpclient
|
||
|
import pkg/chronicles
|
||
|
import ../ethertest
|
||
|
import ./codexclient
|
||
|
import ./nodes
|
||
|
|
||
|
export ethertest
|
||
|
export codexclient
|
||
|
export nodes
|
||
|
|
||
|
template multinodesuite*(name: string,
|
||
|
startNodes: StartNodes, debugNodes: DebugNodes, body: untyped) =
|
||
|
|
||
|
if (debugNodes.client or debugNodes.provider) and
|
||
|
(enabledLogLevel > LogLevel.TRACE or
|
||
|
enabledLogLevel == LogLevel.NONE):
|
||
|
echo ""
|
||
|
echo "More test debug logging is available by running the tests with " &
|
||
|
"'-d:chronicles_log_level=TRACE " &
|
||
|
"-d:chronicles_disabled_topics=websock " &
|
||
|
"-d:chronicles_default_output_device=stdout " &
|
||
|
"-d:chronicles_sinks=textlines'"
|
||
|
echo ""
|
||
|
|
||
|
ethersuite name:
|
||
|
|
||
|
var running: seq[RunningNode]
|
||
|
var bootstrap: string
|
||
|
|
||
|
proc newNodeProcess(index: int,
|
||
|
addlOptions: seq[string],
|
||
|
debug: bool): (NodeProcess, string, Address) =
|
||
|
|
||
|
if index > accounts.len - 1:
|
||
|
raiseAssert("Cannot start node at index " & $index &
|
||
|
", not enough eth accounts.")
|
||
|
|
||
|
let datadir = getTempDir() / "Codex" & $index
|
||
|
var options = @[
|
||
|
"--api-port=" & $(8080 + index),
|
||
|
"--data-dir=" & datadir,
|
||
|
"--nat=127.0.0.1",
|
||
|
"--disc-ip=127.0.0.1",
|
||
|
"--disc-port=" & $(8090 + index),
|
||
|
"--eth-account=" & $accounts[index]]
|
||
|
.concat(addlOptions)
|
||
|
if debug: options.add "--log-level=INFO;TRACE: " & debugNodes.topics
|
||
|
let node = startNode(options, debug = debug)
|
||
|
(node, datadir, accounts[index])
|
||
|
|
||
|
proc newCodexClient(index: int): CodexClient =
|
||
|
CodexClient.new("http://localhost:" & $(8080 + index) & "/api/codex/v1")
|
||
|
|
||
|
proc startClientNode() =
|
||
|
let index = running.len
|
||
|
let (node, datadir, account) = newNodeProcess(
|
||
|
index, @["--persistence"], debugNodes.client)
|
||
|
let restClient = newCodexClient(index)
|
||
|
running.add RunningNode.new(Role.Client, node, restClient, datadir,
|
||
|
account)
|
||
|
if debugNodes.client:
|
||
|
debug "started new client node and codex client",
|
||
|
restApiPort = 8080 + index, discPort = 8090 + index, account
|
||
|
|
||
|
proc startProviderNode(failEveryNProofs: uint = 0) =
|
||
|
let index = running.len
|
||
|
let (node, datadir, account) = newNodeProcess(index, @[
|
||
|
"--bootstrap-node=" & bootstrap,
|
||
|
"--persistence",
|
||
|
"--simulate-proof-failures=" & $failEveryNProofs],
|
||
|
debugNodes.provider)
|
||
|
let restClient = newCodexClient(index)
|
||
|
running.add RunningNode.new(Role.Provider, node, restClient, datadir,
|
||
|
account)
|
||
|
if debugNodes.provider:
|
||
|
debug "started new provider node and codex client",
|
||
|
restApiPort = 8080 + index, discPort = 8090 + index, account
|
||
|
|
||
|
proc startValidatorNode() =
|
||
|
let index = running.len
|
||
|
let (node, datadir, account) = newNodeProcess(index, @[
|
||
|
"--bootstrap-node=" & bootstrap,
|
||
|
"--validator"],
|
||
|
debugNodes.validator)
|
||
|
let restClient = newCodexClient(index)
|
||
|
running.add RunningNode.new(Role.Validator, node, restClient, datadir,
|
||
|
account)
|
||
|
if debugNodes.validator:
|
||
|
debug "started new validator node and codex client",
|
||
|
restApiPort = 8080 + index, discPort = 8090 + index, account
|
||
|
|
||
|
proc clients(): seq[RunningNode] =
|
||
|
running.filter(proc(r: RunningNode): bool = r.role == Role.Client)
|
||
|
|
||
|
proc providers(): seq[RunningNode] =
|
||
|
running.filter(proc(r: RunningNode): bool = r.role == Role.Provider)
|
||
|
|
||
|
proc validators(): seq[RunningNode] =
|
||
|
running.filter(proc(r: RunningNode): bool = r.role == Role.Validator)
|
||
|
|
||
|
setup:
|
||
|
for i in 0..<startNodes.clients:
|
||
|
startClientNode()
|
||
|
if i == 0:
|
||
|
bootstrap = running[0].restClient.info()["spr"].getStr()
|
||
|
|
||
|
for i in 0..<startNodes.providers:
|
||
|
startProviderNode()
|
||
|
|
||
|
for i in 0..<startNodes.validators:
|
||
|
startValidatorNode()
|
||
|
|
||
|
teardown:
|
||
|
for r in running:
|
||
|
r.restClient.close()
|
||
|
r.node.stop()
|
||
|
removeDir(r.datadir)
|
||
|
running = @[]
|
||
|
|
||
|
body
|