import pkg/chronos

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

  setup:
    for setup in setups:
      await setup()

  teardown:
    for teardown in teardowns:
      await teardown()

  template setup(setupBody) {.inject, used.} =
    setups.add(
      proc() {.async: (handleException: true, raises: [AsyncExceptionError]).} =
        setupBody
    )

  template teardown(teardownBody) {.inject, used.} =
    teardowns.insert(
      proc() {.async: (handleException: true, raises: [AsyncExceptionError]).} =
        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.} =
    let setupProc = proc() =
      setupBody
    setups.add(setupProc)

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