prysm-pulse/beacon-chain/db/block_vote_cache_test.go

package db

import (
	"reflect"
	"testing"

	"github.com/prysmaticlabs/prysm/beacon-chain/utils"
)

func TestBlockVoteCacheReadWrite(t *testing.T) {
	db := setupDB(t)
	defer teardownDB(t, db)

	blockVoteCache := utils.NewBlockVoteCache()
	blockVote1 := &utils.BlockVote{VoterIndices: []uint32{1, 2, 3}, VoteTotalDeposit: 6}
	blockVote2 := &utils.BlockVote{VoterIndices: []uint32{4, 5, 6}, VoteTotalDeposit: 15}
	blockHash1 := [32]byte{1}
	blockHash2 := [32]byte{2}
	blockVoteCache[blockHash1] = blockVote1
	blockVoteCache[blockHash2] = blockVote2

	var err error
	if err = db.WriteBlockVoteCache(blockVoteCache); err != nil {
		t.Fatalf("failed to write block vote cache to DB: %v", err)
	}

	blockVoteCache2, err := db.ReadBlockVoteCache([][32]byte{blockHash1, blockHash2})
	if err != nil {
		t.Fatalf("failed to read block vote cache from DB: %v", err)
	}

	if !reflect.DeepEqual(blockVoteCache, blockVoteCache2) {
		t.Error("block vote cache read write don't match")
	}
}

func TestBlockVoteCacheDelete(t *testing.T) {
	db := setupDB(t)
	defer teardownDB(t, db)

	blockVoteCache := utils.NewBlockVoteCache()
	blockVote1 := &utils.BlockVote{VoterIndices: []uint32{1, 2, 3}, VoteTotalDeposit: 6}
	blockVote2 := &utils.BlockVote{VoterIndices: []uint32{4, 5, 6}, VoteTotalDeposit: 15}
	blockHash1 := [32]byte{1}
	blockHash2 := [32]byte{2}
	blockVoteCache[blockHash1] = blockVote1
	blockVoteCache[blockHash2] = blockVote2

	var err error
	if err = db.WriteBlockVoteCache(blockVoteCache); err != nil {
		t.Fatalf("failed to write block vote cache to DB: %v", err)
	}

	if err = db.DeleteBlockVoteCache([][32]byte{blockHash1, blockHash2}); err != nil {
		t.Fatalf("failed to delete block vote cache from DB: %v", err)
	}

	var voteCache utils.BlockVoteCache
	voteCache, err = db.ReadBlockVoteCache([][32]byte{blockHash1})
	if err != nil {
		t.Error("should not expect error when reading already deleted block vote cache")
	}
	if len(voteCache) != 0 {
		t.Error("should expect empty result when reading already deleted block vote cache")
	}
}