57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
package types
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/tendermint/go-crypto"
|
|
)
|
|
|
|
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, hb.String(), "Heartbeat{1:000000000000 11/02 (0) {<nil>}}")
|
|
|
|
var key crypto.PrivKeyEd25519
|
|
hb.Signature = key.Sign([]byte("Tendermint"))
|
|
require.Equal(t, hb.String(), "Heartbeat{1:000000000000 11/02 (0) {/FF41E371B9BF.../}}")
|
|
}
|
|
|
|
func TestHeartbeatWriteSignBytes(t *testing.T) {
|
|
var n int
|
|
var err error
|
|
buf := new(bytes.Buffer)
|
|
|
|
hb := &Heartbeat{ValidatorIndex: 1, Height: 10, Round: 1}
|
|
hb.WriteSignBytes("0xdeadbeef", buf, &n, &err)
|
|
require.Equal(t, buf.String(), `{"chain_id":"0xdeadbeef","heartbeat":{"height":10,"round":1,"sequence":0,"validator_address":"","validator_index":1}}`)
|
|
|
|
buf.Reset()
|
|
plainHb := &Heartbeat{}
|
|
plainHb.WriteSignBytes("0xdeadbeef", buf, &n, &err)
|
|
require.Equal(t, buf.String(), `{"chain_id":"0xdeadbeef","heartbeat":{"height":0,"round":0,"sequence":0,"validator_address":"","validator_index":0}}`)
|
|
|
|
require.Panics(t, func() {
|
|
buf.Reset()
|
|
var nilHb *Heartbeat
|
|
nilHb.WriteSignBytes("0xdeadbeef", buf, &n, &err)
|
|
require.Equal(t, buf.String(), "null")
|
|
})
|
|
}
|