logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-03 21:33:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add tests for clearing the peerPromises map

Browse Source
This commit is contained in:
vyzo 2022-01-13 13:20:32 +02:00
parent 9e39259ffe
commit b57bcc8b36
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
gossip_tracer_test.go
View File

@ -40,7 +40,7 @@ func TestBrokenPromises(t *testing.T) {
gt.ThrottlePeer(peerC)
// make promises break
time.Sleep(GossipSubIWantFollowupTime + 10*time.Millisecond)
time.Sleep(gt.followUpTime + time.Millisecond)
brokenPromises = gt.GetBrokenPromises()
if len(brokenPromises) != 2 {
@ -56,17 +56,17 @@ func TestBrokenPromises(t *testing.T) {
if brokenPromisesB != 1 {
t.Fatalf("expected 1 broken promise from A, got %d", brokenPromisesB)
}
// verify that the peerPromises map has been vacated
if len(gt.peerPromises) != 0 {
t.Fatal("expected empty peerPromises map")
}
}
func TestNoBrokenPromises(t *testing.T) {
// like above, but this time we deliver messages to fullfil the promises
originalGossipSubIWantFollowupTime := GossipSubIWantFollowupTime
GossipSubIWantFollowupTime = 100 * time.Millisecond
defer func() {
GossipSubIWantFollowupTime = originalGossipSubIWantFollowupTime
}()
gt := newGossipTracer()
gt.followUpTime = 100 * time.Millisecond
peerA := peer.ID("A")
peerB := peer.ID("B")
@ -88,11 +88,16 @@ func TestNoBrokenPromises(t *testing.T) {
gt.DeliverMessage(&Message{Message: m})
}
time.Sleep(GossipSubIWantFollowupTime + 10*time.Millisecond)
time.Sleep(gt.followUpTime + time.Millisecond)
// there should be no broken promises
brokenPromises := gt.GetBrokenPromises()
if brokenPromises != nil {
t.Fatal("expected no broken promises")
}
// verify that the peerPromises map has been vacated
if len(gt.peerPromises) != 0 {
t.Fatal("expected empty peerPromises map")
}
}