prysm-pulse/beacon-chain/db/pending_deposits_test.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{
		&depositContainer{deposit: depToRemove},
		&depositContainer{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{&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{
		&depositContainer{block: big.NewInt(2), deposit: &pb.Deposit{MerkleTreeIndex: 2}},
		&depositContainer{block: big.NewInt(4), deposit: &pb.Deposit{MerkleTreeIndex: 4}},
		&depositContainer{block: big.NewInt(6), deposit: &pb.Deposit{MerkleTreeIndex: 6}},
	}

	deposits := db.PendingDeposits(context.Background(), big.NewInt(4))
	expected := []*pb.Deposit{
		&pb.Deposit{MerkleTreeIndex: 2},
		&pb.Deposit{MerkleTreeIndex: 4},
	}

	if !reflect.DeepEqual(deposits, expected) {
		t.Errorf("Unexpected deposits. got=%+v want=%+v", deposits, expected)
	}
}