100 lines
2.6 KiB
Nim
100 lines
2.6 KiB
Nim
## nim-websock
|
|
## Copyright (c) 2021 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import
|
|
std/[strutils],
|
|
pkg/[chronos, chronicles, stew/byteutils],
|
|
../websock/[websock, types, frame, extensions/compression/deflate]
|
|
|
|
const
|
|
clientFlags = {NoVerifyHost, NoVerifyServerName}
|
|
|
|
# we want to run parallel tests in CI,
|
|
# so we are using different port
|
|
when defined tls:
|
|
const
|
|
agent = "websock-secure-client"
|
|
secure = true
|
|
serverPort = 9002
|
|
else:
|
|
const
|
|
agent = "websock-client"
|
|
secure = false
|
|
serverPort = 9001
|
|
|
|
proc connectServer(path: string, factories: seq[ExtFactory] = @[]): Future[WSSession] {.async.} =
|
|
let ws = await WebSocket.connect(
|
|
host = "127.0.0.1:$1" % [$serverPort],
|
|
path = path,
|
|
secure = secure,
|
|
flags = clientFlags,
|
|
factories = factories
|
|
)
|
|
return ws
|
|
|
|
proc getCaseCount(): Future[int] {.async.} =
|
|
var caseCount = 0
|
|
block:
|
|
try:
|
|
let ws = await connectServer("/getCaseCount")
|
|
let buff = await ws.recvMsg()
|
|
let dataStr = string.fromBytes(buff)
|
|
caseCount = parseInt(dataStr)
|
|
await ws.close()
|
|
break
|
|
except WebSocketError as exc:
|
|
error "WebSocket error", exception = exc.msg
|
|
except ValueError as exc:
|
|
error "ParseInt error", exception = exc.msg
|
|
|
|
return caseCount
|
|
|
|
proc generateReport() {.async.} =
|
|
try:
|
|
trace "request autobahn server to generate report"
|
|
let ws = await connectServer("/updateReports?agent=" & agent)
|
|
while true:
|
|
let buff = await ws.recvMsg()
|
|
if buff.len <= 0:
|
|
break
|
|
await ws.close()
|
|
except WebSocketError as exc:
|
|
error "WebSocket error", exception = exc.msg
|
|
|
|
proc main() {.async.} =
|
|
let caseCount = await getCaseCount()
|
|
trace "case count", count=caseCount
|
|
|
|
var deflateFactory = @[deflateFactory()]
|
|
for i in 1..caseCount:
|
|
trace "runcase", no=i
|
|
let path = "/runCase?case=$1&agent=$2" % [$i, agent]
|
|
try:
|
|
let ws = await connectServer(path, deflateFactory)
|
|
|
|
while ws.readyState != ReadyState.Closed:
|
|
# echo back
|
|
let data = await ws.recvMsg()
|
|
let opCode = if ws.binary:
|
|
Opcode.Binary
|
|
else:
|
|
Opcode.Text
|
|
|
|
if ws.readyState == ReadyState.Closed:
|
|
break
|
|
|
|
await ws.send(data, opCode)
|
|
|
|
except WebSocketError as exc:
|
|
error "WebSocket error", exception = exc.msg
|
|
|
|
await generateReport()
|
|
|
|
waitFor main()
|