candidate_relay_test.go

package ice

import (
	"net"
	"testing"
	"time"

	"github.com/pion/logging"
	"github.com/pion/transport/test"
	"github.com/hcm007/turn"
)

func optimisticAuthHandler(username string, srcAddr net.Addr) (password string, ok bool) {
	return "password", true
}

func TestRelayOnlyConnection(t *testing.T) {
	// Limit runtime in case of deadlocks
	lim := test.TimeOut(time.Second * 30)
	defer lim.Stop()

	report := test.CheckRoutines(t)
	defer report()

	loggerFactory := logging.NewDefaultLoggerFactory()

	serverPort := randomPort(t)
	server := turn.NewServer(&turn.ServerConfig{
		Realm:         "pion.ly",
		AuthHandler:   optimisticAuthHandler,
		ListeningPort: serverPort,
		LoggerFactory: loggerFactory,
	})
	err := server.AddListeningIPAddr("127.0.0.1")
	if err != nil {
		t.Fatal(err)
	}

	cfg := &AgentConfig{
		NetworkTypes: supportedNetworkTypes,
		Urls: []*URL{
			{
				Scheme:   SchemeTypeTURN,
				Host:     "localhost",
				Username: "username",
				Password: "password",
				Port:     serverPort,
			},
		},
		CandidateTypes: []CandidateType{CandidateTypeRelay},
	}

	err = server.Start()
	if err != nil {
		t.Fatal(err)
	}

	aAgent, err := NewAgent(cfg)
	if err != nil {
		t.Fatal(err)
	}

	aNotifier, aConnected := onConnected()
	if err = aAgent.OnConnectionStateChange(aNotifier); err != nil {
		t.Fatal(err)
	}

	bAgent, err := NewAgent(cfg)
	if err != nil {
		t.Fatal(err)
	}

	bNotifier, bConnected := onConnected()
	if err = bAgent.OnConnectionStateChange(bNotifier); err != nil {
		t.Fatal(err)
	}

	connect(aAgent, bAgent)
	<-aConnected
	<-bConnected

	if err = aAgent.Close(); err != nil {
		t.Fatal(err)
	}
	if err = bAgent.Close(); err != nil {
		t.Fatal(err)
	}
	if err = server.Close(); err != nil {
		t.Fatal(err)
	}
}