summary refs log tree commit diff stats
path: root/tests/async/tasyncall.nim
blob: 60ba557ccdc8f0d4921d1096c39aa5d082be5a6b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
discard """
  file: "tasyncall.nim"
  exitcode: 0
"""
import times, sequtils
import asyncdispatch

const
  taskCount = 10
  sleepDuration = 500

proc futureWithValue(x: int): Future[int] {.async.} =
  await sleepAsync(sleepDuration)
  return x

proc futureWithoutValue() {.async.} =
  await sleepAsync(1000)

proc testFuturesWithValue(x: int): seq[int] =
  var tasks = newSeq[Future[int]](taskCount)

  for i in 0..<taskCount:
    tasks[i] = futureWithValue(x)

  result = waitFor all(tasks)

proc testFuturesWithoutValues() =
  var tasks = newSeq[Future[void]](taskCount)

  for i in 0..<taskCount:
    tasks[i] = futureWithoutValue()

  waitFor all(tasks)

proc testVarargs(x, y, z: int): seq[int] =
  let
    a = futureWithValue(x)
    b = futureWithValue(y)
    c = futureWithValue(z)

  result = waitFor all(a, b, c)

block:
  let
    startTime = cpuTime()
    results = testFuturesWithValue(42)
    expected = repeat(42, taskCount)
    execTime = cpuTime() - startTime

  doAssert execTime * 1000 < taskCount * sleepDuration
  doAssert results == expected

block:
  let startTime = cpuTime()
  testFuturesWithoutValues()
  let execTime = cpuTime() - startTime

  doAssert execTime * 1000 < taskCount * sleepDuration

block:
  let
    startTime = cpuTime()
    results = testVarargs(1, 2, 3)
    expected = @[1, 2, 3]
    execTime = cpuTime() - startTime

  doAssert execTime * 100 < taskCount * sleepDuration
  doAssert results == expected