89 lines
2.4 KiB
Nim
89 lines
2.4 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 pkg/[chronicles, chronos/unittest2/asynctests, stew/byteutils]
|
|
import ./base64ext, ./hexext
|
|
import ../../websock/websock, ../helpers
|
|
|
|
suite "multiple extensions flow":
|
|
setup:
|
|
var server: HttpServer
|
|
let address = initTAddress("127.0.0.1:8888")
|
|
let hexFactory = hexFactory()
|
|
let base64Factory = base64Factory(padding = true)
|
|
|
|
teardown:
|
|
if server != nil:
|
|
server.stop()
|
|
waitFor server.closeWait()
|
|
|
|
asyncTest "hex to base64 ext flow":
|
|
let testData = "hello world"
|
|
proc handle(request: HttpRequest) {.async.} =
|
|
let server = WSServer.new(
|
|
protos = ["proto"],
|
|
factories = [hexFactory, base64Factory],
|
|
)
|
|
let ws = await server.handleRequest(request)
|
|
let recvData = await ws.recvMsg()
|
|
await ws.send(recvData,
|
|
if ws.binary: Opcode.Binary else: Opcode.Text)
|
|
|
|
await waitForClose(ws)
|
|
|
|
server = HttpServer.create(
|
|
address,
|
|
handle,
|
|
flags = {ReuseAddr})
|
|
server.start()
|
|
|
|
let client = await WebSocket.connect(
|
|
host = "127.0.0.1:8888",
|
|
path = "/ws",
|
|
protocols = @["proto"],
|
|
factories = @[hexFactory, base64Factory]
|
|
)
|
|
|
|
await client.send(testData)
|
|
let res = await client.recvMsg()
|
|
check testData.toBytes() == res
|
|
await client.close()
|
|
|
|
asyncTest "base64 to hex ext flow":
|
|
let testData = "hello world"
|
|
proc handle(request: HttpRequest) {.async.} =
|
|
let server = WSServer.new(
|
|
protos = ["proto"],
|
|
factories = [hexFactory, base64Factory],
|
|
)
|
|
let ws = await server.handleRequest(request)
|
|
let recvData = await ws.recvMsg()
|
|
await ws.send(recvData,
|
|
if ws.binary: Opcode.Binary else: Opcode.Text)
|
|
|
|
await waitForClose(ws)
|
|
|
|
server = HttpServer.create(
|
|
address,
|
|
handle,
|
|
flags = {ReuseAddr})
|
|
server.start()
|
|
|
|
let client = await WebSocket.connect(
|
|
host = "127.0.0.1:8888",
|
|
path = "/ws",
|
|
protocols = @["proto"],
|
|
factories = @[base64Factory, hexFactory]
|
|
)
|
|
|
|
await client.send(testData)
|
|
let res = await client.recvMsg()
|
|
check testData.toBytes() == res
|
|
await client.close()
|