fix tests

gofiber · Sep 15, 2023 · e6700b3 · e6700b3
1 parent 09e12a7
commit e6700b3
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 28 deletions.
diff --git a/arangodb/arangodb_test.go b/arangodb/arangodb_test.go
@@ -1,25 +1,15 @@
 package arangodb
 
 import (
-	"os"
 	"testing"
 	"time"
 
 	"github.com/stretchr/testify/require"
 )
 
-var testStore *Storage
-
-func TestMain(m *testing.M) {
-	testStore = New(Config{
-		Reset: true,
-	})
-
-	code := m.Run()
-
-	_ = testStore.Close()
-	os.Exit(code)
-}
+var testStore = New(Config{
+	Reset: true,
+})
 
 func Test_ArangoDB_Set(t *testing.T) {
 	var (

diff --git a/badger/badger_test.go b/badger/badger_test.go
@@ -1,25 +1,15 @@
 package badger
 
 import (
-	"os"
 	"testing"
 	"time"
 
 	"github.com/stretchr/testify/require"
 )
 
-var testStore *Storage
-
-func TestMain(m *testing.M) {
-	testStore = New(Config{
-		Reset: true,
-	})
-
-	code := m.Run()
-
-	_ = testStore.Close()
-	os.Exit(code)
-}
+var testStore = New(Config{
+	Reset: true,
+})
 
 func Test_Badger_Set(t *testing.T) {
 	var (

diff --git a/bbolt/bbolt_test.go b/bbolt/bbolt_test.go
@@ -11,7 +11,8 @@ var testStore *Storage
 
 func TestMain(m *testing.M) {
 	testStore = New(Config{
-		Reset: true,
+		Bucket: "fiber-bucket",
+		Reset:  true,
 	})
 
 	code := m.Run()

diff --git a/bbolt/utils.go b/bbolt/utils.go
@@ -1,6 +1,7 @@
 package bbolt
 
 import (
+	"errors"
 	"github.com/gofiber/utils/v2"
 	"go.etcd.io/bbolt"
 )
@@ -15,6 +16,11 @@ func createBucket(cfg Config, conn *bbolt.DB) error {
 
 func removeBucket(cfg Config, conn *bbolt.DB) error {
 	return conn.Update(func(tx *bbolt.Tx) error {
-		return tx.DeleteBucket(utils.UnsafeBytes(cfg.Bucket))
+		err := tx.DeleteBucket(utils.UnsafeBytes(cfg.Bucket))
+		if errors.Is(err, bbolt.ErrBucketNotFound) {
+			return nil
+		}
+
+		return err
 	})
 }