update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package internet_test 
 
import ( 
    "context" 
    "testing" 
 
    "github.com/google/go-cmp/cmp" 
 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/common/buf" 
    "github.com/v2fly/v2ray-core/v5/testing/servers/tcp" 
    . "github.com/v2fly/v2ray-core/v5/transport/internet" 
) 
 
func TestTCPFastOpen(t *testing.T) { 
    tcpServer := tcp.Server{ 
        MsgProcessor: func(b []byte) []byte { 
            return b 
        }, 
    } 
    dest, err := tcpServer.StartContext(context.Background(), &SocketConfig{Tfo: SocketConfig_Enable}) 
    common.Must(err) 
    defer tcpServer.Close() 
 
    ctx := context.Background() 
    dialer := DefaultSystemDialer{} 
    conn, err := dialer.Dial(ctx, nil, dest, &SocketConfig{ 
        Tfo: SocketConfig_Enable, 
    }) 
    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) 
    } 
}