test case for freedom connection

proxy/freedom/freedom_test.go

@@ -0,0 +1,82 @@
+package freedom
+
+import (
+	"io/ioutil"
+	"net"
+	"testing"
+
+	"golang.org/x/net/proxy"
+
+	"github.com/v2ray/v2ray-core"
+	_ "github.com/v2ray/v2ray-core/proxy/socks"
+	"github.com/v2ray/v2ray-core/testing/mocks"
+	"github.com/v2ray/v2ray-core/testing/unit"
+)
+
+func TestSocksTcpConnect(t *testing.T) {
+	assert := unit.Assert(t)
+	port := 48274
+
+	data2Send := "Data to be sent to remote"
+	data2Return := "Data to be returned to local"
+
+	go func() {
+		listener, err := net.ListenTCP("tcp", &net.TCPAddr{
+			IP:   []byte{0, 0, 0, 0},
+			Port: port,
+			Zone: "",
+		})
+		assert.Error(err).IsNil()
+		conn, err := listener.Accept()
+		assert.Error(err).IsNil()
+
+		buffer := make([]byte, 1024)
+		nBytes, err := conn.Read(buffer)
+		assert.Error(err).IsNil()
+
+		if string(buffer[:nBytes]) == data2Send {
+			_, err = conn.Write([]byte(data2Return))
+			assert.Error(err).IsNil()
+		}
+
+		conn.Close()
+		listener.Close()
+	}()
+
+	pointPort := uint16(38724)
+	config := mocks.Config{
+		PortValue: pointPort,
+		InboundConfigValue: &mocks.ConnectionConfig{
+			ProtocolValue: "socks",
+			ContentValue:  []byte("{\"auth\": \"noauth\"}"),
+		},
+		OutboundConfigValue: &mocks.ConnectionConfig{
+			ProtocolValue: "freedom",
+			ContentValue:  nil,
+		},
+	}
+
+	point, err := core.NewPoint(&config)
+	assert.Error(err).IsNil()
+
+	err = point.Start()
+	assert.Error(err).IsNil()
+
+	socks5Client, err := proxy.SOCKS5("tcp", "127.0.0.1:38724", nil, proxy.Direct)
+	assert.Error(err).IsNil()
+
+	targetServer := "127.0.0.1:48274"
+	conn, err := socks5Client.Dial("tcp", targetServer)
+	assert.Error(err).IsNil()
+
+	conn.Write([]byte(data2Send))
+	if tcpConn, ok := conn.(*net.TCPConn); ok {
+		tcpConn.CloseWrite()
+	}
+
+	dataReturned, err := ioutil.ReadAll(conn)
+	assert.Error(err).IsNil()
+	conn.Close()
+
+	assert.Bytes(dataReturned).Equals([]byte(data2Return))
+}