natpmp_mock_test.go

package mocknat

import (
	"net"
	"testing"

	natpmp "github.com/jackpal/go-nat-pmp"
)

func TestExternalIP(t *testing.T) {
	nat := New(net.IPv4(127, 0, 0, 1), net.IPv4(1, 1, 1, 1), true)

	nat.Run()
	defer nat.Close()

	client := natpmp.NewClient(net.IPv4(127, 0, 0, 1))
	result, err := client.GetExternalAddress()
	if err != nil {
		t.Fatal(err)
	}

	got := net.IP(result.ExternalIPAddress[:])
	if !got.Equal(nat.externalIP) {
		t.Fatalf("invalid GetExternalAddress, want: %v, got: %v", nat.externalIP, got)
	}
}

func TestAddPortMapping(t *testing.T) {
	nat := New(net.IPv4(127, 0, 0, 1), net.IPv4(1, 1, 1, 1), true)

	nat.Run()
	defer nat.Close()

	var result *natpmp.AddPortMappingResult
	var err error

	client := natpmp.NewClient(net.IPv4(127, 0, 0, 1))
	result, err = client.AddPortMapping("udp", 2000, 5000, 30)
	if err != nil {
		t.Fatal(err)
	}
	_ = result

	if internal := nat.Map("udp", 5000); internal == nil {
		t.Fatal("")
	}
	if internal := nat.Map("udp", 5001); internal != nil {
		t.Fatal("")
	}
}

func TestSuggestPortMapping(t *testing.T) {
	nat := New(net.IPv4(127, 0, 0, 1), net.IPv4(1, 1, 1, 1), true)

	nat.Run()
	defer nat.Close()

	var result *natpmp.AddPortMappingResult
	var err error

	client := natpmp.NewClient(net.IPv4(127, 0, 0, 1))
	result, err = client.AddPortMapping("udp", 2000, 0, 30)
	if err != nil {
		t.Fatal(err)
	}
	if result.MappedExternalPort != 1024 {
		t.Fatal("")
	}
	result, err = client.AddPortMapping("udp", 2000, 0, 30)
	if err != nil {
		t.Fatal(err)
	}
	if result.MappedExternalPort != 1025 {
		t.Fatal("")
	}
}