mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
|
|
leveldberrors "github.com/syndtr/goleveldb/leveldb/errors"
|
|
)
|
|
|
|
var testDB *DB
|
|
|
|
func init() {
|
|
tmp := fmt.Sprintf("%s/datadir", os.TempDir())
|
|
config := &DBConfig{DataDir: tmp, Name: "beaconchaindata", InMemory: false}
|
|
beaconDB, _ := NewDB(config)
|
|
testDB = beaconDB
|
|
}
|
|
|
|
func TestLifecycle(t *testing.T) {
|
|
tmp := fmt.Sprintf("%s/lifecycledir", os.TempDir())
|
|
config := &DBConfig{DataDir: tmp, Name: "beaconchaindata", InMemory: false}
|
|
b, err := NewDB(config)
|
|
if err != nil {
|
|
t.Fatalf("could not initialize a new DB: %v", err)
|
|
}
|
|
|
|
b.Close()
|
|
|
|
// Access DB after it's stopped, this should fail.
|
|
_, err = b._db.Get([]byte("ralph merkle"))
|
|
|
|
if err.Error() != "leveldb: closed" {
|
|
t.Fatalf("beaconDB close function did not work")
|
|
}
|
|
}
|
|
|
|
// Testing the concurrency with multiple processes attempting to write.
|
|
func Test_DBConcurrent(t *testing.T) {
|
|
var wg sync.WaitGroup
|
|
wg.Add(100)
|
|
for i := 0; i < 100; i++ {
|
|
go func(val string) {
|
|
defer wg.Done()
|
|
if err := testDB._db.Put([]byte("ralph merkle"), []byte(val)); err != nil {
|
|
t.Errorf("could not save value in db: %v", err)
|
|
}
|
|
}(strconv.Itoa(i))
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func Test_DBPut(t *testing.T) {
|
|
if err := testDB._db.Put([]byte("ralph merkle"), []byte{1, 2, 3}); err != nil {
|
|
t.Errorf("could not save value in db: %v", err)
|
|
}
|
|
}
|
|
|
|
func Test_DBHas(t *testing.T) {
|
|
key := []byte("ralph merkle")
|
|
|
|
if err := testDB._db.Put(key, []byte{1, 2, 3}); err != nil {
|
|
t.Fatalf("could not save value in db: %v", err)
|
|
}
|
|
|
|
has, err := testDB._db.Has(key)
|
|
if err != nil {
|
|
t.Errorf("could not check if db has key: %v", err)
|
|
}
|
|
if !has {
|
|
t.Errorf("db should have key: %v", key)
|
|
}
|
|
|
|
key2 := []byte{}
|
|
has2, err := testDB._db.Has(key2)
|
|
if err != nil {
|
|
t.Errorf("could not check if db has key: %v", err)
|
|
}
|
|
if has2 {
|
|
t.Errorf("db should not have non-existent key: %v", key2)
|
|
}
|
|
}
|
|
|
|
func Test_DBGet(t *testing.T) {
|
|
key := []byte("ralph merkle")
|
|
|
|
if err := testDB._db.Put(key, []byte{1, 2, 3}); err != nil {
|
|
t.Fatalf("could not save value in db: %v", err)
|
|
}
|
|
|
|
val, err := testDB._db.Get(key)
|
|
if err != nil {
|
|
t.Errorf("get failed: %v", err)
|
|
}
|
|
if len(val) == 0 {
|
|
t.Errorf("no value stored for key")
|
|
}
|
|
|
|
key2 := []byte{}
|
|
val2, err := testDB._db.Get(key2)
|
|
if err == nil || err.Error() != leveldberrors.ErrNotFound.Error() {
|
|
t.Errorf("Expected error %v but got %v", leveldberrors.ErrNotFound, err)
|
|
}
|
|
if len(val2) != 0 {
|
|
t.Errorf("non-existent key should not have a value. key=%v, value=%v", key2, val2)
|
|
}
|
|
}
|
|
|
|
func Test_DBDelete(t *testing.T) {
|
|
key := []byte("ralph merkle")
|
|
|
|
if err := testDB._db.Put(key, []byte{1, 2, 3}); err != nil {
|
|
t.Fatalf("could not save value in db: %v", err)
|
|
}
|
|
|
|
if err := testDB._db.Delete(key); err != nil {
|
|
t.Errorf("could not delete key: %v", key)
|
|
}
|
|
}
|