mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-12-28 09:35:17 +00:00
141 lines
2.8 KiB
Go
141 lines
2.8 KiB
Go
package blossomsub
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
)
|
|
|
|
func TestMapBlacklist(t *testing.T) {
|
|
b := NewMapBlacklist()
|
|
|
|
p := peer.ID("test")
|
|
|
|
b.Add(p)
|
|
if !b.Contains(p) {
|
|
t.Fatal("peer not in the blacklist")
|
|
}
|
|
|
|
}
|
|
|
|
func TestTimeCachedBlacklist(t *testing.T) {
|
|
b, err := NewTimeCachedBlacklist(10 * time.Minute)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
p := peer.ID("test")
|
|
|
|
b.Add(p)
|
|
if !b.Contains(p) {
|
|
t.Fatal("peer not in the blacklist")
|
|
}
|
|
}
|
|
|
|
func TestBlacklist(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
hosts := getDefaultHosts(t, 2)
|
|
psubs := getBlossomSubs(ctx, hosts)
|
|
connect(t, hosts[0], hosts[1])
|
|
|
|
bitmasks, err := psubs[0].Join([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
sub, err := psubs[1].Subscribe([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * 100)
|
|
psubs[1].BlacklistPeer(hosts[0].ID())
|
|
time.Sleep(time.Millisecond * 100)
|
|
|
|
bitmasks[0].Publish(ctx, bitmasks[0].bitmask, []byte("message"))
|
|
|
|
wctx, cancel := context.WithTimeout(ctx, 1*time.Second)
|
|
defer cancel()
|
|
_, err = sub[0].Next(wctx)
|
|
|
|
if err == nil {
|
|
t.Fatal("got message from blacklisted peer")
|
|
}
|
|
}
|
|
|
|
func TestBlacklist2(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
hosts := getDefaultHosts(t, 2)
|
|
psubs := getBlossomSubs(ctx, hosts)
|
|
connect(t, hosts[0], hosts[1])
|
|
|
|
bitmasks, err := psubs[0].Join([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err = psubs[0].Subscribe([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
sub1, err := psubs[1].Subscribe([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * 100)
|
|
psubs[1].BlacklistPeer(hosts[0].ID())
|
|
time.Sleep(time.Millisecond * 100)
|
|
|
|
bitmasks[0].Publish(ctx, bitmasks[0].bitmask, []byte("message"))
|
|
|
|
wctx, cancel := context.WithTimeout(ctx, 1*time.Second)
|
|
defer cancel()
|
|
_, err = sub1[0].Next(wctx)
|
|
|
|
if err == nil {
|
|
t.Fatal("got message from blacklisted peer")
|
|
}
|
|
}
|
|
|
|
func TestBlacklist3(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
hosts := getDefaultHosts(t, 2)
|
|
psubs := getBlossomSubs(ctx, hosts)
|
|
|
|
psubs[1].BlacklistPeer(hosts[0].ID())
|
|
time.Sleep(time.Millisecond * 100)
|
|
connect(t, hosts[0], hosts[1])
|
|
|
|
bitmasks, err := psubs[0].Join([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
sub, err := psubs[1].Subscribe([]byte{0x00, 0x80, 0x00, 0x00})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * 100)
|
|
|
|
bitmasks[0].Publish(ctx, bitmasks[0].bitmask, []byte("message"))
|
|
|
|
wctx, cancel := context.WithTimeout(ctx, 1*time.Second)
|
|
defer cancel()
|
|
_, err = sub[0].Next(wctx)
|
|
|
|
if err == nil {
|
|
t.Fatal("got message from blacklisted peer")
|
|
}
|
|
}
|