mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-07 10:12:19 +00:00
5f76297401
Former-commit-id: 9dfe0415a7809738b7683090d08486d64ca2e4c0 [formerly c8f386658475a6d6a069ab44f63a3d6e39a80ea7] Former-commit-id: 767719430eda75648b408b4f480b25bb68870421
93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
package sharding
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto/sha3"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
"github.com/ethereum/go-ethereum/sharding/database"
|
|
)
|
|
|
|
// Hash returns the hash of a collation's entire contents. Useful for comparison tests.
|
|
func (c *Collation) Hash() (hash common.Hash) {
|
|
hw := sha3.NewKeccak256()
|
|
rlp.Encode(hw, c)
|
|
hw.Sum(hash[:0])
|
|
return hash
|
|
}
|
|
func TestShard_ValidateShardID(t *testing.T) {
|
|
emptyHash := common.StringToHash("")
|
|
emptyAddr := common.StringToAddress("")
|
|
header := NewCollationHeader(big.NewInt(1), &emptyHash, big.NewInt(1), &emptyAddr, []byte{})
|
|
shardDB := database.MakeShardKV()
|
|
shard := MakeShard(big.NewInt(3), shardDB)
|
|
|
|
if err := shard.ValidateShardID(header); err == nil {
|
|
t.Errorf("ShardID validation incorrect. Function should throw error when ShardID's do not match. want=%d. got=%d", header.ShardID().Int64(), shard.ShardID().Int64())
|
|
}
|
|
|
|
header2 := NewCollationHeader(big.NewInt(100), &emptyHash, big.NewInt(1), &emptyAddr, []byte{})
|
|
shard2 := MakeShard(big.NewInt(100), shardDB)
|
|
|
|
if err := shard2.ValidateShardID(header2); err != nil {
|
|
t.Errorf("ShardID validation incorrect. Function should not throw error when ShardID's match. want=%d. got=%d", header2.ShardID().Int64(), shard2.ShardID().Int64())
|
|
}
|
|
}
|
|
|
|
func TestShard_HeaderByHash(t *testing.T) {
|
|
emptyHash := common.StringToHash("")
|
|
emptyAddr := common.StringToAddress("")
|
|
header := NewCollationHeader(big.NewInt(1), &emptyHash, big.NewInt(1), &emptyAddr, []byte{})
|
|
shardDB := database.MakeShardKV()
|
|
shard := MakeShard(big.NewInt(1), shardDB)
|
|
|
|
if err := shard.SaveHeader(header); err != nil {
|
|
t.Fatalf("cannot save collation header: %v", err)
|
|
}
|
|
hash := header.Hash()
|
|
|
|
dbHeader, err := shard.HeaderByHash(&hash)
|
|
if err != nil {
|
|
t.Fatalf("could not fetch collation header by hash: %v", err)
|
|
}
|
|
// Compare the hashes.
|
|
if header.Hash() != dbHeader.Hash() {
|
|
t.Errorf("headers do not match. want=%v. got=%v", header, dbHeader)
|
|
}
|
|
}
|
|
|
|
func TestShard_CollationByHash(t *testing.T) {
|
|
emptyAddr := common.StringToAddress("")
|
|
|
|
// Empty chunk root.
|
|
header := NewCollationHeader(big.NewInt(1), nil, big.NewInt(1), &emptyAddr, []byte{})
|
|
|
|
collation := &Collation{
|
|
header: header,
|
|
body: []byte{1, 2, 3},
|
|
}
|
|
|
|
// We set the chunk root.
|
|
collation.CalculateChunkRoot()
|
|
|
|
shardDB := database.MakeShardKV()
|
|
shard := MakeShard(big.NewInt(1), shardDB)
|
|
|
|
if err := shard.SaveCollation(collation); err != nil {
|
|
t.Fatalf("cannot save collation: %v", err)
|
|
}
|
|
hash := collation.Header().Hash()
|
|
|
|
dbCollation, err := shard.CollationByHash(&hash)
|
|
if err != nil {
|
|
t.Fatalf("could not fetch collation by hash: %v", err)
|
|
}
|
|
|
|
// Compare the hashes.
|
|
if collation.Hash() != dbCollation.Hash() {
|
|
t.Errorf("collations do not match. want=%v. got=%v", collation, dbCollation)
|
|
}
|
|
}
|