mirror of
https://github.com/status-im/nim-dagger.git
synced 2025-01-21 20:19:43 +00:00
3e80de3454
* Simplify `.then` (promise api) and tests * Remove tracked future when cancelled. Add tracked future tests * Track and cancel statemachine futures The futures created in each asyncstatemachine instance are tracked, and each future is cancelled and waited in `stop`. Change `asyncstatemachine.stop` to be async so `machine.trackedFutures.cancelAndWait` could be called. Add a constructor for `asyncstatemachine` that initialises the `trackedFutures` instance, and call the constructor from derived class constructors.
68 lines
1.9 KiB
Nim
68 lines
1.9 KiB
Nim
import pkg/asynctest
|
|
import pkg/chronos
|
|
import codex/utils/trackedfutures
|
|
import ../helpers/eventually
|
|
import ../helpers
|
|
|
|
type Module = object
|
|
trackedFutures: TrackedFutures
|
|
|
|
asyncchecksuite "tracked futures":
|
|
var module: Module
|
|
|
|
setup:
|
|
module = Module(trackedFutures: TrackedFutures.new())
|
|
|
|
test "starts with zero tracked futures":
|
|
check module.trackedFutures.len == 0
|
|
|
|
test "tracks unfinished futures":
|
|
let fut = newFuture[void]("test")
|
|
discard fut.track(module)
|
|
check module.trackedFutures.len == 1
|
|
|
|
test "does not track completed futures":
|
|
let fut = newFuture[void]("test")
|
|
fut.complete()
|
|
discard fut.track(module)
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
test "does not track failed futures":
|
|
let fut = newFuture[void]("test")
|
|
fut.fail((ref CatchableError)(msg: "some error"))
|
|
discard fut.track(module)
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
test "does not track cancelled futures":
|
|
let fut = newFuture[void]("test")
|
|
await fut.cancelAndWait()
|
|
discard fut.track(module)
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
test "removes tracked future when finished":
|
|
let fut = newFuture[void]("test")
|
|
discard fut.track(module)
|
|
fut.complete()
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
test "removes tracked future when cancelled":
|
|
let fut = newFuture[void]("test")
|
|
discard fut.track(module)
|
|
await fut.cancelAndWait()
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
test "cancels and removes all tracked futures":
|
|
let fut1 = newFuture[void]("test1")
|
|
let fut2 = newFuture[void]("test2")
|
|
let fut3 = newFuture[void]("test3")
|
|
discard fut1.track(module)
|
|
discard fut2.track(module)
|
|
discard fut3.track(module)
|
|
await module.trackedFutures.cancelTracked()
|
|
check eventually fut1.cancelled
|
|
check eventually fut2.cancelled
|
|
check eventually fut3.cancelled
|
|
check eventually module.trackedFutures.len == 0
|
|
|
|
|