fix(tests): minor fix for waitgroup

session/session_test.go

@@ -185,7 +185,7 @@ func TestSessionFindMorePeers(t *testing.T) {
 	<-wantReqs
 	<-cancelReqs
 
-	// wait long enough for a tick to occur
+	// wait for a request to get more peers to occur
 	<-fpm.findMorePeersRequested
 
 	// verify a broadcast was made

sessionpeermanager/sessionpeermanager_test.go

@@ -49,7 +49,7 @@ func (fcm *fakeConnManager) TagPeer(p peer.ID, tag string, n int) {
 }
 
 func (fcm *fakeConnManager) UntagPeer(p peer.ID, tag string) {
-	fcm.wait.Done()
+	defer fcm.wait.Done()
 
 	for i := 0; i < len(fcm.taggedPeers); i++ {
 		if fcm.taggedPeers[i] == p {