ice/networktype_test.go

// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package ice

import (
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNetworkTypeParsing_Success(t *testing.T) {
	ipv4 := net.ParseIP("192.168.0.1")
	ipv6 := net.ParseIP("fe80::a3:6ff:fec4:5454")

	for _, test := range []struct {
		name      string
		inNetwork string
		inIP      net.IP
		expected  NetworkType
	}{
		{
			"lowercase UDP4",
			"udp",
			ipv4,
			NetworkTypeUDP4,
		},
		{
			"uppercase UDP4",
			"UDP",
			ipv4,
			NetworkTypeUDP4,
		},
		{
			"lowercase UDP6",
			"udp",
			ipv6,
			NetworkTypeUDP6,
		},
		{
			"uppercase UDP6",
			"UDP",
			ipv6,
			NetworkTypeUDP6,
		},
	} {
		actual, err := determineNetworkType(test.inNetwork, test.inIP)
		if err != nil {
			t.Errorf("NetworkTypeParsing failed: %v", err)
		}
		if actual != test.expected {
			t.Errorf("NetworkTypeParsing: '%s' -- input:%s expected:%s actual:%s",
				test.name, test.inNetwork, test.expected, actual)
		}
	}
}

func TestNetworkTypeParsing_Failure(t *testing.T) {
	ipv6 := net.ParseIP("fe80::a3:6ff:fec4:5454")

	for _, test := range []struct {
		name      string
		inNetwork string
		inIP      net.IP
	}{
		{
			"invalid network",
			"junkNetwork",
			ipv6,
		},
	} {
		actual, err := determineNetworkType(test.inNetwork, test.inIP)
		if err == nil {
			t.Errorf("NetworkTypeParsing should fail: '%s' -- input:%s actual:%s",
				test.name, test.inNetwork, actual)
		}
	}
}

func TestNetworkTypeIsUDP(t *testing.T) {
	assert.True(t, NetworkTypeUDP4.IsUDP())
	assert.True(t, NetworkTypeUDP6.IsUDP())
	assert.False(t, NetworkTypeUDP4.IsTCP())
	assert.False(t, NetworkTypeUDP6.IsTCP())
}

func TestNetworkTypeIsTCP(t *testing.T) {
	assert.True(t, NetworkTypeTCP4.IsTCP())
	assert.True(t, NetworkTypeTCP6.IsTCP())
	assert.False(t, NetworkTypeTCP4.IsUDP())
	assert.False(t, NetworkTypeTCP6.IsUDP())
}