83 lines
2.0 KiB
Nim
83 lines
2.0 KiB
Nim
# Chronos Test Suite
|
|
# (c) Copyright 2020-Present
|
|
# Status Research & Development GmbH
|
|
#
|
|
# Licensed under either of
|
|
# Apache License, version 2.0, (LICENSE-APACHEv2)
|
|
# MIT license (LICENSE-MIT)
|
|
import unittest2
|
|
import ../chronos, ../chronos/config
|
|
|
|
{.used.}
|
|
|
|
suite "Asynchronous utilities test suite":
|
|
when chronosFutureTracking:
|
|
proc getCount(): uint =
|
|
# This procedure counts number of Future[T] in double-linked list via list
|
|
# iteration.
|
|
var res = 0'u
|
|
for item in pendingFutures():
|
|
inc(res)
|
|
res
|
|
|
|
test "Future clean and leaks test":
|
|
when chronosFutureTracking:
|
|
if pendingFuturesCount(WithoutCompleted) == 0'u:
|
|
if pendingFuturesCount(OnlyCompleted) > 0'u:
|
|
poll()
|
|
check pendingFuturesCount() == 0'u
|
|
else:
|
|
echo dumpPendingFutures()
|
|
check false
|
|
else:
|
|
skip()
|
|
|
|
test "FutureList basics test":
|
|
when chronosFutureTracking:
|
|
var fut1 = newFuture[void]()
|
|
check:
|
|
getCount() == 1'u
|
|
pendingFuturesCount() == 1'u
|
|
var fut2 = newFuture[void]()
|
|
check:
|
|
getCount() == 2'u
|
|
pendingFuturesCount() == 2'u
|
|
var fut3 = newFuture[void]()
|
|
check:
|
|
getCount() == 3'u
|
|
pendingFuturesCount() == 3'u
|
|
fut1.complete()
|
|
poll()
|
|
check:
|
|
getCount() == 2'u
|
|
pendingFuturesCount() == 2'u
|
|
fut2.fail(newException(ValueError, ""))
|
|
poll()
|
|
check:
|
|
getCount() == 1'u
|
|
pendingFuturesCount() == 1'u
|
|
discard fut3.tryCancel()
|
|
poll()
|
|
check:
|
|
getCount() == 0'u
|
|
pendingFuturesCount() == 0'u
|
|
else:
|
|
skip()
|
|
|
|
test "FutureList async procedure test":
|
|
when chronosFutureTracking:
|
|
proc simpleProc() {.async.} =
|
|
await sleepAsync(10.milliseconds)
|
|
|
|
var fut = simpleProc()
|
|
check:
|
|
getCount() == 2'u
|
|
pendingFuturesCount() == 2'u
|
|
|
|
waitFor fut
|
|
check:
|
|
getCount() == 0'u
|
|
pendingFuturesCount() == 0'u
|
|
else:
|
|
skip()
|