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

     1  package kv
     2  
     3  import (
     4  	"context"
     5  	"io/ioutil"
     6  	"testing"
     7  
     8  	"github.com/prysmaticlabs/prysm/shared/testutil/require"
     9  	"github.com/sirupsen/logrus"
    10  )
    11  
    12  func TestMain(m *testing.M) {
    13  	logrus.SetLevel(logrus.DebugLevel)
    14  	logrus.SetOutput(ioutil.Discard)
    15  
    16  	m.Run()
    17  }
    18  
    19  // setupDB instantiates and returns a DB instance for the validator client.
    20  func setupDB(t testing.TB, pubkeys [][48]byte) *Store {
    21  	db, err := NewKVStore(context.Background(), t.TempDir(), &Config{
    22  		PubKeys: pubkeys,
    23  	})
    24  	require.NoError(t, err, "Failed to instantiate DB")
    25  	err = db.UpdatePublicKeysBuckets(pubkeys)
    26  	require.NoError(t, err, "Failed to create old buckets for public keys")
    27  	t.Cleanup(func() {
    28  		require.NoError(t, db.Close(), "Failed to close database")
    29  		require.NoError(t, db.ClearDB(), "Failed to clear database")
    30  	})
    31  	return db
    32  }