mirror of
https://github.com/fluencelabs/go-libp2p-kad-dht
synced 2025-04-24 22:32:13 +00:00
6ec48e921a
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).
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package dht
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
tu "github.com/libp2p/go-testutil"
|
|
)
|
|
|
|
func TestNotifieeMultipleConn(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
d1 := setupDHT(ctx, t, false)
|
|
d2 := setupDHT(ctx, t, false)
|
|
|
|
nn1 := (*netNotifiee)(d1)
|
|
nn2 := (*netNotifiee)(d2)
|
|
|
|
connect(t, ctx, d1, d2)
|
|
c12 := d1.host.Network().ConnsToPeer(d2.self)[0]
|
|
c21 := d2.host.Network().ConnsToPeer(d1.self)[0]
|
|
|
|
// Pretend to reestablish/re-kill connection
|
|
nn1.Connected(d1.host.Network(), c12)
|
|
nn2.Connected(d2.host.Network(), c21)
|
|
|
|
if !checkRoutingTable(d1, d2) {
|
|
t.Fatal("no routes")
|
|
}
|
|
nn1.Disconnected(d1.host.Network(), c12)
|
|
nn2.Disconnected(d2.host.Network(), c21)
|
|
|
|
if !checkRoutingTable(d1, d2) {
|
|
t.Fatal("no routes")
|
|
}
|
|
|
|
for _, conn := range d1.host.Network().ConnsToPeer(d2.self) {
|
|
conn.Close()
|
|
}
|
|
for _, conn := range d2.host.Network().ConnsToPeer(d1.self) {
|
|
conn.Close()
|
|
}
|
|
|
|
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.WithTimeout(context.Background(), time.Second*2)
|
|
defer cancel()
|
|
|
|
d1 := setupDHT(ctx, t, false)
|
|
d2 := setupDHT(ctx, t, false)
|
|
|
|
for i := 0; i < 100; i++ {
|
|
connectNoSync(t, ctx, d1, d2)
|
|
for _, conn := range d1.host.Network().ConnsToPeer(d2.self) {
|
|
conn.Close()
|
|
}
|
|
}
|
|
tu.WaitFor(ctx, func() error {
|
|
if checkRoutingTable(d1, d2) {
|
|
return fmt.Errorf("should not have routes")
|
|
}
|
|
return nil
|
|
})
|
|
connect(t, ctx, d1, d2)
|
|
}
|
|
|
|
func checkRoutingTable(a, b *IpfsDHT) bool {
|
|
// loop until connection notification has been received.
|
|
// under high load, this may not happen as immediately as we would like.
|
|
return a.routingTable.Find(b.self) != "" && b.routingTable.Find(a.self) != ""
|
|
}
|