mirror of
https://github.com/libp2p/go-libp2p.git
synced 2025-10-19 22:44:42 +08:00
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package discovery
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
swarmt "github.com/libp2p/go-libp2p-swarm/testing"
|
|
bhost "github.com/libp2p/go-libp2p/p2p/host/basic"
|
|
)
|
|
|
|
type DiscoveryNotifee struct {
|
|
h host.Host
|
|
}
|
|
|
|
func (n *DiscoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
|
|
n.h.Connect(context.Background(), pi)
|
|
}
|
|
|
|
func TestMdnsDiscovery(t *testing.T) {
|
|
//TODO: re-enable when the new lib will get integrated
|
|
t.Skip("TestMdnsDiscovery fails randomly with current lib")
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
a, err := bhost.NewHost(ctx, swarmt.GenSwarm(t, ctx), nil)
|
|
require.NoError(t, err)
|
|
b, err := bhost.NewHost(ctx, swarmt.GenSwarm(t, ctx), nil)
|
|
require.NoError(t, err)
|
|
|
|
sa, err := NewMdnsService(ctx, a, time.Second, "someTag")
|
|
require.NoError(t, err)
|
|
|
|
sb, err := NewMdnsService(ctx, b, time.Second, "someTag")
|
|
require.NoError(t, err)
|
|
_ = sb
|
|
|
|
n := &DiscoveryNotifee{a}
|
|
|
|
sa.RegisterNotifee(n)
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
if err := a.Connect(ctx, peer.AddrInfo{ID: b.ID()}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|