98 lines
3.0 KiB
Nim
98 lines
3.0 KiB
Nim
# Asyncdispatch2 Test Suite
|
|
# (c) Copyright 2018
|
|
# Status Research & Development GmbH
|
|
#
|
|
# Licensed under either of
|
|
# Apache License, version 2.0, (LICENSE-APACHEv2)
|
|
# MIT license (LICENSE-MIT)
|
|
|
|
import strutils, unittest
|
|
import ../asyncdispatch2
|
|
|
|
type
|
|
CustomServer = ref object of StreamServer
|
|
test1: string
|
|
test2: string
|
|
|
|
CustomTransport = ref object of StreamTransport
|
|
test: string
|
|
|
|
proc serveStreamClient(server: StreamServer,
|
|
transp: StreamTransport) {.async.} =
|
|
discard
|
|
|
|
proc serveCustomStreamClient(server: StreamServer,
|
|
transp: StreamTransport) {.async.} =
|
|
var cserver = cast[CustomServer](server)
|
|
var ctransp = cast[CustomTransport](transp)
|
|
cserver.test1 = "CONNECTION"
|
|
cserver.test2 = ctransp.test
|
|
transp.close()
|
|
server.stop()
|
|
server.close()
|
|
|
|
proc customServerTransport(server: StreamServer,
|
|
fd: AsyncFD): StreamTransport =
|
|
var transp = CustomTransport()
|
|
transp.test = "CUSTOM"
|
|
result = cast[StreamTransport](transp)
|
|
|
|
proc serveDatagramClient(transp: DatagramTransport,
|
|
pbytes: pointer, nbytes: int,
|
|
raddr: TransportAddress,
|
|
udata: pointer): Future[void] {.async.} =
|
|
discard
|
|
|
|
proc test1(): bool =
|
|
var ta = initTAddress("127.0.0.1:31354")
|
|
var server1 = createStreamServer(ta, serveStreamClient, {ReuseAddr})
|
|
server1.start()
|
|
server1.stop()
|
|
server1.close()
|
|
var server2 = createStreamServer(ta, serveStreamClient, {ReuseAddr})
|
|
server2.start()
|
|
server2.stop()
|
|
server2.close()
|
|
result = true
|
|
|
|
proc test2(): bool =
|
|
var ta = initTAddress("127.0.0.1:31354")
|
|
var server1 = createDatagramServer(ta, serveDatagramClient, {ReuseAddr})
|
|
server1.start()
|
|
server1.stop()
|
|
server1.close()
|
|
var server2 = createDatagramServer(ta, serveDatagramClient, {ReuseAddr})
|
|
server2.start()
|
|
server2.stop()
|
|
server2.close()
|
|
result = true
|
|
|
|
proc client(server: CustomServer, ta: TransportAddress) {.async.} =
|
|
var transp = CustomTransport()
|
|
transp.test = "CLIENT"
|
|
server.start()
|
|
var ptransp = await connect(ta, child = transp)
|
|
var etransp = cast[CustomTransport](ptransp)
|
|
doAssert(etransp.test == "CLIENT")
|
|
transp.close()
|
|
await server.join()
|
|
|
|
proc test3(): bool =
|
|
var server = CustomServer()
|
|
server.test1 = "TEST"
|
|
var ta = initTAddress("127.0.0.1:31354")
|
|
var pserver = createStreamServer(ta, serveCustomStreamClient, {ReuseAddr},
|
|
child = cast[StreamServer](server),
|
|
init = customServerTransport)
|
|
waitFor client(server, ta)
|
|
result = (server.test1 == "CONNECTION") and (server.test2 == "CUSTOM")
|
|
|
|
when isMainModule:
|
|
suite "Server's test suite":
|
|
test "Stream Server start/stop test":
|
|
check test1() == true
|
|
test "Stream Server inherited object test":
|
|
check test3() == true
|
|
test "Datagram Server start/stop test":
|
|
check test2() == true
|