v2ray-core/transport/internet/header_test.go

package internet_test

import (
	"testing"

	. "v2ray.com/core/transport/internet"
	"v2ray.com/core/transport/internet/headers/noop"
	"v2ray.com/core/transport/internet/headers/srtp"
	"v2ray.com/core/transport/internet/headers/utp"
	. "v2ray.com/ext/assert"
)

func TestAllHeadersLoadable(t *testing.T) {
	assert := With(t)

	noopAuth, err := CreatePacketHeader((*noop.Config)(nil))
	assert(err, IsNil)
	assert(noopAuth.Size(), Equals, int32(0))

	srtp, err := CreatePacketHeader((*srtp.Config)(nil))
	assert(err, IsNil)
	assert(srtp.Size(), Equals, int32(4))

	utp, err := CreatePacketHeader((*utp.Config)(nil))
	assert(err, IsNil)
	assert(utp.Size(), Equals, int32(4))
}