43 lines
1.0 KiB
Nim
43 lines
1.0 KiB
Nim
import std/times
|
|
import pkg/chronos
|
|
import codex/clock
|
|
|
|
export clock
|
|
|
|
type
|
|
MockClock* = ref object of Clock
|
|
time: SecondsSince1970
|
|
waiting: seq[Waiting]
|
|
Waiting = ref object
|
|
until: SecondsSince1970
|
|
future: Future[void]
|
|
|
|
func new*(
|
|
_: type MockClock,
|
|
time: SecondsSince1970 = getTime().toUnix
|
|
): MockClock =
|
|
## Create a mock clock instance
|
|
MockClock(time: time)
|
|
|
|
proc set*(clock: MockClock, time: SecondsSince1970) =
|
|
clock.time = time
|
|
var index = 0
|
|
while index < clock.waiting.len:
|
|
if clock.waiting[index].until <= clock.time:
|
|
clock.waiting[index].future.complete()
|
|
clock.waiting.del(index)
|
|
else:
|
|
inc index
|
|
|
|
proc advance*(clock: MockClock, seconds: int64) =
|
|
clock.set(clock.time + seconds)
|
|
|
|
method now*(clock: MockClock): Future[SecondsSince1970] {.async.} =
|
|
clock.time
|
|
|
|
method waitUntil*(clock: MockClock, time: SecondsSince1970) {.async.} =
|
|
if time > (await clock.now()):
|
|
let future = newFuture[void]()
|
|
clock.waiting.add(Waiting(until: time, future: future))
|
|
await future
|