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
| package srtp_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/srtp"
| )
|
| func TestSRTPWrite(t *testing.T) {
| content := []byte{'a', 'b', 'c', 'd', 'e', 'f', 'g'}
| srtpRaw, err := New(context.Background(), &Config{})
| common.Must(err)
|
| srtp := srtpRaw.(*SRTP)
|
| payload := buf.New()
| srtp.Serialize(payload.Extend(srtp.Size()))
| payload.Write(content)
|
| expectedLen := int32(len(content)) + srtp.Size()
| if payload.Len() != expectedLen {
| t.Error("expected ", expectedLen, " of bytes, but got ", payload.Len())
| }
| }
|
|
