Circuit hop tests

Signed-off-by: Gustavo Sampaio <[email protected]>

Author: GustavoKatel

go.mod

@@ -20,6 +20,7 @@ require (
 	github.com/libp2p/go-libp2p-kad-dht v0.1.0
 	github.com/libp2p/go-libp2p-peerstore v0.1.0
 	github.com/libp2p/go-libp2p-record v0.1.0
+	github.com/libp2p/go-libp2p-swarm v0.1.0
 	github.com/multiformats/go-multiaddr v0.0.4
 	github.com/multiformats/go-multihash v0.0.5
 	github.com/stretchr/testify v1.3.0

host.go

@@ -70,7 +70,12 @@ func HostNew(settings *options.Settings, ds ipfsDatastore.Batching, blockstore b
 
 	opts := []libp2p.Option{
 		libp2p.ListenAddrs(sourceMultiAddrIP4, sourceMultiAddrIP6),
-		libp2p.EnableRelay(libp2pCircuit.OptDiscovery),
+	}
+
+	if settings.RelayOnly {
+		opts = append(opts, libp2p.EnableRelay(libp2pCircuit.OptHop))
+	} else {
+		opts = append(opts, libp2p.EnableRelay(libp2pCircuit.OptDiscovery))
 	}
 
 	id, ok := settings.Identity.(*identity.Libp2pIdentity)
@@ -147,6 +152,10 @@ func (host *hostImpl) Close() error {
 		return err
 	}
 
+	if err := host.p2pHost.Network().Close(); err != nil {
+		return err
+	}
+
 	return host.p2pHost.Close()
 }