prysm-pulse/validator/db/setup_db_test.go
rkapka 277fa3300a
Merge validator databases (#5968)
* merge functionality

* merge testing

* log info when merge completed successfully

* extracted private methods

* fixed linter errors

* fixed compilation errors

* build files clean-up

* fixed failing test

* corrected command description

* close sources inside defer

* corrected documentation of NewKVStore
2020-05-25 11:20:35 -05:00

34 lines
843 B
Go

package db
import (
"crypto/rand"
"fmt"
"math/big"
"os"
"path/filepath"
"testing"
)
func TestClearDB(t *testing.T) {
// Setting up manually is required, since SetupDB() will also register a teardown procedure.
randPath, err := rand.Int(rand.Reader, big.NewInt(1000000))
if err != nil {
t.Fatalf("Could not generate random file path: %v", err)
}
p := filepath.Join(TempDir(), fmt.Sprintf("/%d", randPath))
if err := os.RemoveAll(p); err != nil {
t.Fatalf("Failed to remove directory: %v", err)
}
db, err := NewKVStoreWithPublicKeyBuckets(p, [][48]byte{})
if err != nil {
t.Fatalf("Failed to instantiate DB: %v", err)
}
if err := db.ClearDB(); err != nil {
t.Fatal(err)
}
if _, err := os.Stat(filepath.Join(db.DatabasePath(), databaseFileName)); !os.IsNotExist(err) {
t.Fatalf("DB was not cleared: %v", err)
}
}