v2ray-core/transport/internet/tcp/sockopt_linux_test.go

// +build linux

package tcp_test

import (
	"context"
	"strings"
	"testing"

	"v2ray.com/core/testing/servers/tcp"
	"v2ray.com/core/transport/internet"
	. "v2ray.com/core/transport/internet/tcp"
	. "v2ray.com/ext/assert"
)

func TestGetOriginalDestination(t *testing.T) {
	assert := With(t)

	tcpServer := tcp.Server{}
	dest, err := tcpServer.Start()
	assert(err, IsNil)
	defer tcpServer.Close()

	conn, err := Dial(context.Background(), dest, internet.ToMemoryStreamConfig(nil))
	assert(err, IsNil)
	defer conn.Close()

	originalDest, err := GetOriginalDestination(conn)
	assert(dest == originalDest || strings.Contains(err.Error(), "failed to call getsockopt"), IsTrue)
}