prysm-pulse/validator/db/testing/setup_db_test.go
Victor Farazdagi 2f11e55869
Use t.TempDir() in tests (#7769)
* use t.TempDir()

* remove redundant delete

* simplify setupDB()

* simplify db/testing/setup_db

* fix tests
2020-11-10 22:45:17 +00:00

22 lines
594 B
Go

package testing
import (
"os"
"path/filepath"
"testing"
"github.com/prysmaticlabs/prysm/shared/testutil/require"
"github.com/prysmaticlabs/prysm/validator/db/kv"
)
func TestClearDB(t *testing.T) {
// Setting up manually is required, since SetupDB() will also register a teardown procedure.
testDB, err := kv.NewKVStore(t.TempDir(), [][48]byte{})
require.NoError(t, err, "Failed to instantiate DB")
require.NoError(t, testDB.ClearDB())
if _, err := os.Stat(filepath.Join(testDB.DatabasePath(), "validator.db")); !os.IsNotExist(err) {
t.Fatalf("DB was not cleared: %v", err)
}
}