116 lines
2.3 KiB
Go
116 lines
2.3 KiB
Go
package dbft
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Structures used for type-specific dBFT/payloads implementation to avoid cyclic
|
|
// dependency.
|
|
type (
|
|
hash struct{}
|
|
payloadStub struct {
|
|
height uint32
|
|
typ MessageType
|
|
validatorIndex uint16
|
|
}
|
|
)
|
|
|
|
func (hash) String() string {
|
|
return ""
|
|
}
|
|
|
|
func (p payloadStub) ViewNumber() byte {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) SetViewNumber(byte) {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) Type() MessageType {
|
|
return p.typ
|
|
}
|
|
func (p payloadStub) SetType(MessageType) {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) Payload() any {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) SetPayload(any) {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetChangeView() ChangeView {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetPrepareRequest() PrepareRequest[hash] {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetPrepareResponse() PrepareResponse[hash] {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetCommit() Commit {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetPreCommit() PreCommit { panic("TODO") }
|
|
func (p payloadStub) GetRecoveryRequest() RecoveryRequest {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) GetRecoveryMessage() RecoveryMessage[hash] {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) ValidatorIndex() uint16 {
|
|
return p.validatorIndex
|
|
}
|
|
func (p payloadStub) SetValidatorIndex(uint16) {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) Height() uint32 {
|
|
return p.height
|
|
}
|
|
func (p payloadStub) SetHeight(uint32) {
|
|
panic("TODO")
|
|
}
|
|
func (p payloadStub) Hash() hash {
|
|
panic("TODO")
|
|
}
|
|
|
|
func TestMessageCache(t *testing.T) {
|
|
c := newCache[hash]()
|
|
|
|
p1 := payloadStub{
|
|
height: 3,
|
|
typ: PrepareRequestType,
|
|
}
|
|
c.addMessage(p1)
|
|
|
|
p2 := payloadStub{
|
|
height: 4,
|
|
typ: ChangeViewType,
|
|
}
|
|
c.addMessage(p2)
|
|
|
|
p3 := payloadStub{
|
|
height: 4,
|
|
typ: CommitType,
|
|
}
|
|
c.addMessage(p3)
|
|
|
|
p4 := payloadStub{
|
|
height: 3,
|
|
typ: PreCommitType,
|
|
}
|
|
c.addMessage(p4)
|
|
|
|
box := c.getHeight(3)
|
|
require.Len(t, box.chViews, 0)
|
|
require.Len(t, box.prepare, 1)
|
|
require.Len(t, box.preCommit, 1)
|
|
require.Len(t, box.commit, 0)
|
|
|
|
box = c.getHeight(4)
|
|
require.Len(t, box.chViews, 1)
|
|
require.Len(t, box.prepare, 0)
|
|
require.Len(t, box.preCommit, 0)
|
|
require.Len(t, box.commit, 1)
|
|
}
|