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.
104 lines
3.1 KiB
104 lines
3.1 KiB
package types
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
"github.com/tendermint/tendermint/crypto/secp256k1"
|
|
)
|
|
|
|
func TestHeartbeatCopy(t *testing.T) {
|
|
hb := &Heartbeat{ValidatorIndex: 1, Height: 10, Round: 1}
|
|
hbCopy := hb.Copy()
|
|
require.Equal(t, hbCopy, hb, "heartbeat copy should be the same")
|
|
hbCopy.Round = hb.Round + 10
|
|
require.NotEqual(t, hbCopy, hb, "heartbeat copy mutation should not change original")
|
|
|
|
var nilHb *Heartbeat
|
|
nilHbCopy := nilHb.Copy()
|
|
require.Nil(t, nilHbCopy, "copy of nil should also return nil")
|
|
}
|
|
|
|
func TestHeartbeatString(t *testing.T) {
|
|
var nilHb *Heartbeat
|
|
require.Contains(t, nilHb.String(), "nil", "expecting a string and no panic")
|
|
|
|
hb := &Heartbeat{ValidatorIndex: 1, Height: 11, Round: 2}
|
|
require.Equal(t, "Heartbeat{1:000000000000 11/02 (0) /000000000000.../}", hb.String())
|
|
|
|
var key ed25519.PrivKeyEd25519
|
|
sig, err := key.Sign([]byte("Tendermint"))
|
|
require.NoError(t, err)
|
|
hb.Signature = sig
|
|
require.Equal(t, "Heartbeat{1:000000000000 11/02 (0) /FF41E371B9BF.../}", hb.String())
|
|
}
|
|
|
|
func TestHeartbeatWriteSignBytes(t *testing.T) {
|
|
chainID := "test_chain_id"
|
|
|
|
{
|
|
testHeartbeat := &Heartbeat{ValidatorIndex: 1, Height: 10, Round: 1}
|
|
signBytes := testHeartbeat.SignBytes(chainID)
|
|
expected, err := cdc.MarshalBinaryLengthPrefixed(CanonicalizeHeartbeat(chainID, testHeartbeat))
|
|
require.NoError(t, err)
|
|
require.Equal(t, expected, signBytes, "Got unexpected sign bytes for Heartbeat")
|
|
}
|
|
|
|
{
|
|
testHeartbeat := &Heartbeat{}
|
|
signBytes := testHeartbeat.SignBytes(chainID)
|
|
expected, err := cdc.MarshalBinaryLengthPrefixed(CanonicalizeHeartbeat(chainID, testHeartbeat))
|
|
require.NoError(t, err)
|
|
require.Equal(t, expected, signBytes, "Got unexpected sign bytes for Heartbeat")
|
|
}
|
|
|
|
require.Panics(t, func() {
|
|
var nilHb *Heartbeat
|
|
signBytes := nilHb.SignBytes(chainID)
|
|
require.Equal(t, string(signBytes), "null")
|
|
})
|
|
}
|
|
|
|
func TestHeartbeatValidateBasic(t *testing.T) {
|
|
testCases := []struct {
|
|
testName string
|
|
malleateHeartBeat func(*Heartbeat)
|
|
expectErr bool
|
|
}{
|
|
{"Good HeartBeat", func(hb *Heartbeat) {}, false},
|
|
{"Invalid address size", func(hb *Heartbeat) {
|
|
hb.ValidatorAddress = nil
|
|
}, true},
|
|
{"Negative validator index", func(hb *Heartbeat) {
|
|
hb.ValidatorIndex = -1
|
|
}, true},
|
|
{"Negative height", func(hb *Heartbeat) {
|
|
hb.Height = -1
|
|
}, true},
|
|
{"Negative round", func(hb *Heartbeat) {
|
|
hb.Round = -1
|
|
}, true},
|
|
{"Negative sequence", func(hb *Heartbeat) {
|
|
hb.Sequence = -1
|
|
}, true},
|
|
{"Missing signature", func(hb *Heartbeat) {
|
|
hb.Signature = nil
|
|
}, true},
|
|
{"Signature too big", func(hb *Heartbeat) {
|
|
hb.Signature = make([]byte, MaxSignatureSize+1)
|
|
}, true},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.testName, func(t *testing.T) {
|
|
hb := &Heartbeat{
|
|
ValidatorAddress: secp256k1.GenPrivKey().PubKey().Address(),
|
|
Signature: make([]byte, 4),
|
|
ValidatorIndex: 1, Height: 10, Round: 1}
|
|
|
|
tc.malleateHeartBeat(hb)
|
|
assert.Equal(t, tc.expectErr, hb.ValidateBasic() != nil, "Validate Basic had an unexpected result")
|
|
})
|
|
}
|
|
}
|