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

168 lines
3.2 KiB
Go

package blossomsub
import (
"encoding/binary"
"fmt"
"testing"
pb "source.quilibrium.com/quilibrium/monorepo/go-libp2p-blossomsub/pb"
)
func TestMessageCache(t *testing.T) {
mcache := NewMessageCache(3, 5)
msgID := DefaultMsgIdFn
msgs := make([]*pb.Message, 60)
for i := range msgs {
msgs[i] = makeTestMessage(i)
}
for i := 0; i < 10; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
for i := 0; i < 10; i++ {
mid := msgID(msgs[i])
m, ok := mcache.Get(mid)
if !ok {
t.Fatalf("Message %d not in cache", i)
}
if m.Message != msgs[i] {
t.Fatalf("Message %d does not match cache", i)
}
}
gids := mcache.GetGossipIDs([]byte{0x7e, 0x57})
if len(gids) != 10 {
t.Fatalf("Expected 10 gossip IDs; got %d", len(gids))
}
for i := 0; i < 10; i++ {
mid := msgID(msgs[i])
if mid != gids[i] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
mcache.Shift()
for i := 10; i < 20; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
for i := 0; i < 20; i++ {
mid := msgID(msgs[i])
m, ok := mcache.Get(mid)
if !ok {
t.Fatalf("Message %d not in cache", i)
}
if m.Message != msgs[i] {
t.Fatalf("Message %d does not match cache", i)
}
}
gids = mcache.GetGossipIDs([]byte{0x7e, 0x57})
if len(gids) != 20 {
t.Fatalf("Expected 20 gossip IDs; got %d", len(gids))
}
for i := 0; i < 10; i++ {
mid := msgID(msgs[i])
if mid != gids[10+i] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
for i := 10; i < 20; i++ {
mid := msgID(msgs[i])
if mid != gids[i-10] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
mcache.Shift()
for i := 20; i < 30; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
mcache.Shift()
for i := 30; i < 40; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
mcache.Shift()
for i := 40; i < 50; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
mcache.Shift()
for i := 50; i < 60; i++ {
mcache.Put(&Message{Message: msgs[i]})
}
if len(mcache.msgs) != 50 {
t.Fatalf("Expected 50 messages in the cache; got %d", len(mcache.msgs))
}
for i := 0; i < 10; i++ {
mid := msgID(msgs[i])
_, ok := mcache.Get(mid)
if ok {
t.Fatalf("Message %d still in cache", i)
}
}
for i := 10; i < 60; i++ {
mid := msgID(msgs[i])
m, ok := mcache.Get(mid)
if !ok {
t.Fatalf("Message %d not in cache", i)
}
if m.Message != msgs[i] {
t.Fatalf("Message %d does not match cache", i)
}
}
gids = mcache.GetGossipIDs([]byte{0x7e, 0x57})
if len(gids) != 30 {
t.Fatalf("Expected 30 gossip IDs; got %d", len(gids))
}
for i := 0; i < 10; i++ {
mid := msgID(msgs[50+i])
if mid != gids[i] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
for i := 10; i < 20; i++ {
mid := msgID(msgs[30+i])
if mid != gids[i] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
for i := 20; i < 30; i++ {
mid := msgID(msgs[10+i])
if mid != gids[i] {
t.Fatalf("GossipID mismatch for message %d", i)
}
}
}
func makeTestMessage(n int) *pb.Message {
seqno := make([]byte, 8)
binary.BigEndian.PutUint64(seqno, uint64(n))
data := []byte(fmt.Sprintf("%d", n))
bitmask := []byte{0x7e, 0x57}
return &pb.Message{
Data: data,
Bitmask: bitmask,
From: []byte([]byte{0x7e, 0x57}),
Seqno: seqno,
}
}