Commit 6b3042fe authored by hannahhoward's avatar hannahhoward

fix(tests): minor fix for waitgroup

parent 78d4f387
...@@ -185,7 +185,7 @@ func TestSessionFindMorePeers(t *testing.T) { ...@@ -185,7 +185,7 @@ func TestSessionFindMorePeers(t *testing.T) {
<-wantReqs <-wantReqs
<-cancelReqs <-cancelReqs
// wait long enough for a tick to occur // wait for a request to get more peers to occur
<-fpm.findMorePeersRequested <-fpm.findMorePeersRequested
// verify a broadcast was made // verify a broadcast was made
......
...@@ -49,7 +49,7 @@ func (fcm *fakeConnManager) TagPeer(p peer.ID, tag string, n int) { ...@@ -49,7 +49,7 @@ func (fcm *fakeConnManager) TagPeer(p peer.ID, tag string, n int) {
} }
func (fcm *fakeConnManager) UntagPeer(p peer.ID, tag string) { func (fcm *fakeConnManager) UntagPeer(p peer.ID, tag string) {
fcm.wait.Done() defer fcm.wait.Done()
for i := 0; i < len(fcm.taggedPeers); i++ { for i := 0; i < len(fcm.taggedPeers); i++ {
if fcm.taggedPeers[i] == p { if fcm.taggedPeers[i] == p {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment