import pkg/chronos

# Allow multiple setups and teardowns in a test suite
template asyncmultisetup* =
  var setups: seq[proc: Future[void] {.gcsafe.}]
  var teardowns: seq[proc: Future[void] {.gcsafe.}]

  setup:
    for setup in setups:
      await setup()

  teardown:
    for teardown in teardowns:
      await teardown()

  template setup(setupBody) {.inject, used.} =
    setups.add(proc {.async.} = setupBody)

  template teardown(teardownBody) {.inject, used.} =
    teardowns.insert(proc {.async.} = teardownBody)

template multisetup* =
  var setups: seq[proc() {.gcsafe.}]
  var teardowns: seq[proc() {.gcsafe.}]

  setup:
    for setup in setups:
      setup()

  teardown:
    for teardown in teardowns:
      teardown()

  template setup(setupBody) {.inject, used.} =
    setups.add(proc = setupBody)

  template teardown(teardownBody) {.inject, used.} =
    teardowns.insert(proc = teardownBody)