logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-02 12:53:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test races

Browse Source
This commit is contained in:
vyzo 2020-09-03 12:34:20 +03:00
parent be1b155d45
commit 8408e700e2
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
peer_gater_test.go
View File

@ -93,20 +93,29 @@ func TestPeerGater(t *testing.T) {
}
pg.RemovePeer(peerA)
pg.Lock()
_, ok := pg.peerStats[peerA]
pg.Unlock()
if ok {
t.Fatal("still have a stat record for peerA")
}
pg.Lock()
_, ok = pg.ipStats[peerAip]
pg.Unlock()
if !ok {
t.Fatal("expected to still have a stat record for peerA's ip")
}
pg.Lock()
pg.ipStats[peerAip].expire = time.Now()
pg.Unlock()
time.Sleep(2 * time.Second)
pg.Lock()
_, ok = pg.ipStats["1.2.3.4"]
pg.Unlock()
if ok {
t.Fatal("still have a stat record for peerA's ip")
}