## nim-websock
## Copyright (c) 2021-2022 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/[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()