mirror of
https://github.com/taigrr/shorturl
synced 2025-01-18 04:03:16 -08:00
* Changed URL format to shortuuid * Fixed drone CI failure * Moved database to Bitcask and some minor changes * Moved database to Bitcask and some minor changes * Delete coverage.txt
34 lines
423 B
Go
34 lines
423 B
Go
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)
|
|
}
|