53 lines
1.5 KiB
Nim
53 lines
1.5 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, ../asyncdispatch2/timer
|
||
|
|
|
||
|
|
const TimersCount = 10
|
||
|
|
|
||
|
|
proc timeWorker(time: int): Future[int] {.async.} =
|
||
|
|
var st = fastEpochTime()
|
||
|
|
await sleepAsync(time)
|
||
|
|
var et = fastEpochTime()
|
||
|
|
result = int(et - st)
|
||
|
|
|
||
|
|
proc waitAll[T](futs: seq[Future[T]]): 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 test(timeout: int): Future[int] {.async.} =
|
||
|
|
var workers = newSeq[Future[int]](TimersCount)
|
||
|
|
for i in 0..<TimersCount:
|
||
|
|
workers[i] = timeWorker(timeout)
|
||
|
|
await waitAll(workers)
|
||
|
|
var sum = 0
|
||
|
|
for i in 0..<TimersCount:
|
||
|
|
var time = workers[i].read()
|
||
|
|
sum = sum + time
|
||
|
|
result = sum div 10
|
||
|
|
|
||
|
|
when isMainModule:
|
||
|
|
suite "Asynchronous timers test suite":
|
||
|
|
test $TimersCount & " timers with 10ms timeout":
|
||
|
|
var res = waitFor(test(10))
|
||
|
|
check (res >= 10) and (res <= 100)
|
||
|
|
test $TimersCount & " timers with 100ms timeout":
|
||
|
|
var res = waitFor(test(100))
|
||
|
|
check (res >= 100) and (res <= 1000)
|
||
|
|
test $TimersCount & " timers with 1000ms timeout":
|
||
|
|
var res = waitFor(test(1000))
|
||
|
|
check (res >= 1000) and (res <= 2000)
|