mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
58ca71af23
* Update config for deposits processing * Add pending deposits database methods * oops forgot the files * move the lock after the nil check * move the lock after the nil check * test inclusion behavior * Insert pending deposits on process deposit log
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
)
|
|
|
|
func TestInsertPendingDeposit(t *testing.T) {
|
|
db := BeaconDB{}
|
|
db.InsertPendingDeposit(context.Background(), &pb.Deposit{}, big.NewInt(111))
|
|
|
|
if len(db.deposits) != 1 {
|
|
t.Error("Deposit not inserted")
|
|
}
|
|
}
|
|
|
|
func TestInsertPendingDeposit_ignoresNilDeposit(t *testing.T) {
|
|
db := BeaconDB{}
|
|
db.InsertPendingDeposit(context.Background(), nil /*deposit*/, nil /*blockNum*/)
|
|
|
|
if len(db.deposits) > 0 {
|
|
t.Error("Unexpected deposit insertion")
|
|
}
|
|
}
|
|
|
|
func TestRemovePendingDeposit(t *testing.T) {
|
|
db := BeaconDB{}
|
|
depToRemove := &pb.Deposit{MerkleTreeIndex: 1}
|
|
otherDep := &pb.Deposit{MerkleTreeIndex: 5}
|
|
db.deposits = []*depositContainer{
|
|
{deposit: depToRemove},
|
|
{deposit: otherDep},
|
|
}
|
|
db.RemovePendingDeposit(context.Background(), depToRemove)
|
|
|
|
if len(db.deposits) != 1 || !proto.Equal(db.deposits[0].deposit, otherDep) {
|
|
t.Error("Failed to remove deposit")
|
|
}
|
|
}
|
|
|
|
func TestRemovePendingDeposit_IgnoresNilDeposit(t *testing.T) {
|
|
db := BeaconDB{}
|
|
db.deposits = []*depositContainer{{deposit: &pb.Deposit{}}}
|
|
db.RemovePendingDeposit(context.Background(), nil /*deposit*/)
|
|
if len(db.deposits) != 1 {
|
|
t.Errorf("Deposit unexpectedly removed")
|
|
}
|
|
}
|
|
|
|
func TestPendingDeposit_RoundTrip(t *testing.T) {
|
|
db := BeaconDB{}
|
|
dep := &pb.Deposit{MerkleTreeIndex: 123}
|
|
db.InsertPendingDeposit(context.Background(), dep, big.NewInt(111))
|
|
db.RemovePendingDeposit(context.Background(), dep)
|
|
if len(db.deposits) != 0 {
|
|
t.Error("Failed to insert & delete a pending deposit")
|
|
}
|
|
}
|
|
|
|
func TestPendingDeposits(t *testing.T) {
|
|
db := BeaconDB{}
|
|
|
|
db.deposits = []*depositContainer{
|
|
{block: big.NewInt(2), deposit: &pb.Deposit{MerkleTreeIndex: 2}},
|
|
{block: big.NewInt(4), deposit: &pb.Deposit{MerkleTreeIndex: 4}},
|
|
{block: big.NewInt(6), deposit: &pb.Deposit{MerkleTreeIndex: 6}},
|
|
}
|
|
|
|
deposits := db.PendingDeposits(context.Background(), big.NewInt(4))
|
|
expected := []*pb.Deposit{
|
|
{MerkleTreeIndex: 2},
|
|
{MerkleTreeIndex: 4},
|
|
}
|
|
|
|
if !reflect.DeepEqual(deposits, expected) {
|
|
t.Errorf("Unexpected deposits. got=%+v want=%+v", deposits, expected)
|
|
}
|
|
|
|
all := db.PendingDeposits(context.Background(), nil)
|
|
if len(all) != len(db.deposits) {
|
|
t.Error("PendingDeposits(ctx, nil) did not return all deposits")
|
|
}
|
|
}
|