nim-dagger/tests/helpers/templeveldb.nim

import os
import std/monotimes
import pkg/datastore
import pkg/chronos
import pkg/questionable/results

type TempLevelDb* = ref object
  currentPath: string
  ds: LevelDbDatastore

var number = 0

proc newDb*(self: TempLevelDb): Datastore =
  if self.currentPath.len > 0:
    raiseAssert("TempLevelDb already active.")
  self.currentPath = getTempDir() / "templeveldb" / $number / $getMonoTime()
  inc number
  createDir(self.currentPath)
  self.ds = LevelDbDatastore.new(self.currentPath).tryGet()
  return self.ds

proc destroyDb*(self: TempLevelDb): Future[void] {.async.} =
  if self.currentPath.len == 0:
    raiseAssert("TempLevelDb not active.")
  try:
    (await self.ds.close()).tryGet()
  finally:
    removeDir(self.currentPath)
    self.currentPath = ""