gather_test.go

package ice

import (
	"net"
	"testing"
)

func TestListenUDP(t *testing.T) {
	a, err := NewAgent(&AgentConfig{})
	if err != nil {
		t.Fatalf("Failed to create agent: %s", err)
	}

	localIPs, err := a.localInterfaces([]NetworkType{NetworkTypeUDP4})
	if len(localIPs) == 0 {
		t.Fatal("localInterfaces found no interfaces, unable to test")
	} else if err != nil {
		t.Fatal(err)
	}

	ip := localIPs[0]

	conn, err := a.listenUDP(0, 0, udp, &net.UDPAddr{IP: ip, Port: 0})
	if err != nil {
		t.Fatalf("listenUDP error with no port restriction %v", err)
	} else if conn == nil {
		t.Fatalf("listenUDP error with no port restriction return a nil conn")
	}

	_, err = a.listenUDP(4999, 5000, udp, &net.UDPAddr{IP: ip, Port: 0})
	if err == nil {
		t.Fatal("listenUDP with invalid port range did not fail")
	}
	if err != ErrPort {
		t.Fatal("listenUDP with invalid port range did not return ErrPort")
	}

	conn, err = a.listenUDP(5000, 5000, udp, &net.UDPAddr{IP: ip, Port: 0})
	if err != nil {
		t.Fatalf("listenUDP error with no port restriction %v", err)
	} else if conn == nil {
		t.Fatalf("listenUDP error with no port restriction return a nil conn")
	}

	_, port, err := net.SplitHostPort(conn.LocalAddr().String())
	if err != nil {
		t.Fatal(err)
	} else if port != "5000" {
		t.Fatalf("listenUDP with port restriction of 5000 listened on incorrect port (%s)", port)
	}
}