You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
188 lines
5.4 KiB
188 lines
5.4 KiB
package types
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/tendermint/go-amino"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto"
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
"github.com/tendermint/tendermint/crypto/secp256k1"
|
|
"github.com/tendermint/tendermint/version"
|
|
)
|
|
|
|
func TestABCIPubKey(t *testing.T) {
|
|
pkEd := ed25519.GenPrivKey().PubKey()
|
|
pkSecp := secp256k1.GenPrivKey().PubKey()
|
|
testABCIPubKey(t, pkEd, ABCIPubKeyTypeEd25519)
|
|
testABCIPubKey(t, pkSecp, ABCIPubKeyTypeSecp256k1)
|
|
}
|
|
|
|
func testABCIPubKey(t *testing.T, pk crypto.PubKey, typeStr string) {
|
|
abciPubKey := TM2PB.PubKey(pk)
|
|
pk2, err := PB2TM.PubKey(abciPubKey)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, pk, pk2)
|
|
}
|
|
|
|
func TestABCIValidators(t *testing.T) {
|
|
pkEd := ed25519.GenPrivKey().PubKey()
|
|
|
|
// correct validator
|
|
tmValExpected := NewValidator(pkEd, 10)
|
|
|
|
tmVal := NewValidator(pkEd, 10)
|
|
|
|
abciVal := TM2PB.ValidatorUpdate(tmVal)
|
|
tmVals, err := PB2TM.ValidatorUpdates([]abci.ValidatorUpdate{abciVal})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, tmValExpected, tmVals[0])
|
|
|
|
abciVals := TM2PB.ValidatorUpdates(NewValidatorSet(tmVals))
|
|
assert.Equal(t, []abci.ValidatorUpdate{abciVal}, abciVals)
|
|
|
|
// val with address
|
|
tmVal.Address = pkEd.Address()
|
|
|
|
abciVal = TM2PB.ValidatorUpdate(tmVal)
|
|
tmVals, err = PB2TM.ValidatorUpdates([]abci.ValidatorUpdate{abciVal})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, tmValExpected, tmVals[0])
|
|
|
|
// val with incorrect pubkey data
|
|
abciVal = TM2PB.ValidatorUpdate(tmVal)
|
|
abciVal.PubKey.Data = []byte("incorrect!")
|
|
tmVals, err = PB2TM.ValidatorUpdates([]abci.ValidatorUpdate{abciVal})
|
|
assert.NotNil(t, err)
|
|
assert.Nil(t, tmVals)
|
|
}
|
|
|
|
func TestABCIConsensusParams(t *testing.T) {
|
|
cp := DefaultConsensusParams()
|
|
abciCP := TM2PB.ConsensusParams(cp)
|
|
cp2 := PB2TM.ConsensusParams(abciCP)
|
|
|
|
assert.Equal(t, *cp, cp2)
|
|
}
|
|
|
|
func newHeader(
|
|
height, numTxs int64,
|
|
commitHash, dataHash, evidenceHash []byte,
|
|
) *Header {
|
|
return &Header{
|
|
Height: height,
|
|
NumTxs: numTxs,
|
|
LastCommitHash: commitHash,
|
|
DataHash: dataHash,
|
|
EvidenceHash: evidenceHash,
|
|
}
|
|
}
|
|
|
|
func TestABCIHeader(t *testing.T) {
|
|
// build a full header
|
|
var height int64 = 5
|
|
var numTxs int64 = 3
|
|
header := newHeader(
|
|
height, numTxs,
|
|
[]byte("lastCommitHash"), []byte("dataHash"), []byte("evidenceHash"),
|
|
)
|
|
protocolVersion := version.Consensus{7, 8}
|
|
timestamp := time.Now()
|
|
lastBlockID := BlockID{
|
|
Hash: []byte("hash"),
|
|
PartsHeader: PartSetHeader{
|
|
Total: 10,
|
|
Hash: []byte("hash"),
|
|
},
|
|
}
|
|
var totalTxs int64 = 100
|
|
header.Populate(
|
|
protocolVersion, "chainID",
|
|
timestamp, lastBlockID, totalTxs,
|
|
[]byte("valHash"), []byte("nextValHash"),
|
|
[]byte("consHash"), []byte("appHash"), []byte("lastResultsHash"),
|
|
[]byte("proposerAddress"),
|
|
)
|
|
|
|
cdc := amino.NewCodec()
|
|
headerBz := cdc.MustMarshalBinaryBare(header)
|
|
|
|
pbHeader := TM2PB.Header(header)
|
|
pbHeaderBz, err := proto.Marshal(&pbHeader)
|
|
assert.NoError(t, err)
|
|
|
|
// assert some fields match
|
|
assert.EqualValues(t, protocolVersion.Block, pbHeader.Version.Block)
|
|
assert.EqualValues(t, protocolVersion.App, pbHeader.Version.App)
|
|
assert.EqualValues(t, "chainID", pbHeader.ChainID)
|
|
assert.EqualValues(t, height, pbHeader.Height)
|
|
assert.EqualValues(t, timestamp, pbHeader.Time)
|
|
assert.EqualValues(t, numTxs, pbHeader.NumTxs)
|
|
assert.EqualValues(t, totalTxs, pbHeader.TotalTxs)
|
|
assert.EqualValues(t, lastBlockID.Hash, pbHeader.LastBlockId.Hash)
|
|
assert.EqualValues(t, []byte("lastCommitHash"), pbHeader.LastCommitHash)
|
|
assert.Equal(t, []byte("proposerAddress"), pbHeader.ProposerAddress)
|
|
|
|
// assert the encodings match
|
|
// NOTE: they don't yet because Amino encodes
|
|
// int64 as zig-zag and we're using non-zigzag in the protobuf.
|
|
// See https://github.com/tendermint/tendermint/issues/2682
|
|
_, _ = headerBz, pbHeaderBz
|
|
// assert.EqualValues(t, headerBz, pbHeaderBz)
|
|
|
|
}
|
|
|
|
func TestABCIEvidence(t *testing.T) {
|
|
val := NewMockPV()
|
|
blockID := makeBlockID([]byte("blockhash"), 1000, []byte("partshash"))
|
|
blockID2 := makeBlockID([]byte("blockhash2"), 1000, []byte("partshash"))
|
|
const chainID = "mychain"
|
|
ev := &DuplicateVoteEvidence{
|
|
PubKey: val.GetPubKey(),
|
|
VoteA: makeVote(val, chainID, 0, 10, 2, 1, blockID),
|
|
VoteB: makeVote(val, chainID, 0, 10, 2, 1, blockID2),
|
|
}
|
|
abciEv := TM2PB.Evidence(
|
|
ev,
|
|
NewValidatorSet([]*Validator{NewValidator(val.GetPubKey(), 10)}),
|
|
time.Now(),
|
|
)
|
|
|
|
assert.Equal(t, "duplicate/vote", abciEv.Type)
|
|
}
|
|
|
|
type pubKeyEddie struct{}
|
|
|
|
func (pubKeyEddie) Address() Address { return []byte{} }
|
|
func (pubKeyEddie) Bytes() []byte { return []byte{} }
|
|
func (pubKeyEddie) VerifyBytes(msg []byte, sig []byte) bool { return false }
|
|
func (pubKeyEddie) Equals(crypto.PubKey) bool { return false }
|
|
|
|
func TestABCIValidatorFromPubKeyAndPower(t *testing.T) {
|
|
pubkey := ed25519.GenPrivKey().PubKey()
|
|
|
|
abciVal := TM2PB.NewValidatorUpdate(pubkey, 10)
|
|
assert.Equal(t, int64(10), abciVal.Power)
|
|
|
|
assert.Panics(t, func() { TM2PB.NewValidatorUpdate(nil, 10) })
|
|
assert.Panics(t, func() { TM2PB.NewValidatorUpdate(pubKeyEddie{}, 10) })
|
|
}
|
|
|
|
func TestABCIValidatorWithoutPubKey(t *testing.T) {
|
|
pkEd := ed25519.GenPrivKey().PubKey()
|
|
|
|
abciVal := TM2PB.Validator(NewValidator(pkEd, 10))
|
|
|
|
// pubkey must be nil
|
|
tmValExpected := abci.Validator{
|
|
Address: pkEd.Address(),
|
|
Power: 10,
|
|
}
|
|
|
|
assert.Equal(t, tmValExpected, abciVal)
|
|
}
|