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.
305 lines
8.6 KiB
305 lines
8.6 KiB
package v2
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
tmstate "github.com/tendermint/tendermint/internal/state"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
// pcBlock is a test helper structure with simple types. Its purpose is to help with test readability.
|
|
type pcBlock struct {
|
|
pid string
|
|
height int64
|
|
}
|
|
|
|
// params is a test structure used to create processor state.
|
|
type params struct {
|
|
height int64
|
|
items []pcBlock
|
|
blocksSynced int
|
|
verBL []int64
|
|
appBL []int64
|
|
draining bool
|
|
}
|
|
|
|
// makePcBlock makes an empty block.
|
|
func makePcBlock(height int64) *types.Block {
|
|
return &types.Block{Header: types.Header{Height: height}}
|
|
}
|
|
|
|
// makeState takes test parameters and creates a specific processor state.
|
|
func makeState(p *params) *pcState {
|
|
var (
|
|
tmState = tmstate.State{LastBlockHeight: p.height}
|
|
context = newMockProcessorContext(tmState, p.verBL, p.appBL)
|
|
)
|
|
state := newPcState(context)
|
|
|
|
for _, item := range p.items {
|
|
state.enqueue(types.NodeID(item.pid), makePcBlock(item.height), item.height)
|
|
}
|
|
|
|
state.blocksSynced = p.blocksSynced
|
|
state.draining = p.draining
|
|
return state
|
|
}
|
|
|
|
func mBlockResponse(peerID types.NodeID, height int64) scBlockReceived {
|
|
return scBlockReceived{
|
|
peerID: peerID,
|
|
block: makePcBlock(height),
|
|
}
|
|
}
|
|
|
|
type pcFsmMakeStateValues struct {
|
|
currentState *params
|
|
event Event
|
|
wantState *params
|
|
wantNextEvent Event
|
|
wantErr error
|
|
wantPanic bool
|
|
}
|
|
|
|
type testFields struct {
|
|
name string
|
|
steps []pcFsmMakeStateValues
|
|
}
|
|
|
|
func executeProcessorTests(t *testing.T, tests []testFields) {
|
|
for _, tt := range tests {
|
|
tt := tt
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
var state *pcState
|
|
for _, step := range tt.steps {
|
|
defer func() {
|
|
r := recover()
|
|
if (r != nil) != step.wantPanic {
|
|
t.Errorf("recover = %v, wantPanic = %v", r, step.wantPanic)
|
|
}
|
|
}()
|
|
|
|
// First step must always initialize the currentState as state.
|
|
if step.currentState != nil {
|
|
state = makeState(step.currentState)
|
|
}
|
|
if state == nil {
|
|
panic("Bad (initial?) step")
|
|
}
|
|
|
|
nextEvent, err := state.handle(step.event)
|
|
t.Log(state)
|
|
assert.Equal(t, step.wantErr, err)
|
|
assert.Equal(t, makeState(step.wantState), state)
|
|
assert.Equal(t, step.wantNextEvent, nextEvent)
|
|
// Next step may use the wantedState as their currentState.
|
|
state = makeState(step.wantState)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRProcessPeerError(t *testing.T) {
|
|
tests := []testFields{
|
|
{
|
|
name: "error for existing peer",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}},
|
|
event: scPeerError{peerID: "P2"},
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 1}}},
|
|
wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "error for unknown peer",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}},
|
|
event: scPeerError{peerID: "P3"},
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}},
|
|
wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
executeProcessorTests(t, tests)
|
|
}
|
|
|
|
func TestPcBlockResponse(t *testing.T) {
|
|
tests := []testFields{
|
|
{
|
|
name: "add one block",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{}, event: mBlockResponse("P1", 1),
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 1}}}, wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "add two blocks",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{}, event: mBlockResponse("P1", 3),
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 3}}}, wantNextEvent: noOp,
|
|
},
|
|
{ // use previous wantState as currentState,
|
|
event: mBlockResponse("P1", 4),
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 3}, {"P1", 4}}}, wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
executeProcessorTests(t, tests)
|
|
}
|
|
|
|
func TestRProcessBlockSuccess(t *testing.T) {
|
|
tests := []testFields{
|
|
{
|
|
name: "noop - no blocks over current height",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{}, event: rProcessBlock{},
|
|
wantState: ¶ms{}, wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "noop - high new blocks",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{height: 5, items: []pcBlock{{"P1", 30}, {"P2", 31}}}, event: rProcessBlock{},
|
|
wantState: ¶ms{height: 5, items: []pcBlock{{"P1", 30}, {"P2", 31}}}, wantNextEvent: noOp,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "blocks H+1 and H+2 present",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}}, event: rProcessBlock{},
|
|
wantState: ¶ms{height: 1, items: []pcBlock{{"P2", 2}}, blocksSynced: 1},
|
|
wantNextEvent: pcBlockProcessed{height: 1, peerID: "P1"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "blocks H+1 and H+2 present after draining",
|
|
steps: []pcFsmMakeStateValues{
|
|
{ // some contiguous blocks - on stop check draining is set
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}, {"P1", 4}}},
|
|
event: scFinishedEv{},
|
|
wantState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}, {"P1", 4}}, draining: true},
|
|
wantNextEvent: noOp,
|
|
},
|
|
{
|
|
event: rProcessBlock{},
|
|
wantState: ¶ms{height: 1, items: []pcBlock{{"P2", 2}, {"P1", 4}}, blocksSynced: 1, draining: true},
|
|
wantNextEvent: pcBlockProcessed{height: 1, peerID: "P1"},
|
|
},
|
|
{ // finish when H+1 or/and H+2 are missing
|
|
event: rProcessBlock{},
|
|
wantState: ¶ms{height: 1, items: []pcBlock{{"P2", 2}, {"P1", 4}}, blocksSynced: 1, draining: true},
|
|
wantNextEvent: pcFinished{tmState: tmstate.State{LastBlockHeight: 1}, blocksSynced: 1},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
executeProcessorTests(t, tests)
|
|
}
|
|
|
|
func TestRProcessBlockFailures(t *testing.T) {
|
|
tests := []testFields{
|
|
{
|
|
name: "blocks H+1 and H+2 present from different peers - H+1 verification fails ",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}, verBL: []int64{1}}, event: rProcessBlock{},
|
|
wantState: ¶ms{items: []pcBlock{}, verBL: []int64{1}},
|
|
wantNextEvent: pcBlockVerificationFailure{height: 1, firstPeerID: "P1", secondPeerID: "P2"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "blocks H+1 and H+2 present from same peer - H+1 applyBlock fails ",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}}, appBL: []int64{1}}, event: rProcessBlock{},
|
|
wantState: ¶ms{items: []pcBlock{}, appBL: []int64{1}}, wantPanic: true,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "blocks H+1 and H+2 present from same peers - H+1 verification fails ",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{height: 0, items: []pcBlock{{"P1", 1}, {"P1", 2}, {"P2", 3}},
|
|
verBL: []int64{1}}, event: rProcessBlock{},
|
|
wantState: ¶ms{height: 0, items: []pcBlock{{"P2", 3}}, verBL: []int64{1}},
|
|
wantNextEvent: pcBlockVerificationFailure{height: 1, firstPeerID: "P1", secondPeerID: "P1"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "blocks H+1 and H+2 present from different peers - H+1 applyBlock fails ",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{items: []pcBlock{{"P1", 1}, {"P2", 2}, {"P2", 3}}, appBL: []int64{1}},
|
|
event: rProcessBlock{},
|
|
wantState: ¶ms{items: []pcBlock{{"P2", 3}}, appBL: []int64{1}}, wantPanic: true,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
executeProcessorTests(t, tests)
|
|
}
|
|
|
|
func TestScFinishedEv(t *testing.T) {
|
|
tests := []testFields{
|
|
{
|
|
name: "no blocks",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{height: 100, items: []pcBlock{}, blocksSynced: 100}, event: scFinishedEv{},
|
|
wantState: ¶ms{height: 100, items: []pcBlock{}, blocksSynced: 100},
|
|
wantNextEvent: pcFinished{tmState: tmstate.State{LastBlockHeight: 100}, blocksSynced: 100},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "maxHeight+1 block present",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{height: 100, items: []pcBlock{
|
|
{"P1", 101}}, blocksSynced: 100}, event: scFinishedEv{},
|
|
wantState: ¶ms{height: 100, items: []pcBlock{{"P1", 101}}, blocksSynced: 100},
|
|
wantNextEvent: pcFinished{tmState: tmstate.State{LastBlockHeight: 100}, blocksSynced: 100},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "more blocks present",
|
|
steps: []pcFsmMakeStateValues{
|
|
{
|
|
currentState: ¶ms{height: 100, items: []pcBlock{
|
|
{"P1", 101}, {"P1", 102}}, blocksSynced: 100}, event: scFinishedEv{},
|
|
wantState: ¶ms{height: 100, items: []pcBlock{
|
|
{"P1", 101}, {"P1", 102}}, blocksSynced: 100, draining: true},
|
|
wantNextEvent: noOp,
|
|
wantErr: nil,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
executeProcessorTests(t, tests)
|
|
}
|