github.com/prysmaticlabs/prysm@v1.4.4/slasher/db/kv/kv_test.go (about)

     1  package kv
     2  
     3  import (
     4  	"io/ioutil"
     5  	"testing"
     6  
     7  	"github.com/prysmaticlabs/prysm/shared/testutil/require"
     8  	"github.com/sirupsen/logrus"
     9  )
    10  
    11  func TestMain(m *testing.M) {
    12  	logrus.SetLevel(logrus.DebugLevel)
    13  	logrus.SetOutput(ioutil.Discard)
    14  
    15  	m.Run()
    16  }
    17  
    18  func setupDB(t testing.TB) *Store {
    19  	cfg := &Config{}
    20  	db, err := NewKVStore(t.TempDir(), cfg)
    21  	require.NoError(t, err, "Failed to instantiate DB")
    22  	t.Cleanup(func() {
    23  		require.NoError(t, db.Close(), "Failed to close database")
    24  	})
    25  	return db
    26  }