remove db file before testing.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
appleboy · Aug 2, 2016 · 9dd3357 · 9dd3357
1 parent 601472a
commit 9dd3357
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/storage/buntdb/buntdb_test.go b/storage/buntdb/buntdb_test.go
@@ -3,6 +3,7 @@ package buntdb
 import (
 	c "github.com/appleboy/gorush/config"
 	"github.com/stretchr/testify/assert"
+	"os"
 	"testing"
 )
 
@@ -11,6 +12,10 @@ func TestBuntDBEngine(t *testing.T) {
 
 	config := c.BuildDefaultPushConf()
 
+	if _, err := os.Stat(config.Stat.BuntDB.Path); os.IsNotExist(err) {
+		os.RemoveAll(config.Stat.BuntDB.Path)
+	}
+
 	buntDB := New(config)
 	buntDB.Init()
 	buntDB.Reset()