nim-codex/tests/integration/nodes.nim

49 lines
1.1 KiB
Nim

import std/osproc
import std/os
import std/streams
import std/strutils
const workingDir = currentSourcePath() / ".." / ".." / ".."
const executable = "build" / "codex"
type NodeProcess* = ref object
process: Process
arguments: seq[string]
debug: bool
proc start(node: NodeProcess) =
if node.debug:
node.process = startProcess(
executable,
workingDir,
node.arguments,
options={poParentStreams}
)
sleep(1000)
else:
node.process = startProcess(
executable,
workingDir,
node.arguments
)
for line in node.process.outputStream.lines:
if line.contains("Started codex node"):
break
proc startNode*(args: openArray[string], debug = false): NodeProcess =
## Starts a Codex Node with the specified arguments.
## Set debug to 'true' to see output of the node.
let node = NodeProcess(arguments: @args, debug: debug)
node.start()
node
proc stop*(node: NodeProcess) =
let process = node.process
process.terminate()
discard process.waitForExit(timeout=5_000)
process.close()
proc restart*(node: NodeProcess) =
node.stop()
node.start()