test: make sure to wait for the stream sub-tests to finish

test/stream.go

@@ -221,20 +221,35 @@ func SubtestStress(t *testing.T, ta, tb tpt.Transport, maddr ma.Multiaddr, peerA
 		c, err := tb.Dial(context.Background(), l.Multiaddr(), peerA)
 		checkErr(t, err)
 
+		var wg sync.WaitGroup
+
 		// serve the outgoing conn, because some muxers assume
 		// that we _always_ call serve. (this is an error?)
+		wg.Add(1)
 		go func() {
+			defer wg.Done()
+
 			debugLog(t, "serving connection")
-			for {
+			for i := 0; i < opt.streamNum; i++ {
 				str, err := c.AcceptStream()
 				if err != nil {
+					t.Error(err)
 					break
 				}
-				go echoStream(t, str)
+
+				wg.Add(1)
+				go func() {
+					defer wg.Done()
+					echoStream(t, str)
+				}()
+			}
+			s, err := c.AcceptStream()
+			if err == nil {
+				s.Reset()
+				t.Error("got too many streams")
 			}
 		}()
 
-		var wg sync.WaitGroup
 		for i := 0; i < opt.streamNum; i++ {
 			wg.Add(1)
 			go rateLimit(func() {