# 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, net, unittest import ../asyncdispatch2 when sizeof(int) == 8: const TestsCount = 10000 ClientsCount = 100 MessagesCount = 100 elif sizeof(int) == 4: const TestsCount = 2000 ClientsCount = 20 MessagesCount = 20 proc client1(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("REQUEST"): var numstr = data[7..^1] var num = parseInt(numstr) var ans = "ANSWER" & $num await transp.sendTo(raddr, addr ans[0], len(ans)) else: var err = "ERROR" await transp.sendTo(raddr, addr err[0], len(err)) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client2(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var ta = initTAddress("127.0.0.1:33336") var req = "REQUEST" & $counterPtr[] await transp.sendTo(ta, addr req[0], len(req)) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client3(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var req = "REQUEST" & $counterPtr[] await transp.send(addr req[0], len(req)) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client4(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == MessagesCount: transp.close() else: var req = "REQUEST" & $counterPtr[] await transp.send(addr req[0], len(req)) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client5(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == MessagesCount: transp.close() else: var ta = initTAddress("127.0.0.1:33337") var req = "REQUEST" & $counterPtr[] await transp.sendTo(ta, addr req[0], len(req)) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client6(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("REQUEST"): var numstr = data[7..^1] var num = parseInt(numstr) var ans = "ANSWER" & $num await transp.sendTo(raddr, ans) else: var err = "ERROR" await transp.sendTo(raddr, err) else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client7(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var ta = initTAddress("127.0.0.1:33336") var req = "REQUEST" & $counterPtr[] await transp.sendTo(ta, req) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client8(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var req = "REQUEST" & $counterPtr[] await transp.send(req) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client9(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("REQUEST"): var numstr = data[7..^1] var num = parseInt(numstr) var ans = "ANSWER" & $num var ansseq = newSeq[byte](len(ans)) copyMem(addr ansseq[0], addr ans[0], len(ans)) await transp.sendTo(raddr, ansseq) else: var err = "ERROR" var errseq = newSeq[byte](len(err)) copyMem(addr errseq[0], addr err[0], len(err)) await transp.sendTo(raddr, errseq) else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client10(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var ta = initTAddress("127.0.0.1:33336") var req = "REQUEST" & $counterPtr[] var reqseq = newSeq[byte](len(req)) copyMem(addr reqseq[0], addr req[0], len(req)) await transp.sendTo(ta, reqseq) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc client11(transp: DatagramTransport, raddr: TransportAddress): Future[void] {.async.} = var pbytes: seq[byte] var nbytes: int transp.peekMessage(pbytes, nbytes) if nbytes > 0: var data = newString(nbytes + 1) copyMem(addr data[0], addr pbytes[0], nbytes) data.setLen(nbytes) if data.startsWith("ANSWER"): var counterPtr = cast[ptr int](transp.udata) counterPtr[] = counterPtr[] + 1 if counterPtr[] == TestsCount: transp.close() else: var req = "REQUEST" & $counterPtr[] var reqseq = newSeq[byte](len(req)) copyMem(addr reqseq[0], addr req[0], len(req)) await transp.send(reqseq) else: var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() else: ## Read operation failed with error var counterPtr = cast[ptr int](transp.udata) counterPtr[] = -1 transp.close() proc testPointerSendTo(): Future[int] {.async.} = ## sendTo(pointer) test var ta = initTAddress("127.0.0.1:33336") var counter = 0 var dgram1 = newDatagramTransport(client1, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client2, udata = addr counter) var data = "REQUEST0" await dgram2.sendTo(ta, addr data[0], len(data)) await dgram2.join() dgram1.close() dgram2.close() result = counter proc testPointerSend(): Future[int] {.async.} = ## send(pointer) test var ta = initTAddress("127.0.0.1:33337") var counter = 0 var dgram1 = newDatagramTransport(client1, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client3, udata = addr counter, remote = ta) var data = "REQUEST0" await dgram2.send(addr data[0], len(data)) await dgram2.join() dgram1.close() dgram2.close() result = counter proc testStringSendTo(): Future[int] {.async.} = ## sendTo(string) test var ta = initTAddress("127.0.0.1:33336") var counter = 0 var dgram1 = newDatagramTransport(client6, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client7, udata = addr counter) var data = "REQUEST0" await dgram2.sendTo(ta, data) await dgram2.join() dgram1.close() dgram2.close() result = counter proc testStringSend(): Future[int] {.async.} = ## send(string) test var ta = initTAddress("127.0.0.1:33337") var counter = 0 var dgram1 = newDatagramTransport(client6, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client8, udata = addr counter, remote = ta) var data = "REQUEST0" await dgram2.send(data) await dgram2.join() dgram1.close() dgram2.close() result = counter proc testSeqSendTo(): Future[int] {.async.} = ## sendTo(string) test var ta = initTAddress("127.0.0.1:33336") var counter = 0 var dgram1 = newDatagramTransport(client9, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client10, udata = addr counter) var data = "REQUEST0" var dataseq = newSeq[byte](len(data)) copyMem(addr dataseq[0], addr data[0], len(data)) await dgram2.sendTo(ta, dataseq) await dgram2.join() dgram1.close() dgram2.close() result = counter proc testSeqSend(): Future[int] {.async.} = ## send(string) test var ta = initTAddress("127.0.0.1:33337") var counter = 0 var dgram1 = newDatagramTransport(client9, udata = addr counter, local = ta) var dgram2 = newDatagramTransport(client11, udata = addr counter, remote = ta) var data = "REQUEST0" var dataseq = newSeq[byte](len(data)) copyMem(addr dataseq[0], addr data[0], len(data)) await dgram2.send(data) await dgram2.join() dgram1.close() dgram2.close() result = counter # proc waitAll(futs: seq[Future[void]]): Future[void] = var counter = len(futs) var retFuture = newFuture[void]("waitAll") proc cb(udata: pointer) = dec(counter) if counter == 0: retFuture.complete() for fut in futs: fut.addCallback(cb) return retFuture proc test3(bounded: bool): Future[int] {.async.} = var ta = initTAddress("127.0.0.1:33337") var counter = 0 var dgram1 = newDatagramTransport(client1, udata = addr counter, local = ta) var clients = newSeq[Future[void]](ClientsCount) var grams = newSeq[DatagramTransport](ClientsCount) var counters = newSeq[int](ClientsCount) for i in 0..