package behaviour_test import ( "sync" "testing" bh "github.com/tendermint/tendermint/behaviour" "github.com/tendermint/tendermint/p2p" ) // TestMockReporter tests the MockReporter's ability to store reported // peer behaviour in memory indexed by the peerID. func TestMockReporter(t *testing.T) { var peerID p2p.ID = "MockPeer" pr := bh.NewMockReporter() behaviours := pr.GetBehaviours(peerID) if len(behaviours) != 0 { t.Error("Expected to have no behaviours reported") } badMessage := bh.BadMessage(peerID, "bad message") pr.Report(badMessage) behaviours = pr.GetBehaviours(peerID) if len(behaviours) != 1 { t.Error("Expected the peer have one reported behaviour") } if behaviours[0] != badMessage { t.Error("Expected Bad Message to have been reported") } } type scriptItem struct { peerID p2p.ID behaviour bh.PeerBehaviour } // equalBehaviours returns true if a and b contain the same PeerBehaviours with // the same freequencies and otherwise false. func equalBehaviours(a []bh.PeerBehaviour, b []bh.PeerBehaviour) bool { aHistogram := map[bh.PeerBehaviour]int{} bHistogram := map[bh.PeerBehaviour]int{} for _, behaviour := range a { aHistogram[behaviour]++ } for _, behaviour := range b { bHistogram[behaviour]++ } if len(aHistogram) != len(bHistogram) { return false } for _, behaviour := range a { if aHistogram[behaviour] != bHistogram[behaviour] { return false } } for _, behaviour := range b { if bHistogram[behaviour] != aHistogram[behaviour] { return false } } return true } // TestEqualPeerBehaviours tests that equalBehaviours can tell that two slices // of peer behaviours can be compared for the behaviours they contain and the // freequencies that those behaviours occur. func TestEqualPeerBehaviours(t *testing.T) { var ( peerID p2p.ID = "MockPeer" consensusVote = bh.ConsensusVote(peerID, "voted") blockPart = bh.BlockPart(peerID, "blocked") equals = []struct { left []bh.PeerBehaviour right []bh.PeerBehaviour }{ // Empty sets {[]bh.PeerBehaviour{}, []bh.PeerBehaviour{}}, // Single behaviours {[]bh.PeerBehaviour{consensusVote}, []bh.PeerBehaviour{consensusVote}}, // Equal Frequencies {[]bh.PeerBehaviour{consensusVote, consensusVote}, []bh.PeerBehaviour{consensusVote, consensusVote}}, // Equal frequencies different orders {[]bh.PeerBehaviour{consensusVote, blockPart}, []bh.PeerBehaviour{blockPart, consensusVote}}, } unequals = []struct { left []bh.PeerBehaviour right []bh.PeerBehaviour }{ // Comparing empty sets to non empty sets {[]bh.PeerBehaviour{}, []bh.PeerBehaviour{consensusVote}}, // Different behaviours {[]bh.PeerBehaviour{consensusVote}, []bh.PeerBehaviour{blockPart}}, // Same behaviour with different frequencies {[]bh.PeerBehaviour{consensusVote}, []bh.PeerBehaviour{consensusVote, consensusVote}}, } ) for _, test := range equals { if !equalBehaviours(test.left, test.right) { t.Errorf("expected %#v and %#v to be equal", test.left, test.right) } } for _, test := range unequals { if equalBehaviours(test.left, test.right) { t.Errorf("expected %#v and %#v to be unequal", test.left, test.right) } } } // TestPeerBehaviourConcurrency constructs a scenario in which // multiple goroutines are using the same MockReporter instance. // This test reproduces the conditions in which MockReporter will // be used within a Reactor `Receive` method tests to ensure thread safety. func TestMockPeerBehaviourReporterConcurrency(t *testing.T) { var ( behaviourScript = []struct { peerID p2p.ID behaviours []bh.PeerBehaviour }{ {"1", []bh.PeerBehaviour{bh.ConsensusVote("1", "")}}, {"2", []bh.PeerBehaviour{bh.ConsensusVote("2", ""), bh.ConsensusVote("2", ""), bh.ConsensusVote("2", "")}}, { "3", []bh.PeerBehaviour{bh.BlockPart("3", ""), bh.ConsensusVote("3", ""), bh.BlockPart("3", ""), bh.ConsensusVote("3", "")}}, { "4", []bh.PeerBehaviour{bh.ConsensusVote("4", ""), bh.ConsensusVote("4", ""), bh.ConsensusVote("4", ""), bh.ConsensusVote("4", "")}}, { "5", []bh.PeerBehaviour{bh.BlockPart("5", ""), bh.ConsensusVote("5", ""), bh.BlockPart("5", ""), bh.ConsensusVote("5", "")}}, } ) var receiveWg sync.WaitGroup pr := bh.NewMockReporter() scriptItems := make(chan scriptItem) done := make(chan int) numConsumers := 3 for i := 0; i < numConsumers; i++ { receiveWg.Add(1) go func() { defer receiveWg.Done() for { select { case pb := <-scriptItems: pr.Report(pb.behaviour) case <-done: return } } }() } var sendingWg sync.WaitGroup sendingWg.Add(1) go func() { defer sendingWg.Done() for _, item := range behaviourScript { for _, reason := range item.behaviours { scriptItems <- scriptItem{item.peerID, reason} } } }() sendingWg.Wait() for i := 0; i < numConsumers; i++ { done <- 1 } receiveWg.Wait() for _, items := range behaviourScript { reported := pr.GetBehaviours(items.peerID) if !equalBehaviours(reported, items.behaviours) { t.Errorf("expected peer %s to have behaved \nExpected: %#v \nGot %#v \n", items.peerID, items.behaviours, reported) } } }