84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package libp2pwebrtcdirect
|
|
|
|
import (
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
|
|
logging "github.com/ipfs/go-log"
|
|
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
tpt "github.com/libp2p/go-libp2p-transport"
|
|
utils "github.com/libp2p/go-libp2p-transport/test"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/pions/webrtc"
|
|
mplex "github.com/whyrusleeping/go-smux-multiplex"
|
|
)
|
|
|
|
var Subtests = []func(t *testing.T, ta, tb tpt.Transport, maddr ma.Multiaddr, peerA peer.ID){
|
|
utils.SubtestProtocols,
|
|
utils.SubtestBasic,
|
|
|
|
utils.SubtestCancel,
|
|
// utils.SubtestPingPong, // TODO: ICE state becomes disconnected
|
|
|
|
// Stolen from the stream muxer test suite.
|
|
utils.SubtestStress1Conn1Stream1Msg,
|
|
utils.SubtestStress1Conn1Stream100Msg,
|
|
// utils.SubtestStress1Conn100Stream100Msg, // TODO: ICE state becomes disconnected
|
|
// utils.SubtestStress50Conn10Stream50Msg, // TODO
|
|
// utils.SubtestStress1Conn1000Stream10Msg, // TODO
|
|
// utils.SubtestStress1Conn100Stream100Msg10MB, // TODO
|
|
// utils.SubtestStreamOpenStress, // panic: Fail in goroutine after TestTransport has completed
|
|
utils.SubtestStreamReset,
|
|
}
|
|
|
|
func TestTransport(t *testing.T) {
|
|
logging.SetLogLevel("*", "warning")
|
|
|
|
ta := NewTransport(
|
|
webrtc.Configuration{},
|
|
new(mplex.Transport),
|
|
)
|
|
tb := NewTransport(
|
|
webrtc.Configuration{},
|
|
new(mplex.Transport),
|
|
)
|
|
|
|
addr := "/ip4/127.0.0.1/tcp/0/http/p2p-webrtc-direct"
|
|
SubtestTransport(t, ta, tb, addr, "peerA")
|
|
}
|
|
|
|
func getFunctionName(i interface{}) string {
|
|
return runtime.FuncForPC(reflect.ValueOf(i).Pointer()).Name()
|
|
}
|
|
|
|
func SubtestTransport(t *testing.T, ta, tb tpt.Transport, addr string, peerA peer.ID) {
|
|
maddr, err := ma.NewMultiaddr(addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, f := range Subtests {
|
|
t.Run(getFunctionName(f), func(t *testing.T) {
|
|
f(t, ta, tb, maddr, peerA)
|
|
})
|
|
}
|
|
}
|
|
func TestTransportCantListenUtp(t *testing.T) {
|
|
utpa, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/50000")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tpt := NewTransport(
|
|
webrtc.Configuration{},
|
|
new(mplex.Transport),
|
|
)
|
|
|
|
_, err = tpt.Listen(utpa)
|
|
if err == nil {
|
|
t.Fatal("shouldnt be able to listen on utp addr with tcp transport")
|
|
}
|
|
|
|
}
|