mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
36 lines
975 B
Go
36 lines
975 B
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")
|
|
}
|
|
}
|