forked from TrueCloudLab/neoneo-go
45 lines
917 B
Go
45 lines
917 B
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
type tempBadgerDB struct {
|
||
|
|
*BadgerDBStore
|
||
|
|
dir string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tbdb *tempBadgerDB) Close() error {
|
||
|
|
err := tbdb.BadgerDBStore.Close()
|
||
|
|
// Make test fail if failed to cleanup, even though technically it's
|
||
|
|
// not a BadgerDBStore problem.
|
||
|
|
osErr := os.RemoveAll(tbdb.dir)
|
||
|
|
if osErr != nil {
|
||
|
|
return osErr
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func newBadgerDBForTesting(t *testing.T) Store {
|
||
|
|
bdbDir, err := ioutil.TempDir(os.TempDir(), "testbadgerdb")
|
||
|
|
require.Nil(t, err, "failed to setup temporary directory")
|
||
|
|
|
||
|
|
dbConfig := DBConfiguration{
|
||
|
|
Type: "badgerdb",
|
||
|
|
BadgerDBOptions: BadgerDBOptions{
|
||
|
|
Dir: bdbDir,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
newBadgerStore, err := NewBadgerDBStore(dbConfig.BadgerDBOptions)
|
||
|
|
require.Nil(t, err, "NewBadgerDBStore error")
|
||
|
|
tbdb := &tempBadgerDB{
|
||
|
|
BadgerDBStore: newBadgerStore,
|
||
|
|
dir: bdbDir,
|
||
|
|
}
|
||
|
|
return tbdb
|
||
|
|
}
|