2018-03-04 05:35:18 +00:00
|
|
|
package libp2p
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"fmt"
|
2018-03-10 01:56:02 +00:00
|
|
|
"strings"
|
2018-03-04 05:35:18 +00:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
crypto "github.com/libp2p/go-libp2p-crypto"
|
|
|
|
host "github.com/libp2p/go-libp2p-host"
|
2018-07-25 19:38:20 +00:00
|
|
|
"github.com/libp2p/go-tcp-transport"
|
2018-03-04 05:35:18 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestNewHost(t *testing.T) {
|
2018-03-10 01:56:02 +00:00
|
|
|
h, err := makeRandomHost(t, 9000)
|
2018-03-04 05:35:18 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
2018-03-10 01:56:02 +00:00
|
|
|
h.Close()
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestBadTransportConstructor(t *testing.T) {
|
|
|
|
ctx := context.Background()
|
|
|
|
h, err := New(ctx, Transport(func() {}))
|
|
|
|
if err == nil {
|
|
|
|
h.Close()
|
|
|
|
t.Fatal("expected an error")
|
|
|
|
}
|
|
|
|
if !strings.Contains(err.Error(), "libp2p_test.go") {
|
|
|
|
t.Error("expected error to contain debugging info")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestInsecure(t *testing.T) {
|
|
|
|
ctx := context.Background()
|
|
|
|
h, err := New(ctx, NoSecurity)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
h.Close()
|
2018-03-04 05:35:18 +00:00
|
|
|
}
|
|
|
|
|
2018-07-25 19:38:20 +00:00
|
|
|
func TestDefaultListenAddrs(t *testing.T) {
|
|
|
|
ctx := context.Background()
|
|
|
|
|
|
|
|
// Test 1: Listen addr should not set if user defined transport is passed.
|
|
|
|
h, err := New(
|
|
|
|
ctx,
|
2018-07-25 19:44:09 +00:00
|
|
|
Transport(tcp.NewTCPTransport),
|
2018-07-25 19:38:20 +00:00
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(h.Addrs()) != 0 {
|
|
|
|
t.Error("expected zero listen addrs as none is set with user defined transport")
|
|
|
|
}
|
|
|
|
h.Close()
|
|
|
|
|
|
|
|
// Test 2: User defined listener addrs should overwrite the default options.
|
|
|
|
h, err = New(
|
|
|
|
ctx,
|
2018-07-25 19:44:09 +00:00
|
|
|
Transport(tcp.NewTCPTransport),
|
2018-07-25 19:38:20 +00:00
|
|
|
ListenAddrStrings("/ip4/127.0.0.1/tcp/0"),
|
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(h.Addrs()) != 1 {
|
|
|
|
t.Error("expected one listen addr")
|
|
|
|
}
|
|
|
|
h.Close()
|
|
|
|
}
|
|
|
|
|
2018-03-04 05:35:18 +00:00
|
|
|
func makeRandomHost(t *testing.T, port int) (host.Host, error) {
|
|
|
|
ctx := context.Background()
|
|
|
|
priv, _, err := crypto.GenerateKeyPair(crypto.RSA, 2048)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
opts := []Option{
|
|
|
|
ListenAddrStrings(fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", port)),
|
|
|
|
Identity(priv),
|
2018-03-10 01:56:02 +00:00
|
|
|
DefaultTransports,
|
|
|
|
DefaultMuxers,
|
|
|
|
DefaultSecurity,
|
2018-03-04 05:35:18 +00:00
|
|
|
NATPortMap(),
|
|
|
|
}
|
|
|
|
|
|
|
|
return New(ctx, opts...)
|
|
|
|
}
|