fluencelabs/go-libp2p-kad-dht
1
0
Fork 0
mirror of https://github.com/fluencelabs/go-libp2p-kad-dht synced 2025-04-24 22:32:13 +00:00
Code Issues Projects Releases Wiki Activity

don't assume that routes are removed immediately after close

Browse Source 
We may fire notifications asynchronously. Well, we have to, really. We'd like:

1. Begin Open
2. Open Notifications
3. End Open
4. Begin Close
5. Close Notifications
6. End Close

However, this breaks if we allow open notifications to close connections (which,
IMO, we want).
This commit is contained in:
Steven Allen 2018-02-16 18:55:42 -08:00
parent b02b9245be
commit 6ec48e921a
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
notify_test.go
View File

@ -2,7 +2,11 @@ package dht
import (
"context"
"fmt"
"testing"
"time"
tu "github.com/libp2p/go-testutil"
)
func TestNotifieeMultipleConn(t *testing.T) {
@ -40,13 +44,16 @@ func TestNotifieeMultipleConn(t *testing.T) {
conn.Close()
}
if checkRoutingTable(d1, d2) {
t.Fatal("routes")
}
tu.WaitFor(ctx, func() error {
if checkRoutingTable(d1, d2) {
return fmt.Errorf("should not have routes")
}
return nil
})
}
func TestNotifieeFuzz(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
defer cancel()
d1 := setupDHT(ctx, t, false)
@ -58,9 +65,12 @@ func TestNotifieeFuzz(t *testing.T) {
conn.Close()
}
}
if checkRoutingTable(d1, d2) {
t.Fatal("should not have routes")
}
tu.WaitFor(ctx, func() error {
if checkRoutingTable(d1, d2) {
return fmt.Errorf("should not have routes")
}
return nil
})
connect(t, ctx, d1, d2)
}