summary refs log tree commit diff stats
path: root/tests/async/tfuturevar.nim
blob: 73c0fddf75dc8896fd662552ea0e76e3ff51bfe1 (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
import asyncdispatch

proc completeOnReturn(fut: FutureVar[string], x: bool) {.async.} =
  if x:
    fut.mget() = ""
    fut.mget.add("foobar")
    return

proc completeOnImplicitReturn(fut: FutureVar[string], x: bool) {.async.} =
  if x:
    fut.mget() = ""
    fut.mget.add("foobar")

proc failureTest(fut: FutureVar[string], x: bool) {.async.} =
  if x:
    raise newException(Exception, "Test")

proc manualComplete(fut: FutureVar[string], x: bool) {.async.} =
  if x:
    fut.mget() = "Hello World"
    fut.complete()
    return

proc main() {.async.} =
  var fut: FutureVar[string]

  fut = newFutureVar[string]()
  await completeOnReturn(fut, true)
  doAssert(fut.read() == "foobar")

  fut = newFutureVar[string]()
  await completeOnImplicitReturn(fut, true)
  doAssert(fut.read() == "foobar")

  fut = newFutureVar[string]()
  let retFut = failureTest(fut, true)
  yield retFut
  doAssert(fut.read().isNil)
  doAssert(fut.finished)

  fut = newFutureVar[string]()
  await manualComplete(fut, true)
  doAssert(fut.read() == "Hello World")


waitFor main()