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 }