## Nim-Codex
## 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 pkg/chronos
import pkg/asynctest/chronos/unittest

import codex/utils/timer
import ../helpers

asyncchecksuite "Timer":
  var timer1: Timer
  var timer2: Timer
  var output: string
  var numbersState = 0
  var lettersState = 'a'

  proc numbersCallback(): Future[void] {.async.} =
    output &= $numbersState
    inc numbersState

  proc lettersCallback(): Future[void] {.async.} =
    output &= $lettersState
    inc lettersState

  proc exceptionCallback(): Future[void] {.async.} =
    raise newException(CatchableError, "Test Exception")

  proc startNumbersTimer() =
    timer1.start(numbersCallback, 10.milliseconds)

  proc startLettersTimer() =
    timer2.start(lettersCallback, 10.milliseconds)

  setup:
    timer1 = Timer.new()
    timer2 = Timer.new()

    output = ""
    numbersState = 0
    lettersState = 'a'

  teardown:
    await timer1.stop()
    await timer2.stop()

  test "Start timer1 should execute callback":
    startNumbersTimer()
    check eventually output == "0"

  test "Start timer1 should execute callback multiple times":
    startNumbersTimer()
    check eventually output == "012"

  test "Starting timer1 multiple times has no impact":
    startNumbersTimer()
    startNumbersTimer()
    startNumbersTimer()
    check eventually output == "01234"

  test "Stop timer1 should stop execution of the callback":
    startNumbersTimer()
    check eventually output == "012"
    await timer1.stop()
    await sleepAsync(30.milliseconds)
    let stoppedOutput = output
    await sleepAsync(30.milliseconds)
    check output == stoppedOutput

  test "Exceptions raised in timer callback are handled":
    timer1.start(exceptionCallback, 10.milliseconds)
    await sleepAsync(30.milliseconds)
    await timer1.stop()

  test "Starting both timers should execute callbacks sequentially":
    startNumbersTimer()
    startLettersTimer()
    check eventually output == "0a1b2c3d4e"