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

make randomsub test reliable

Browse Source
This commit is contained in:
vyzo 2020-04-23 20:35:14 +03:00
parent c353348592
commit 2c7e59ea03
1 changed files with 35 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
randomsub_test.go
View File

@ -25,6 +25,17 @@ func getRandomsubs(ctx context.Context, hs []host.Host, size int, opts ...Option
return psubs return psubs
} }
func tryReceive(sub *Subscription) *Message {
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
defer cancel()
m, err := sub.Next(ctx)
if err != nil {
return nil
} else {
return m
}
}
func TestRandomsubSmall(t *testing.T) { func TestRandomsubSmall(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
@ -45,14 +56,21 @@ func TestRandomsubSmall(t *testing.T) {
time.Sleep(time.Second) time.Sleep(time.Second)
count := 0
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
msg := []byte(fmt.Sprintf("message %d", i)) msg := []byte(fmt.Sprintf("message %d", i))
psubs[i].Publish("test", msg) psubs[i].Publish("test", msg)
for _, sub := range subs { for _, sub := range subs {
assertReceive(t, sub, msg) if tryReceive(sub) != nil {
count++
}
} }
} }
if count < 9*len(hosts) {
t.Fatalf("received too few messages; expected at least %d but got %d", 9*len(hosts), count)
}
} }
func TestRandomsubBig(t *testing.T) { func TestRandomsubBig(t *testing.T) {
@ -75,14 +93,21 @@ func TestRandomsubBig(t *testing.T) {
time.Sleep(time.Second) time.Sleep(time.Second)
count := 0
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
msg := []byte(fmt.Sprintf("message %d", i)) msg := []byte(fmt.Sprintf("message %d", i))
psubs[i].Publish("test", msg) psubs[i].Publish("test", msg)
for _, sub := range subs { for _, sub := range subs {
assertReceive(t, sub, msg) if tryReceive(sub) != nil {
count++
}
} }
} }
if count < 9*len(hosts) {
t.Fatalf("received too few messages; expected at least %d but got %d", 9*len(hosts), count)
}
} }
func TestRandomsubMixed(t *testing.T) { func TestRandomsubMixed(t *testing.T) {
@ -107,14 +132,21 @@ func TestRandomsubMixed(t *testing.T) {
time.Sleep(time.Second) time.Sleep(time.Second)
count := 0
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
msg := []byte(fmt.Sprintf("message %d", i)) msg := []byte(fmt.Sprintf("message %d", i))
psubs[i].Publish("test", msg) psubs[i].Publish("test", msg)
for _, sub := range subs { for _, sub := range subs {
assertReceive(t, sub, msg) if tryReceive(sub) != nil {
count++
}
} }
} }
if count < 9*len(hosts) {
t.Fatalf("received too few messages; expected at least %d but got %d", 9*len(hosts), count)
}
} }
func TestRandomsubEnoughPeers(t *testing.T) { func TestRandomsubEnoughPeers(t *testing.T) {