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
package tls_test
 
import (
    "context"
    "testing"
 
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/common/buf"
    . "github.com/v2fly/v2ray-core/v5/transport/internet/headers/tls"
)
 
func TestDTLSWrite(t *testing.T) {
    content := []byte{'a', 'b', 'c', 'd', 'e', 'f', 'g'}
    dtlsRaw, err := New(context.Background(), &PacketConfig{})
    common.Must(err)
 
    dtls := dtlsRaw.(*DTLS)
 
    payload := buf.New()
    dtls.Serialize(payload.Extend(dtls.Size()))
    payload.Write(content)
 
    if payload.Len() != int32(len(content))+dtls.Size() {
        t.Error("payload len: ", payload.Len(), " want ", int32(len(content))+dtls.Size())
    }
}