mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-09-06 00:09:20 +03:00
50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
|
package internet_test
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/xtls/xray-core/v1/common"
|
||
|
. "github.com/xtls/xray-core/v1/transport/internet"
|
||
|
"github.com/xtls/xray-core/v1/transport/internet/headers/noop"
|
||
|
"github.com/xtls/xray-core/v1/transport/internet/headers/srtp"
|
||
|
"github.com/xtls/xray-core/v1/transport/internet/headers/utp"
|
||
|
"github.com/xtls/xray-core/v1/transport/internet/headers/wechat"
|
||
|
"github.com/xtls/xray-core/v1/transport/internet/headers/wireguard"
|
||
|
)
|
||
|
|
||
|
func TestAllHeadersLoadable(t *testing.T) {
|
||
|
testCases := []struct {
|
||
|
Input interface{}
|
||
|
Size int32
|
||
|
}{
|
||
|
{
|
||
|
Input: new(noop.Config),
|
||
|
Size: 0,
|
||
|
},
|
||
|
{
|
||
|
Input: new(srtp.Config),
|
||
|
Size: 4,
|
||
|
},
|
||
|
{
|
||
|
Input: new(utp.Config),
|
||
|
Size: 4,
|
||
|
},
|
||
|
{
|
||
|
Input: new(wechat.VideoConfig),
|
||
|
Size: 13,
|
||
|
},
|
||
|
{
|
||
|
Input: new(wireguard.WireguardConfig),
|
||
|
Size: 4,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, testCase := range testCases {
|
||
|
header, err := CreatePacketHeader(testCase.Input)
|
||
|
common.Must(err)
|
||
|
if header.Size() != testCase.Size {
|
||
|
t.Error("expected size ", testCase.Size, " but got ", header.Size())
|
||
|
}
|
||
|
}
|
||
|
}
|