github.com/muhammadn/cortex@v1.9.1-0.20220510110439-46bb7000d03d/pkg/configs/db/dbtest/unit.go (about)

     1  //go:build !integration
     2  // +build !integration
     3  
     4  package dbtest
     5  
     6  import (
     7  	"testing"
     8  
     9  	"github.com/stretchr/testify/require"
    10  	"github.com/weaveworks/common/logging"
    11  
    12  	"github.com/cortexproject/cortex/pkg/configs/db"
    13  )
    14  
    15  // Setup sets up stuff for testing, creating a new database
    16  func Setup(t *testing.T) db.DB {
    17  	require.NoError(t, logging.Setup("debug"))
    18  	database, err := db.New(db.Config{
    19  		URI: "memory://",
    20  	})
    21  	require.NoError(t, err)
    22  	return database
    23  }
    24  
    25  // Cleanup cleans up after a test
    26  func Cleanup(t *testing.T, database db.DB) {
    27  	require.NoError(t, database.Close())
    28  }