discovery_test.go

package p2pnet

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

var testAdvertPropagationDelay = time.Millisecond * 100

func TestHost_Discover(t *testing.T) {
	namespaces := []string{"", "test"}

	ha := newHost(t, basicTestConfig(t, namespaces))
	err := ha.Start()
	require.NoError(t, err)

	hb := newHost(t, basicTestConfig(t, namespaces))
	err = hb.Start()
	require.NoError(t, err)

	hc := newHost(t, basicTestConfig(t, namespaces))
	err = hc.Start()
	require.NoError(t, err)

	// connect a + b and b + c, see if c can discover a via DHT
	err = ha.h.Connect(ha.ctx, hb.AddrInfo())
	require.NoError(t, err)

	err = hc.h.Connect(hc.ctx, hb.AddrInfo())
	require.NoError(t, err)

	// now that the nodes can form a DHT, advertise
	ha.Advertise()
	hb.Advertise()
	hc.Addresses()
	time.Sleep(testAdvertPropagationDelay)

	require.GreaterOrEqual(t, len(ha.h.Network().Peers()), 2)
	require.GreaterOrEqual(t, len(hb.h.Network().Peers()), 2)
	require.GreaterOrEqual(t, len(hc.h.Network().Peers()), 2)

	peerIDs, err := hc.Discover("test", time.Second)
	require.NoError(t, err)
	require.GreaterOrEqual(t, len(peerIDs), 1)
	require.NotEmpty(t, peerIDs[0])
	require.NotEqual(t, peerIDs[0], hc.PeerID()) // should be ha or hb's ID
}