v1.0.0

common/net/destination_test.go

@@ -0,0 +1,111 @@
+package net_test
+
+import (
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+
+	. "github.com/xtls/xray-core/v1/common/net"
+)
+
+func TestDestinationProperty(t *testing.T) {
+	testCases := []struct {
+		Input     Destination
+		Network   Network
+		String    string
+		NetString string
+	}{
+		{
+			Input:     TCPDestination(IPAddress([]byte{1, 2, 3, 4}), 80),
+			Network:   Network_TCP,
+			String:    "tcp:1.2.3.4:80",
+			NetString: "1.2.3.4:80",
+		},
+		{
+			Input:     UDPDestination(IPAddress([]byte{0x20, 0x01, 0x48, 0x60, 0x48, 0x60, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x88, 0x88}), 53),
+			Network:   Network_UDP,
+			String:    "udp:[2001:4860:4860::8888]:53",
+			NetString: "[2001:4860:4860::8888]:53",
+		},
+		{
+			Input:     UnixDestination(DomainAddress("/tmp/test.sock")),
+			Network:   Network_UNIX,
+			String:    "unix:/tmp/test.sock",
+			NetString: "/tmp/test.sock",
+		},
+	}
+
+	for _, testCase := range testCases {
+		dest := testCase.Input
+		if r := cmp.Diff(dest.Network, testCase.Network); r != "" {
+			t.Error("unexpected Network in ", dest.String(), ": ", r)
+		}
+		if r := cmp.Diff(dest.String(), testCase.String); r != "" {
+			t.Error(r)
+		}
+		if r := cmp.Diff(dest.NetAddr(), testCase.NetString); r != "" {
+			t.Error(r)
+		}
+	}
+}
+
+func TestDestinationParse(t *testing.T) {
+	cases := []struct {
+		Input  string
+		Output Destination
+		Error  bool
+	}{
+		{
+			Input:  "tcp:127.0.0.1:80",
+			Output: TCPDestination(LocalHostIP, Port(80)),
+		},
+		{
+			Input:  "udp:8.8.8.8:53",
+			Output: UDPDestination(IPAddress([]byte{8, 8, 8, 8}), Port(53)),
+		},
+		{
+			Input:  "unix:/tmp/test.sock",
+			Output: UnixDestination(DomainAddress("/tmp/test.sock")),
+		},
+		{
+			Input: "8.8.8.8:53",
+			Output: Destination{
+				Address: IPAddress([]byte{8, 8, 8, 8}),
+				Port:    Port(53),
+			},
+		},
+		{
+			Input: ":53",
+			Output: Destination{
+				Address: AnyIP,
+				Port:    Port(53),
+			},
+		},
+		{
+			Input: "8.8.8.8",
+			Error: true,
+		},
+		{
+			Input: "8.8.8.8:http",
+			Error: true,
+		},
+		{
+			Input: "/tmp/test.sock",
+			Error: true,
+		},
+	}
+
+	for _, testcase := range cases {
+		d, err := ParseDestination(testcase.Input)
+		if !testcase.Error {
+			if err != nil {
+				t.Error("for test case: ", testcase.Input, " expected no error, but got ", err)
+			}
+			if d != testcase.Output {
+				t.Error("for test case: ", testcase.Input, " expected output: ", testcase.Output.String(), " but got ", d.String())
+			}
+		} else if err == nil {
+			t.Error("for test case: ", testcase.Input, " expected error, but got nil")
+		}
+	}
+}