ceremonyclient/go-libp2p/p2p/host/routed/routed_test.go
2023-08-20 23:07:43 -05:00

92 lines
2.9 KiB
Go

package routedhost
import (
"context"
"testing"
"github.com/libp2p/go-libp2p/core/peer"
basic "github.com/libp2p/go-libp2p/p2p/host/basic"
swarmt "github.com/libp2p/go-libp2p/p2p/net/swarm/testing"
ma "github.com/multiformats/go-multiaddr"
"github.com/stretchr/testify/require"
)
var _ Routing = (*mockRouting)(nil)
type mockRouting struct {
callCount int
findPeerFn func(ctx context.Context, id peer.ID) (peer.AddrInfo, error)
}
func (m *mockRouting) FindPeer(ctx context.Context, pid peer.ID) (peer.AddrInfo, error) {
m.callCount++
return m.findPeerFn(ctx, pid)
}
func TestRoutedHostConnectToObsoleteAddresses(t *testing.T) {
h1, err := basic.NewHost(swarmt.GenSwarm(t), nil)
require.NoError(t, err)
defer h1.Close()
h1.Start()
h2, err := basic.NewHost(swarmt.GenSwarm(t), nil)
require.NoError(t, err)
defer h2.Close()
// assemble the AddrInfo struct to use for the connection attempt
pi := peer.AddrInfo{
ID: h2.ID(),
// Use a wrong multi address for host 2, so that the initial connection attempt will fail
// (we have obsolete, old multi address information)
Addrs: []ma.Multiaddr{ma.StringCast("/ip4/127.0.0.1/tcp/1234")},
}
// Build mock routing module and replace the FindPeer function.
// Now, that function will return the correct multi addresses for host 2
// (we have fetched the most up-to-date data from the DHT)
mr := &mockRouting{
findPeerFn: func(context.Context, peer.ID) (peer.AddrInfo, error) {
return peer.AddrInfo{
ID: h2.ID(),
Addrs: h2.Addrs(),
}, nil
},
}
// Build routed host
rh := Wrap(h1, mr)
// Connection establishment should have worked without an error
require.NoError(t, rh.Connect(context.Background(), pi))
require.Equal(t, 1, mr.callCount, "the mocked FindPeer function should have been called")
}
func TestRoutedHostConnectFindPeerNoUsefulAddrs(t *testing.T) {
h1, err := basic.NewHost(swarmt.GenSwarm(t), nil)
require.NoError(t, err)
defer h1.Close()
h2, err := basic.NewHost(swarmt.GenSwarm(t), nil)
require.NoError(t, err)
defer h2.Close()
// assemble the AddrInfo struct to use for the connection attempt
pi := peer.AddrInfo{
ID: h2.ID(),
// Use a wrong multi address for host 2, so that the initial connection attempt will fail
// (we have obsolete, old multi address information)
Addrs: []ma.Multiaddr{ma.StringCast("/ip4/127.0.0.1/tcp/1234")},
}
// Build mock routing module and replace the FindPeer function.
// Now, that function will return the correct multi addresses for host 2
// (we have fetched the most up-to-date data from the DHT)
mr := &mockRouting{findPeerFn: func(context.Context, peer.ID) (peer.AddrInfo, error) { return pi, nil }}
// Build routed host
rh := Wrap(h1, mr)
// Connection establishment should fail, since we didn't provide any useful addresses in FindPeer.
require.Error(t, rh.Connect(context.Background(), pi))
require.Equal(t, 1, mr.callCount, "the mocked FindPeer function should have been called")
}