72 lines
1.8 KiB
Nim
72 lines
1.8 KiB
Nim
|
# Nim-Libp2p
|
||
|
|
# Copyright (c) 2023 Status Research & Development GmbH
|
||
|
|
# Licensed under either of
|
||
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
||
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
||
|
|
# at your option.
|
||
|
|
# This file may not be copied, modified, or distributed except according to
|
||
|
|
# those terms.
|
||
|
|
|
||
|
|
import chronos
|
||
|
|
|
||
|
|
import ../libp2p/utils/future
|
||
|
|
import ./helpers
|
||
|
|
|
||
|
|
suite "Future":
|
||
|
|
asyncTest "anyCompleted must complete with first completed future":
|
||
|
|
proc fut1() {.async.} =
|
||
|
|
await sleepAsync(100.milliseconds)
|
||
|
|
|
||
|
|
proc fut2() {.async.} =
|
||
|
|
await sleepAsync(200.milliseconds)
|
||
|
|
|
||
|
|
proc fut3() {.async.} =
|
||
|
|
raise newException(CatchableError, "fut3")
|
||
|
|
|
||
|
|
var f1 = fut1()
|
||
|
|
var f2 = fut2()
|
||
|
|
var f3 = fut3()
|
||
|
|
var f = await anyCompleted(@[f1, f2, f3])
|
||
|
|
|
||
|
|
check f == f1
|
||
|
|
|
||
|
|
asyncTest "anyCompleted must fail with empty list":
|
||
|
|
expect AllFuturesFailedError:
|
||
|
|
discard await anyCompleted(newSeq[Future[void]]())
|
||
|
|
|
||
|
|
asyncTest "anyCompleted must fail if all futures fail":
|
||
|
|
proc fut1() {.async.} =
|
||
|
|
raise newException(CatchableError, "fut1")
|
||
|
|
|
||
|
|
proc fut2() {.async.} =
|
||
|
|
raise newException(CatchableError, "fut2")
|
||
|
|
|
||
|
|
proc fut3() {.async.} =
|
||
|
|
raise newException(CatchableError, "fut3")
|
||
|
|
|
||
|
|
var f1 = fut1()
|
||
|
|
var f2 = fut2()
|
||
|
|
var f3 = fut3()
|
||
|
|
|
||
|
|
expect AllFuturesFailedError:
|
||
|
|
discard await anyCompleted(@[f1, f2, f3])
|
||
|
|
|
||
|
|
asyncTest "anyCompleted with timeout":
|
||
|
|
proc fut1() {.async.} =
|
||
|
|
await sleepAsync(100.milliseconds)
|
||
|
|
|
||
|
|
proc fut2() {.async.} =
|
||
|
|
await sleepAsync(200.milliseconds)
|
||
|
|
|
||
|
|
proc fut3() {.async.} =
|
||
|
|
raise newException(ValueError, "fut3")
|
||
|
|
|
||
|
|
var f1 = fut1()
|
||
|
|
var f2 = fut2()
|
||
|
|
var f3 = fut3()
|
||
|
|
var f = anyCompleted(@[f1, f2, f3])
|
||
|
|
if not await f.withTimeout(50.milliseconds):
|
||
|
|
f.cancel()
|
||
|
|
|
||
|
|
check f.cancelled()
|