mirror of https://github.com/portainer/portainer
74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
package bolttest
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/portainer/portainer/api/bolt"
|
|
"github.com/portainer/portainer/api/filesystem"
|
|
)
|
|
|
|
var errTempDir = errors.New("can't create a temp dir")
|
|
|
|
func MustNewTestStore(init bool) (*bolt.Store, func()) {
|
|
store, teardown, err := NewTestStore(init)
|
|
if err != nil {
|
|
if !errors.Is(err, errTempDir) {
|
|
teardown()
|
|
}
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return store, teardown
|
|
}
|
|
|
|
func NewTestStore(init bool) (*bolt.Store, func(), error) {
|
|
// Creates unique temp directory in a concurrency friendly manner.
|
|
dataStorePath, err := ioutil.TempDir("", "boltdb")
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(errTempDir, err.Error())
|
|
}
|
|
|
|
fileService, err := filesystem.NewService(dataStorePath, "")
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
store, err := bolt.NewStore(dataStorePath, fileService)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
err = store.Open()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
if init {
|
|
err = store.Init()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
}
|
|
|
|
teardown := func() {
|
|
teardown(store, dataStorePath)
|
|
}
|
|
|
|
return store, teardown, nil
|
|
}
|
|
|
|
func teardown(store *bolt.Store, dataStorePath string) {
|
|
err := store.Close()
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
err = os.RemoveAll(dataStorePath)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}
|