new tester; all tests categorized

tests/async/tasynciossl.nim

@@ -0,0 +1,91 @@
+discard """
+  file: "tasynciossl.nim"
+  cmd: "nimrod cc --hints:on --define:ssl $# $#"
+  output: "20000"
+"""
+import sockets, asyncio, strutils, times
+
+var disp = newDispatcher()
+var msgCount = 0
+
+when defined(ssl):
+  var ctx = newContext(verifyMode = CVerifyNone, 
+      certFile = "tests/testdata/mycert.pem", keyFile = "tests/testdata/mycert.pem")
+
+  var ctx1 = newContext(verifyMode = CVerifyNone)
+
+const 
+  swarmSize = 50
+  messagesToSend = 100
+
+proc swarmConnect(s: PAsyncSocket) =
+  #echo("Connected")
+  for i in 1..messagesToSend:
+    s.send("Message " & $i & "\c\L")
+  s.close()
+
+proc serverRead(s: PAsyncSocket) =
+  var line = ""
+  assert s.readLine(line)
+  if line != "":
+    #echo(line)
+    if line.startsWith("Message "):
+      msgCount.inc()
+    else:
+      assert(false)
+  else:
+    s.close()
+
+proc serverAccept(s: PAsyncSocket) =
+  var client: PAsyncSocket
+  new(client)
+  s.accept(client)
+  client.handleRead = serverRead
+  disp.register(client)
+
+proc launchSwarm(disp: var PDispatcher, port: TPort, count: int,
+                 buffered = true, useSSL = false) =
+  for i in 1..count:
+    var client = AsyncSocket()
+    when defined(ssl):
+      if useSSL:
+        ctx1.wrapSocket(client)
+    client.handleConnect = swarmConnect
+    disp.register(client)
+    client.connect("localhost", port)
+
+proc createSwarm(port: TPort, buffered = true, useSSL = false) =
+  var server = AsyncSocket()
+  when defined(ssl):
+    if useSSL:
+      ctx.wrapSocket(server)
+  server.handleAccept = serverAccept
+  disp.register(server)
+  server.bindAddr(port)
+  server.listen()
+  disp.launchSwarm(port, swarmSize, buffered, useSSL)
+
+when defined(ssl):
+  const serverCount = 4
+else:
+  const serverCount = 2
+
+createSwarm(TPort(10235))
+createSwarm(TPort(10236), false)
+
+when defined(ssl):
+  createSwarm(TPort(10237), true, true)
+  createSwarm(TPort(10238), false, true)
+
+var startTime = epochTime()
+while true:
+  if epochTime() - startTime >= 300.0:
+    break
+  if not disp.poll(): break
+  if disp.len == serverCount:
+    # Only the servers are left in the dispatcher. All clients finished,
+    # we need to therefore break.
+    break
+
+assert msgCount == (swarmSize * messagesToSend) * serverCount
+echo(msgCount)

tests/async/tasyncudp.nim

@@ -0,0 +1,78 @@
+discard """
+  file: "tasyncudp.nim"
+  output: "2000"
+"""
+import asyncio, sockets, strutils, times
+
+const
+  swarmSize = 5
+  messagesToSend = 200
+
+var
+  disp = newDispatcher()
+  msgCount = 0
+  currentClient = 0
+
+proc serverRead(s: PAsyncSocket) =
+  var data = ""
+  var address = ""
+  var port: TPort
+  if s.recvFromAsync(data, 9, address, port):
+    assert address == "127.0.0.1"
+    msgCount.inc()
+  
+  discard """
+  
+  var line = ""
+  assert s.recvLine(line)
+  
+  if line == "":
+    assert(false)
+  else:
+    if line.startsWith("Message "):
+      msgCount.inc()
+    else:
+      assert(false)
+  """
+
+proc swarmConnect(s: PAsyncSocket) =
+  for i in 1..messagesToSend:
+    s.send("Message\c\L")
+
+proc createClient(disp: var PDispatcher, port: TPort,
+                  buffered = true) =
+  currentClient.inc()
+  var client = AsyncSocket(typ = SOCK_DGRAM, protocol = IPPROTO_UDP,
+                           buffered = buffered)
+  client.handleConnect = swarmConnect
+  disp.register(client)
+  client.connect("localhost", port)
+
+proc createServer(port: TPort, buffered = true) =
+  var server = AsyncSocket(typ = SOCK_DGRAM, protocol = IPPROTO_UDP,
+                           buffered = buffered)
+  server.handleRead = serverRead
+  disp.register(server)
+  server.bindAddr(port)
+
+let serverCount = 2
+
+createServer(TPort(10335), false)
+createServer(TPort(10336), true)
+var startTime = epochTime()
+while true:
+  if epochTime() - startTime >= 300.0:
+    break
+
+  if not disp.poll():
+    break
+  
+  if (msgCount div messagesToSend) * serverCount == currentClient:
+    createClient(disp, TPort(10335), false)
+    createClient(disp, TPort(10336), true)
+  
+  if msgCount == messagesToSend * serverCount * swarmSize:
+    break
+
+assert msgCount == messagesToSend * serverCount * swarmSize
+echo(msgCount)