1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-23 02:26:49 -05:00
v2fly/transport/internet/headers/srtp/srtp_test.go
2018-11-02 21:34:04 +01:00

27 lines
529 B
Go

package srtp_test
import (
"context"
"testing"
"v2ray.com/core/common/buf"
. "v2ray.com/core/transport/internet/headers/srtp"
. "v2ray.com/ext/assert"
)
func TestSRTPWrite(t *testing.T) {
assert := With(t)
content := []byte{'a', 'b', 'c', 'd', 'e', 'f', 'g'}
srtpRaw, err := New(context.Background(), &Config{})
assert(err, IsNil)
srtp := srtpRaw.(*SRTP)
payload := buf.New()
srtp.Serialize(payload.Extend(srtp.Size()))
payload.Write(content)
assert(payload.Len(), Equals, int32(len(content))+srtp.Size())
}