prysm-pulse/beacon-chain/db/pending_deposits_test.go
Preston Van Loon 58ca71af23 Insert pending deposits (#1495)
* 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
2019-02-05 11:25:09 -06:00

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")
}
}