107 lines
2.8 KiB
Nim
107 lines
2.8 KiB
Nim
# Chronos Test Suite
|
|
# (c) Copyright 2018-Present
|
|
# Status Research & Development GmbH
|
|
#
|
|
# Licensed under either of
|
|
# Apache License, version 2.0, (LICENSE-APACHEv2)
|
|
# MIT license (LICENSE-MIT)
|
|
import unittest2
|
|
import macros
|
|
import ../chronos
|
|
|
|
when defined(nimHasUsed): {.used.}
|
|
|
|
proc asyncRetValue(n: int): Future[int] {.async.} =
|
|
await sleepAsync(n.milliseconds)
|
|
result = n * 10
|
|
|
|
proc asyncRetVoid(n: int) {.async.} =
|
|
await sleepAsync(n.milliseconds)
|
|
|
|
proc asyncRetExceptionValue(n: int): Future[int] {.async.} =
|
|
await sleepAsync(n.milliseconds)
|
|
result = n * 10
|
|
if true:
|
|
raise newException(ValueError, "Test exception")
|
|
|
|
proc asyncRetExceptionVoid(n: int) {.async.} =
|
|
await sleepAsync(n.milliseconds)
|
|
if true:
|
|
raise newException(ValueError, "Test exception")
|
|
|
|
proc testAwait(): Future[bool] {.async.} =
|
|
var res: int
|
|
await asyncRetVoid(100)
|
|
res = await asyncRetValue(100)
|
|
if res != 1000:
|
|
return false
|
|
if (await asyncRetValue(100)) != 1000:
|
|
return false
|
|
try:
|
|
await asyncRetExceptionVoid(100)
|
|
return false
|
|
except ValueError:
|
|
discard
|
|
res = 0
|
|
try:
|
|
discard await asyncRetExceptionValue(100)
|
|
return false
|
|
except ValueError:
|
|
discard
|
|
if res != 0:
|
|
return false
|
|
return true
|
|
|
|
proc testAwaitne(): Future[bool] {.async.} =
|
|
var res1: Future[void]
|
|
var res2: Future[int]
|
|
|
|
res1 = awaitne asyncRetVoid(100)
|
|
res2 = awaitne asyncRetValue(100)
|
|
if res1.failed():
|
|
return false
|
|
if res2.read() != 1000:
|
|
return false
|
|
|
|
res1 = awaitne asyncRetExceptionVoid(100)
|
|
if not(res1.failed()):
|
|
return false
|
|
|
|
res2 = awaitne asyncRetExceptionValue(100)
|
|
try:
|
|
discard res2.read()
|
|
return false
|
|
except ValueError:
|
|
discard
|
|
|
|
return true
|
|
|
|
suite "Macro transformations test suite":
|
|
test "`await` command test":
|
|
check waitFor(testAwait()) == true
|
|
test "`awaitne` command test":
|
|
check waitFor(testAwaitne()) == true
|
|
|
|
|
|
test "template async macro transformation":
|
|
template templatedAsync(name, restype: untyped): untyped =
|
|
proc name(): Future[restype] {.async.} = return @[4]
|
|
|
|
templatedAsync(testTemplate, seq[int])
|
|
check waitFor(testTemplate()) == @[4]
|
|
|
|
macro macroAsync(name, restype, innerrestype: untyped): untyped =
|
|
quote do:
|
|
proc `name`(): Future[`restype`[`innerrestype`]] {.async.} = return
|
|
|
|
type OpenObject = object
|
|
macroAsync(testMacro, seq, OpenObject)
|
|
check waitFor(testMacro()).len == 0
|
|
|
|
macro macroAsync2(name, restype, inner1, inner2, inner3, inner4: untyped): untyped =
|
|
quote do:
|
|
proc `name`(): Future[`restype`[`inner1`[`inner2`[`inner3`, `inner4`]]]] {.async.} = return
|
|
|
|
macroAsync2(testMacro2, seq, Opt, Result, OpenObject, cstring)
|
|
check waitFor(testMacro2()).len == 0
|