48 lines
1.3 KiB
Nim
48 lines
1.3 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 unittest
|
|
import ../asyncdispatch2
|
|
|
|
const HELLO_PORT = 45679
|
|
const TEST_MSG = "testmsg"
|
|
const MSG_LEN = TEST_MSG.len()
|
|
|
|
type
|
|
CustomData = ref object
|
|
test: string
|
|
|
|
proc udp4DataAvailable(transp: DatagramTransport,
|
|
remote: TransportAddress): Future[void] {.async, gcsafe.} =
|
|
var udata = getUserData[CustomData](transp)
|
|
var expect = TEST_MSG
|
|
var data: seq[byte]
|
|
var datalen: int
|
|
transp.peekMessage(data, datalen)
|
|
if udata.test == "CHECK" and datalen == MSG_LEN and
|
|
equalMem(addr data[0], addr expect[0], datalen):
|
|
udata.test = "OK"
|
|
transp.close()
|
|
|
|
proc issue6(): Future[bool] {.async.} =
|
|
var myself = initTAddress("127.0.0.1:" & $HELLO_PORT)
|
|
var data = CustomData()
|
|
data.test = "CHECK"
|
|
var dsock4 = newDatagramTransport(udp4DataAvailable, udata = data,
|
|
local = myself)
|
|
await dsock4.sendTo(myself, TEST_MSG, MSG_LEN)
|
|
await dsock4.join()
|
|
if data.test == "OK":
|
|
result = true
|
|
|
|
when isMainModule:
|
|
suite "Asynchronous issues test suite":
|
|
test "Issue #6":
|
|
var res = waitFor(issue6())
|
|
check res == true
|