39 lines
730 B
Go
39 lines
730 B
Go
|
package util
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
bolt "go.etcd.io/bbolt"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBolt(tb testing.TB, buckets ...[]byte) (*bolt.DB, func()) {
|
||
|
|
tb.Helper()
|
||
|
|
|
||
|
|
f, err := os.CreateTemp("", "bbolt_*.db")
|
||
|
|
require.NoError(tb, err)
|
||
|
|
|
||
|
|
filePath := f.Name()
|
||
|
|
assert.NoError(tb, f.Close())
|
||
|
|
|
||
|
|
db, err := bolt.Open(filePath, os.ModePerm, nil)
|
||
|
|
require.NoError(tb, err)
|
||
|
|
|
||
|
|
for _, bucket := range buckets {
|
||
|
|
bucket := bucket
|
||
|
|
|
||
|
|
assert.NoError(tb, db.Update(func(tx *bolt.Tx) error {
|
||
|
|
_, err := tx.CreateBucket(bucket)
|
||
|
|
|
||
|
|
return err //nolint: errcheck
|
||
|
|
}))
|
||
|
|
}
|
||
|
|
|
||
|
|
return db, func() {
|
||
|
|
db.Close() //nolint: errcheck
|
||
|
|
os.Remove(filePath) //nolint: errcheck
|
||
|
|
}
|
||
|
|
}
|