Xray-core/transport/internet/sockopt_test.go

41 lines
866 B
Go
Raw Normal View History

2020-11-25 11:01:53 +00:00
package internet_test
import (
"context"
"testing"
"github.com/google/go-cmp/cmp"
2020-12-04 01:36:16 +00:00
"github.com/xtls/xray-core/common"
"github.com/xtls/xray-core/common/buf"
"github.com/xtls/xray-core/testing/servers/tcp"
. "github.com/xtls/xray-core/transport/internet"
2020-11-25 11:01:53 +00:00
)
func TestTCPFastOpen(t *testing.T) {
tcpServer := tcp.Server{
MsgProcessor: func(b []byte) []byte {
return b
},
}
2021-03-06 14:45:12 +00:00
dest, err := tcpServer.StartContext(context.Background(), &SocketConfig{Tfo: 256})
2020-11-25 11:01:53 +00:00
common.Must(err)
defer tcpServer.Close()
ctx := context.Background()
dialer := DefaultSystemDialer{}
conn, err := dialer.Dial(ctx, nil, dest, &SocketConfig{
2021-03-06 14:45:12 +00:00
Tfo: 1,
2020-11-25 11:01:53 +00:00
})
common.Must(err)
defer conn.Close()
_, err = conn.Write([]byte("abcd"))
common.Must(err)
b := buf.New()
common.Must2(b.ReadFrom(conn))
if r := cmp.Diff(b.Bytes(), []byte("abcd")); r != "" {
t.Fatal(r)
}
}