p2p: fix switch test for Broadcast returning success channel

p2p/switch.go

@@ -166,7 +166,7 @@ func (sw *Switch) IsDialing(addr *NetAddress) bool {
 // which receives success values for each attempted send (false if times out)
 func (sw *Switch) Broadcast(chId byte, msg interface{}) chan bool {
 	if atomic.LoadUint32(&sw.stopped) == 1 {
-		return
+		return nil
 	}
 	successChan := make(chan bool, len(sw.peers.List()))
 	log.Debug("Broadcast", "channel", chId, "msg", msg)

p2p/switch_test.go

@@ -197,9 +197,14 @@ func BenchmarkSwitches(b *testing.B) {
 	// Send random message from one channel to another
 	for i := 0; i < b.N; i++ {
 		chId := byte(i % 4)
-		nS, nF := s1.Broadcast(chId, "test data")
-		numSuccess += nS
-		numFailure += nF
+		successChan := s1.Broadcast(chId, "test data")
+		for s := range successChan {
+			if s {
+				numSuccess += 1
+			} else {
+				numFailure += 1
+			}
+		}
 	}
 
 	log.Warn(Fmt("success: %v, failure: %v", numSuccess, numFailure))