ceremonyclient/go-libp2p-blossomsub/gossip_tracer_test.go
2023-07-07 01:07:10 -05:00

104 lines
2.4 KiB
Go

package blossomsub
import (
"testing"
"time"
pb "source.quilibrium.com/quilibrium/monorepo/go-libp2p-blossomsub/pb"
"github.com/libp2p/go-libp2p/core/peer"
)
func TestBrokenPromises(t *testing.T) {
// tests that unfullfilled promises are tracked correctly
gt := newGossipTracer()
gt.followUpTime = 100 * time.Millisecond
peerA := peer.ID("A")
peerB := peer.ID("B")
peerC := peer.ID("C")
var mids []string
for i := 0; i < 100; i++ {
m := makeTestMessage(i)
m.From = []byte(peerA)
mid := DefaultMsgIdFn(m)
mids = append(mids, mid)
}
gt.AddPromise(peerA, mids)
gt.AddPromise(peerB, mids)
gt.AddPromise(peerC, mids)
// no broken promises yet
brokenPromises := gt.GetBrokenPromises()
if brokenPromises != nil {
t.Fatal("expected no broken promises")
}
// throttle one of the peers to save his promises
gt.ThrottlePeer(peerC)
// make promises break
time.Sleep(gt.followUpTime + time.Millisecond)
brokenPromises = gt.GetBrokenPromises()
if len(brokenPromises) != 2 {
t.Fatalf("expected 2 broken prmises, got %d", len(brokenPromises))
}
brokenPromisesA := brokenPromises[peerA]
if brokenPromisesA != 1 {
t.Fatalf("expected 1 broken promise from A, got %d", brokenPromisesA)
}
brokenPromisesB := brokenPromises[peerB]
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
gt := newGossipTracer()
gt.followUpTime = 100 * time.Millisecond
peerA := peer.ID("A")
peerB := peer.ID("B")
var msgs []*pb.Message
var mids []string
for i := 0; i < 100; i++ {
m := makeTestMessage(i)
m.From = []byte(peerA)
msgs = append(msgs, m)
mid := DefaultMsgIdFn(m)
mids = append(mids, mid)
}
gt.AddPromise(peerA, mids)
gt.AddPromise(peerB, mids)
for _, m := range msgs {
gt.DeliverMessage(&Message{Message: m})
}
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")
}
}