package main

import (
	"io/ioutil"
	"log"
	"os"
	"testing"

	"github.com/prologic/bitcask"
)

func TestMain(m *testing.M) {
	tmpdir, err := ioutil.TempDir("", "shorturl")
	if err != nil {
		log.Fatal(err)
	}
	defer os.Remove(tmpdir)

	db, err := bitcask.Open(tmpdir)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	ret := m.Run()

	err = os.RemoveAll(tmpdir)
	if err != nil {
		log.Fatal(err)
	}

	os.Exit(ret)
}