Merge pull request #1149 from libp2p/fix/deflake-echo-test

fix: deflake multipro echo test
libp2p · Aug 3, 2021 · 159a6fc · 159a6fc
2 parents aaef2e0 + 8217b1f
commit 159a6fc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/examples/multipro/echo.go b/examples/multipro/echo.go
@@ -86,6 +86,7 @@ func (e *EchoProtocol) onEchoRequest(s network.Stream) {
 	if ok {
 		log.Printf("%s: Echo response to %s sent.", s.Conn().LocalPeer().String(), s.Conn().RemotePeer().String())
 	}
+	e.done <- true
 }
 
 // remote echo response handler

diff --git a/examples/multipro/main.go b/examples/multipro/main.go
@@ -57,7 +57,7 @@ func run(h1, h2 *Node, done <-chan bool) {
 	h2.Echo(h1.Host)
 
 	// block until all responses have been processed
-	for i := 0; i < 4; i++ {
+	for i := 0; i < 8; i++ {
 		<-done
 	}
 }
diff --git a/examples/multipro/ping.go b/examples/multipro/ping.go
@@ -82,6 +82,7 @@ func (p *PingProtocol) onPingRequest(s network.Stream) {
 	if ok {
 		log.Printf("%s: Ping response to %s sent.", s.Conn().LocalPeer().String(), s.Conn().RemotePeer().String())
 	}
+	p.done <- true
 }
 
 // remote ping response handler