- package v1
-
- import (
- "fmt"
- "testing"
- "time"
-
- "github.com/stretchr/testify/assert"
-
- "github.com/tendermint/tendermint/libs/log"
- tmmath "github.com/tendermint/tendermint/libs/math"
- tmrand "github.com/tendermint/tendermint/libs/rand"
- "github.com/tendermint/tendermint/p2p"
- "github.com/tendermint/tendermint/types"
- )
-
- type lastBlockRequestT struct {
- peerID p2p.ID
- height int64
- }
-
- type lastPeerErrorT struct {
- peerID p2p.ID
- err error
- }
-
- // reactor for FSM testing
- type testReactor struct {
- logger log.Logger
- fsm *BcReactorFSM
- numStatusRequests int
- numBlockRequests int
- lastBlockRequest lastBlockRequestT
- lastPeerError lastPeerErrorT
- stateTimerStarts map[string]int
- }
-
- func sendEventToFSM(fsm *BcReactorFSM, ev bReactorEvent, data bReactorEventData) error {
- return fsm.Handle(&bcReactorMessage{event: ev, data: data})
- }
-
- type fsmStepTestValues struct {
- currentState string
- event bReactorEvent
- data bReactorEventData
-
- wantErr error
- wantState string
- wantStatusReqSent bool
- wantReqIncreased bool
- wantNewBlocks []int64
- wantRemovedPeers []p2p.ID
- }
-
- // ---------------------------------------------------------------------------
- // helper test function for different FSM events, state and expected behavior
- func sStopFSMEv(current, expected string) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: stopFSMEv,
- wantState: expected,
- wantErr: errNoErrorFinished}
- }
-
- func sUnknownFSMEv(current string) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: 1234,
- wantState: current,
- wantErr: errInvalidEvent}
- }
-
- func sStartFSMEv() fsmStepTestValues {
- return fsmStepTestValues{
- currentState: "unknown",
- event: startFSMEv,
- wantState: "waitForPeer",
- wantStatusReqSent: true}
- }
-
- func sStateTimeoutEv(current, expected string, timedoutState string, wantErr error) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: stateTimeoutEv,
- data: bReactorEventData{
- stateName: timedoutState,
- },
- wantState: expected,
- wantErr: wantErr,
- }
- }
-
- func sProcessedBlockEv(current, expected string, reactorError error) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: processedBlockEv,
- data: bReactorEventData{
- err: reactorError,
- },
- wantState: expected,
- wantErr: reactorError,
- }
- }
-
- func sStatusEv(current, expected string, peerID p2p.ID, height int64, err error) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: statusResponseEv,
- data: bReactorEventData{peerID: peerID, height: height},
- wantState: expected,
- wantErr: err}
- }
-
- func sMakeRequestsEv(current, expected string, maxPendingRequests int) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: makeRequestsEv,
- data: bReactorEventData{maxNumRequests: maxPendingRequests},
- wantState: expected,
- wantReqIncreased: true,
- }
- }
-
- func sMakeRequestsEvErrored(current, expected string,
- maxPendingRequests int, err error, peersRemoved []p2p.ID) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: makeRequestsEv,
- data: bReactorEventData{maxNumRequests: maxPendingRequests},
- wantState: expected,
- wantErr: err,
- wantRemovedPeers: peersRemoved,
- wantReqIncreased: true,
- }
- }
-
- func sBlockRespEv(current, expected string, peerID p2p.ID, height int64, prevBlocks []int64) fsmStepTestValues {
- txs := []types.Tx{types.Tx("foo"), types.Tx("bar")}
- return fsmStepTestValues{
- currentState: current,
- event: blockResponseEv,
- data: bReactorEventData{
- peerID: peerID,
- height: height,
- block: types.MakeBlock(height, txs, nil, nil),
- length: 100},
- wantState: expected,
- wantNewBlocks: append(prevBlocks, height),
- }
- }
-
- func sBlockRespEvErrored(current, expected string,
- peerID p2p.ID, height int64, prevBlocks []int64, wantErr error, peersRemoved []p2p.ID) fsmStepTestValues {
- txs := []types.Tx{types.Tx("foo"), types.Tx("bar")}
-
- return fsmStepTestValues{
- currentState: current,
- event: blockResponseEv,
- data: bReactorEventData{
- peerID: peerID,
- height: height,
- block: types.MakeBlock(height, txs, nil, nil),
- length: 100},
- wantState: expected,
- wantErr: wantErr,
- wantRemovedPeers: peersRemoved,
- wantNewBlocks: prevBlocks,
- }
- }
-
- func sPeerRemoveEv(current, expected string, peerID p2p.ID, err error, peersRemoved []p2p.ID) fsmStepTestValues {
- return fsmStepTestValues{
- currentState: current,
- event: peerRemoveEv,
- data: bReactorEventData{
- peerID: peerID,
- err: err,
- },
- wantState: expected,
- wantRemovedPeers: peersRemoved,
- }
- }
-
- // --------------------------------------------
-
- func newTestReactor(height int64) *testReactor {
- testBcR := &testReactor{logger: log.TestingLogger(), stateTimerStarts: make(map[string]int)}
- testBcR.fsm = NewFSM(height, testBcR)
- testBcR.fsm.SetLogger(testBcR.logger)
- return testBcR
- }
-
- func fixBlockResponseEvStep(step *fsmStepTestValues, testBcR *testReactor) {
- // There is currently no good way to know to which peer a block request was sent.
- // So in some cases where it does not matter, before we simulate a block response
- // we cheat and look where it is expected from.
- if step.event == blockResponseEv {
- height := step.data.height
- peerID, ok := testBcR.fsm.pool.blocks[height]
- if ok {
- step.data.peerID = peerID
- }
- }
- }
-
- type testFields struct {
- name string
- startingHeight int64
- maxRequestsPerPeer int
- maxPendingRequests int
- steps []fsmStepTestValues
- }
-
- func executeFSMTests(t *testing.T, tests []testFields, matchRespToReq bool) {
- for _, tt := range tests {
- tt := tt
- t.Run(tt.name, func(t *testing.T) {
- // Create test reactor
- testBcR := newTestReactor(tt.startingHeight)
-
- if tt.maxRequestsPerPeer != 0 {
- maxRequestsPerPeer = tt.maxRequestsPerPeer
- }
-
- for _, step := range tt.steps {
- step := step
- assert.Equal(t, step.currentState, testBcR.fsm.state.name)
-
- var heightBefore int64
- if step.event == processedBlockEv && step.data.err == errBlockVerificationFailure {
- heightBefore = testBcR.fsm.pool.Height
- }
- oldNumStatusRequests := testBcR.numStatusRequests
- oldNumBlockRequests := testBcR.numBlockRequests
- if matchRespToReq {
- fixBlockResponseEvStep(&step, testBcR)
- }
-
- fsmErr := sendEventToFSM(testBcR.fsm, step.event, step.data)
- assert.Equal(t, step.wantErr, fsmErr)
-
- if step.wantStatusReqSent {
- assert.Equal(t, oldNumStatusRequests+1, testBcR.numStatusRequests)
- } else {
- assert.Equal(t, oldNumStatusRequests, testBcR.numStatusRequests)
- }
-
- if step.wantReqIncreased {
- assert.True(t, oldNumBlockRequests < testBcR.numBlockRequests)
- } else {
- assert.Equal(t, oldNumBlockRequests, testBcR.numBlockRequests)
- }
-
- for _, height := range step.wantNewBlocks {
- _, err := testBcR.fsm.pool.BlockAndPeerAtHeight(height)
- assert.Nil(t, err)
- }
- if step.event == processedBlockEv && step.data.err == errBlockVerificationFailure {
- heightAfter := testBcR.fsm.pool.Height
- assert.Equal(t, heightBefore, heightAfter)
- firstAfter, err1 := testBcR.fsm.pool.BlockAndPeerAtHeight(testBcR.fsm.pool.Height)
- secondAfter, err2 := testBcR.fsm.pool.BlockAndPeerAtHeight(testBcR.fsm.pool.Height + 1)
- assert.NotNil(t, err1)
- assert.NotNil(t, err2)
- assert.Nil(t, firstAfter)
- assert.Nil(t, secondAfter)
- }
-
- assert.Equal(t, step.wantState, testBcR.fsm.state.name)
-
- if step.wantState == "finished" {
- assert.True(t, testBcR.fsm.isCaughtUp())
- }
- }
- })
- }
- }
-
- func TestFSMBasic(t *testing.T) {
- tests := []testFields{
- {
- name: "one block, one peer - TS2",
- startingHeight: 1,
- maxRequestsPerPeer: 2,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 2, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 2, []int64{1}),
- sProcessedBlockEv("waitForBlock", "finished", nil),
- },
- },
- {
- name: "multi block, multi peer - TS2",
- startingHeight: 1,
- maxRequestsPerPeer: 2,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 4, nil),
- sStatusEv("waitForBlock", "waitForBlock", "P2", 4, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 2, []int64{1}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 3, []int64{1, 2}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 4, []int64{1, 2, 3}),
-
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
- sProcessedBlockEv("waitForBlock", "finished", nil),
- },
- },
- }
-
- executeFSMTests(t, tests, true)
- }
-
- func TestFSMBlockVerificationFailure(t *testing.T) {
- tests := []testFields{
- {
- name: "block verification failure - TS2 variant",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
-
- // add P1 and get blocks 1-3 from it
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 2, []int64{1}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 3, []int64{1, 2}),
-
- // add P2
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, nil),
-
- // process block failure, should remove P1 and all blocks
- sProcessedBlockEv("waitForBlock", "waitForBlock", errBlockVerificationFailure),
-
- // get blocks 1-3 from P2
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 2, []int64{1}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 3, []int64{1, 2}),
-
- // finish after processing blocks 1 and 2
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
- sProcessedBlockEv("waitForBlock", "finished", nil),
- },
- },
- }
-
- executeFSMTests(t, tests, false)
- }
-
- func TestFSMBadBlockFromPeer(t *testing.T) {
- tests := []testFields{
- {
- name: "block we haven't asked for",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1 and ask for blocks 1-3
- sStatusEv("waitForPeer", "waitForBlock", "P1", 300, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- // blockResponseEv for height 100 should cause an error
- sBlockRespEvErrored("waitForBlock", "waitForPeer",
- "P1", 100, []int64{}, errMissingBlock, []p2p.ID{}),
- },
- },
- {
- name: "block we already have",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1 and get block 1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 100, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock",
- "P1", 1, []int64{}),
-
- // Get block 1 again. Since peer is removed together with block 1,
- // the blocks present in the pool should be {}
- sBlockRespEvErrored("waitForBlock", "waitForPeer",
- "P1", 1, []int64{}, errDuplicateBlock, []p2p.ID{"P1"}),
- },
- },
- {
- name: "block from unknown peer",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1 and get block 1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
-
- // get block 1 from unknown peer P2
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEvErrored("waitForBlock", "waitForBlock",
- "P2", 1, []int64{}, errBadDataFromPeer, []p2p.ID{"P2"}),
- },
- },
- {
- name: "block from wrong peer",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1, make requests for blocks 1-3 to P1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- // add P2
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, nil),
-
- // receive block 1 from P2
- sBlockRespEvErrored("waitForBlock", "waitForBlock",
- "P2", 1, []int64{}, errBadDataFromPeer, []p2p.ID{"P2"}),
- },
- },
- }
-
- executeFSMTests(t, tests, false)
- }
-
- func TestFSMBlockAtCurrentHeightDoesNotArriveInTime(t *testing.T) {
- tests := []testFields{
- {
- name: "block at current height undelivered - TS5",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1, get blocks 1 and 2, process block 1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock",
- "P1", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock",
- "P1", 2, []int64{1}),
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
-
- // add P2
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, nil),
-
- // timeout on block 3, P1 should be removed
- sStateTimeoutEv("waitForBlock", "waitForBlock", "waitForBlock", errNoPeerResponseForCurrentHeights),
-
- // make requests and finish by receiving blocks 2 and 3 from P2
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 2, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P2", 3, []int64{2}),
- sProcessedBlockEv("waitForBlock", "finished", nil),
- },
- },
- {
- name: "block at current height undelivered, at maxPeerHeight after peer removal - TS3",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1, request blocks 1-3 from P1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- // add P2 (tallest)
- sStatusEv("waitForBlock", "waitForBlock", "P2", 30, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- // receive blocks 1-3 from P1
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 1, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 2, []int64{1}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 3, []int64{1, 2}),
-
- // process blocks at heights 1 and 2
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
-
- // timeout on block at height 4
- sStateTimeoutEv("waitForBlock", "finished", "waitForBlock", nil),
- },
- },
- }
-
- executeFSMTests(t, tests, true)
- }
-
- func TestFSMPeerRelatedEvents(t *testing.T) {
- tests := []testFields{
- {
- name: "peer remove event with no blocks",
- startingHeight: 1,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1, P2, P3
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, nil),
- sStatusEv("waitForBlock", "waitForBlock", "P3", 3, nil),
-
- // switch removes P2
- sPeerRemoveEv("waitForBlock", "waitForBlock", "P2", errSwitchRemovesPeer, []p2p.ID{"P2"}),
- },
- },
- {
- name: "only peer removed while in waitForBlock state",
- startingHeight: 100,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 200, nil),
-
- // switch removes P1
- sPeerRemoveEv("waitForBlock", "waitForPeer", "P1", errSwitchRemovesPeer, []p2p.ID{"P1"}),
- },
- },
- {
- name: "highest peer removed while in waitForBlock state, node reaches maxPeerHeight - TS4 ",
- startingHeight: 100,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1 and make requests
- sStatusEv("waitForPeer", "waitForBlock", "P1", 101, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- // add P2
- sStatusEv("waitForBlock", "waitForBlock", "P2", 200, nil),
-
- // get blocks 100 and 101 from P1 and process block at height 100
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 100, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 101, []int64{100}),
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
-
- // switch removes peer P1, should be finished
- sPeerRemoveEv("waitForBlock", "finished", "P2", errSwitchRemovesPeer, []p2p.ID{"P2"}),
- },
- },
- {
- name: "highest peer lowers its height in waitForBlock state, node reaches maxPeerHeight - TS4",
- startingHeight: 100,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1 and make requests
- sStatusEv("waitForPeer", "waitForBlock", "P1", 101, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
-
- // add P2
- sStatusEv("waitForBlock", "waitForBlock", "P2", 200, nil),
-
- // get blocks 100 and 101 from P1
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 100, []int64{}),
- sBlockRespEv("waitForBlock", "waitForBlock", "P1", 101, []int64{100}),
-
- // processed block at heights 100
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil),
-
- // P2 becomes short
- sStatusEv("waitForBlock", "finished", "P2", 100, errPeerLowersItsHeight),
- },
- },
- {
- name: "new short peer while in waitForPeer state",
- startingHeight: 100,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForPeer", "P1", 3, errPeerTooShort),
- },
- },
- {
- name: "new short peer while in waitForBlock state",
- startingHeight: 100,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 200, nil),
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, errPeerTooShort),
- },
- },
- {
- name: "only peer updated with low height while in waitForBlock state",
- startingHeight: 100,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 200, nil),
- sStatusEv("waitForBlock", "waitForPeer", "P1", 3, errPeerLowersItsHeight),
- },
- },
- {
- name: "peer does not exist in the switch",
- startingHeight: 9999999,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- // add P1
- sStatusEv("waitForPeer", "waitForBlock", "P1", 20000000, nil),
- // send request for block 9999999
- // Note: For this block request the "switch missing the peer" error is simulated,
- // see implementation of bcReactor interface, sendBlockRequest(), in this file.
- sMakeRequestsEvErrored("waitForBlock", "waitForBlock",
- maxNumRequests, nil, []p2p.ID{"P1"}),
- },
- },
- }
-
- executeFSMTests(t, tests, true)
- }
-
- func TestFSMStopFSM(t *testing.T) {
- tests := []testFields{
- {
- name: "stopFSMEv in unknown",
- steps: []fsmStepTestValues{
- sStopFSMEv("unknown", "finished"),
- },
- },
- {
- name: "stopFSMEv in waitForPeer",
- startingHeight: 1,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStopFSMEv("waitForPeer", "finished"),
- },
- },
- {
- name: "stopFSMEv in waitForBlock",
- startingHeight: 1,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sStopFSMEv("waitForBlock", "finished"),
- },
- },
- }
-
- executeFSMTests(t, tests, false)
- }
-
- func TestFSMUnknownElements(t *testing.T) {
- tests := []testFields{
- {
- name: "unknown event for state unknown",
- steps: []fsmStepTestValues{
- sUnknownFSMEv("unknown"),
- },
- },
- {
- name: "unknown event for state waitForPeer",
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sUnknownFSMEv("waitForPeer"),
- },
- },
- {
- name: "unknown event for state waitForBlock",
- startingHeight: 1,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sUnknownFSMEv("waitForBlock"),
- },
- },
- }
-
- executeFSMTests(t, tests, false)
- }
-
- func TestFSMPeerStateTimeoutEvent(t *testing.T) {
- tests := []testFields{
- {
- name: "timeout event for state waitForPeer while in state waitForPeer - TS1",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStateTimeoutEv("waitForPeer", "finished", "waitForPeer", errNoTallerPeer),
- },
- },
- {
- name: "timeout event for state waitForPeer while in a state != waitForPeer",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStateTimeoutEv("waitForPeer", "waitForPeer", "waitForBlock", errTimeoutEventWrongState),
- },
- },
- {
- name: "timeout event for state waitForBlock while in state waitForBlock ",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sStateTimeoutEv("waitForBlock", "waitForPeer", "waitForBlock", errNoPeerResponseForCurrentHeights),
- },
- },
- {
- name: "timeout event for state waitForBlock while in a state != waitForBlock",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sStateTimeoutEv("waitForBlock", "waitForBlock", "waitForPeer", errTimeoutEventWrongState),
- },
- },
- {
- name: "timeout event for state waitForBlock with multiple peers",
- startingHeight: 1,
- maxRequestsPerPeer: 3,
- steps: []fsmStepTestValues{
- sStartFSMEv(),
- sStatusEv("waitForPeer", "waitForBlock", "P1", 3, nil),
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- sStatusEv("waitForBlock", "waitForBlock", "P2", 3, nil),
- sStateTimeoutEv("waitForBlock", "waitForBlock", "waitForBlock", errNoPeerResponseForCurrentHeights),
- },
- },
- }
-
- executeFSMTests(t, tests, false)
- }
-
- func makeCorrectTransitionSequence(startingHeight int64, numBlocks int64, numPeers int, randomPeerHeights bool,
- maxRequestsPerPeer int, maxPendingRequests int) testFields {
-
- // Generate numPeers peers with random or numBlocks heights according to the randomPeerHeights flag.
- peerHeights := make([]int64, numPeers)
- for i := 0; i < numPeers; i++ {
- if i == 0 {
- peerHeights[0] = numBlocks
- continue
- }
- if randomPeerHeights {
- peerHeights[i] = int64(tmmath.MaxInt(tmrand.Intn(int(numBlocks)), int(startingHeight)+1))
- } else {
- peerHeights[i] = numBlocks
- }
- }
-
- // Approximate the slice capacity to save time for appends.
- testSteps := make([]fsmStepTestValues, 0, 3*numBlocks+int64(numPeers))
-
- testName := fmt.Sprintf("%v-blocks %v-startingHeight %v-peers %v-maxRequestsPerPeer %v-maxNumRequests",
- numBlocks, startingHeight, numPeers, maxRequestsPerPeer, maxPendingRequests)
-
- // Add startFSMEv step.
- testSteps = append(testSteps, sStartFSMEv())
-
- // For each peer, add statusResponseEv step.
- for i := 0; i < numPeers; i++ {
- peerName := fmt.Sprintf("P%d", i)
- if i == 0 {
- testSteps = append(
- testSteps,
- sStatusEv("waitForPeer", "waitForBlock", p2p.ID(peerName), peerHeights[i], nil))
- } else {
- testSteps = append(testSteps,
- sStatusEv("waitForBlock", "waitForBlock", p2p.ID(peerName), peerHeights[i], nil))
- }
- }
-
- height := startingHeight
- numBlocksReceived := 0
- prevBlocks := make([]int64, 0, maxPendingRequests)
-
- forLoop:
- for i := 0; i < int(numBlocks); i++ {
-
- // Add the makeRequestEv step periodically.
- if i%maxRequestsPerPeer == 0 {
- testSteps = append(
- testSteps,
- sMakeRequestsEv("waitForBlock", "waitForBlock", maxNumRequests),
- )
- }
-
- // Add the blockRespEv step
- testSteps = append(
- testSteps,
- sBlockRespEv("waitForBlock", "waitForBlock",
- "P0", height, prevBlocks))
- prevBlocks = append(prevBlocks, height)
- height++
- numBlocksReceived++
-
- // Add the processedBlockEv step periodically.
- if numBlocksReceived >= maxRequestsPerPeer || height >= numBlocks {
- for j := int(height) - numBlocksReceived; j < int(height); j++ {
- if j >= int(numBlocks) {
- // This is the last block that is processed, we should be in "finished" state.
- testSteps = append(
- testSteps,
- sProcessedBlockEv("waitForBlock", "finished", nil))
- break forLoop
- }
- testSteps = append(
- testSteps,
- sProcessedBlockEv("waitForBlock", "waitForBlock", nil))
- }
- numBlocksReceived = 0
- prevBlocks = make([]int64, 0, maxPendingRequests)
- }
- }
-
- return testFields{
- name: testName,
- startingHeight: startingHeight,
- maxRequestsPerPeer: maxRequestsPerPeer,
- maxPendingRequests: maxPendingRequests,
- steps: testSteps,
- }
- }
-
- const (
- maxStartingHeightTest = 100
- maxRequestsPerPeerTest = 20
- maxTotalPendingRequestsTest = 600
- maxNumPeersTest = 1000
- maxNumBlocksInChainTest = 10000 //should be smaller than 9999999
- )
-
- func makeCorrectTransitionSequenceWithRandomParameters() testFields {
- // Generate a starting height for fast sync.
- startingHeight := int64(tmrand.Intn(maxStartingHeightTest) + 1)
-
- // Generate the number of requests per peer.
- maxRequestsPerPeer := tmrand.Intn(maxRequestsPerPeerTest) + 1
-
- // Generate the maximum number of total pending requests, >= maxRequestsPerPeer.
- maxPendingRequests := tmrand.Intn(maxTotalPendingRequestsTest-maxRequestsPerPeer) + maxRequestsPerPeer
-
- // Generate the number of blocks to be synced.
- numBlocks := int64(tmrand.Intn(maxNumBlocksInChainTest)) + startingHeight
-
- // Generate a number of peers.
- numPeers := tmrand.Intn(maxNumPeersTest) + 1
-
- return makeCorrectTransitionSequence(startingHeight, numBlocks, numPeers, true, maxRequestsPerPeer, maxPendingRequests)
- }
-
- func shouldApplyProcessedBlockEvStep(step *fsmStepTestValues, testBcR *testReactor) bool {
- if step.event == processedBlockEv {
- _, err := testBcR.fsm.pool.BlockAndPeerAtHeight(testBcR.fsm.pool.Height)
- if err == errMissingBlock {
- return false
- }
- _, err = testBcR.fsm.pool.BlockAndPeerAtHeight(testBcR.fsm.pool.Height + 1)
- if err == errMissingBlock {
- return false
- }
- }
- return true
- }
-
- func TestFSMCorrectTransitionSequences(t *testing.T) {
-
- tests := []testFields{
- makeCorrectTransitionSequence(1, 100, 10, true, 10, 40),
- makeCorrectTransitionSequenceWithRandomParameters(),
- }
-
- for _, tt := range tests {
- tt := tt
- t.Run(tt.name, func(t *testing.T) {
- // Create test reactor
- testBcR := newTestReactor(tt.startingHeight)
-
- if tt.maxRequestsPerPeer != 0 {
- maxRequestsPerPeer = tt.maxRequestsPerPeer
- }
-
- for _, step := range tt.steps {
- step := step
- assert.Equal(t, step.currentState, testBcR.fsm.state.name)
-
- oldNumStatusRequests := testBcR.numStatusRequests
- fixBlockResponseEvStep(&step, testBcR)
- if !shouldApplyProcessedBlockEvStep(&step, testBcR) {
- continue
- }
-
- fsmErr := sendEventToFSM(testBcR.fsm, step.event, step.data)
- assert.Equal(t, step.wantErr, fsmErr)
-
- if step.wantStatusReqSent {
- assert.Equal(t, oldNumStatusRequests+1, testBcR.numStatusRequests)
- } else {
- assert.Equal(t, oldNumStatusRequests, testBcR.numStatusRequests)
- }
-
- assert.Equal(t, step.wantState, testBcR.fsm.state.name)
- if step.wantState == "finished" {
- assert.True(t, testBcR.fsm.isCaughtUp())
- }
- }
-
- })
- }
- }
-
- // ----------------------------------------
- // implements the bcRNotifier
- func (testR *testReactor) sendPeerError(err error, peerID p2p.ID) {
- testR.logger.Info("Reactor received sendPeerError call from FSM", "peer", peerID, "err", err)
- testR.lastPeerError.peerID = peerID
- testR.lastPeerError.err = err
- }
-
- func (testR *testReactor) sendStatusRequest() {
- testR.logger.Info("Reactor received sendStatusRequest call from FSM")
- testR.numStatusRequests++
- }
-
- func (testR *testReactor) sendBlockRequest(peerID p2p.ID, height int64) error {
- testR.logger.Info("Reactor received sendBlockRequest call from FSM", "peer", peerID, "height", height)
- testR.numBlockRequests++
- testR.lastBlockRequest.peerID = peerID
- testR.lastBlockRequest.height = height
- if height == 9999999 {
- // simulate switch does not have peer
- return errNilPeerForBlockRequest
- }
- return nil
- }
-
- func (testR *testReactor) resetStateTimer(name string, timer **time.Timer, timeout time.Duration) {
- testR.logger.Info("Reactor received resetStateTimer call from FSM", "state", name, "timeout", timeout)
- if _, ok := testR.stateTimerStarts[name]; !ok {
- testR.stateTimerStarts[name] = 1
- } else {
- testR.stateTimerStarts[name]++
- }
- }
-
- func (testR *testReactor) switchToConsensus() {
- }
-
- // ----------------------------------------
|