evidence: reactor test

This commit is contained in:
Ethan Buchman 2017-11-20 04:32:53 +00:00
parent 666ae244b3
commit cfbedec719

View File

@ -71,7 +71,7 @@ func (evR *EvidenceReactor) AddPeer(peer p2p.Peer) {
// send the peer our high-priority evidence. // send the peer our high-priority evidence.
// the rest will be sent by the broadcastRoutine // the rest will be sent by the broadcastRoutine
evidence := evR.evpool.PriorityEvidence() evidence := evR.evpool.PriorityEvidence()
msg := EvidenceListMessage{evidence} msg := &EvidenceListMessage{evidence}
success := peer.Send(EvidenceChannel, struct{ EvidenceMessage }{msg}) success := peer.Send(EvidenceChannel, struct{ EvidenceMessage }{msg})
if !success { if !success {
// TODO: remove peer ? // TODO: remove peer ?
@ -120,7 +120,7 @@ func (evR *EvidenceReactor) broadcastRoutine() {
select { select {
case evidence := <-evR.evpool.EvidenceChan(): case evidence := <-evR.evpool.EvidenceChan():
// broadcast some new evidence // broadcast some new evidence
msg := EvidenceListMessage{[]types.Evidence{evidence}} msg := &EvidenceListMessage{[]types.Evidence{evidence}}
evR.Switch.Broadcast(EvidenceChannel, struct{ EvidenceMessage }{msg}) evR.Switch.Broadcast(EvidenceChannel, struct{ EvidenceMessage }{msg})
// TODO: Broadcast runs asynchronously, so this should wait on the successChan // TODO: Broadcast runs asynchronously, so this should wait on the successChan
@ -128,7 +128,7 @@ func (evR *EvidenceReactor) broadcastRoutine() {
evR.evpool.evidenceStore.MarkEvidenceAsBroadcasted(evidence) evR.evpool.evidenceStore.MarkEvidenceAsBroadcasted(evidence)
case <-ticker.C: case <-ticker.C:
// broadcast all pending evidence // broadcast all pending evidence
msg := EvidenceListMessage{evR.evpool.PendingEvidence()} msg := &EvidenceListMessage{evR.evpool.PendingEvidence()}
evR.Switch.Broadcast(EvidenceChannel, struct{ EvidenceMessage }{msg}) evR.Switch.Broadcast(EvidenceChannel, struct{ EvidenceMessage }{msg})
case <-evR.Quit: case <-evR.Quit:
return return