mirror of https://github.com/XTLS/Xray-core
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package encoding_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/xtls/xray-core/common"
|
|
"github.com/xtls/xray-core/common/buf"
|
|
"github.com/xtls/xray-core/common/protocol"
|
|
"github.com/xtls/xray-core/common/uuid"
|
|
. "github.com/xtls/xray-core/proxy/vmess/encoding"
|
|
)
|
|
|
|
func TestSwitchAccount(t *testing.T) {
|
|
sa := &protocol.CommandSwitchAccount{
|
|
Port: 1234,
|
|
ID: uuid.New(),
|
|
Level: 128,
|
|
ValidMin: 16,
|
|
}
|
|
|
|
buffer := buf.New()
|
|
common.Must(MarshalCommand(sa, buffer))
|
|
|
|
cmd, err := UnmarshalCommand(1, buffer.BytesFrom(2))
|
|
common.Must(err)
|
|
|
|
sa2, ok := cmd.(*protocol.CommandSwitchAccount)
|
|
if !ok {
|
|
t.Fatal("failed to convert command to CommandSwitchAccount")
|
|
}
|
|
if r := cmp.Diff(sa2, sa); r != "" {
|
|
t.Error(r)
|
|
}
|
|
}
|
|
|
|
func TestSwitchAccountBugOffByOne(t *testing.T) {
|
|
sa := &protocol.CommandSwitchAccount{
|
|
Port: 1234,
|
|
ID: uuid.New(),
|
|
Level: 128,
|
|
ValidMin: 16,
|
|
}
|
|
|
|
buffer := buf.New()
|
|
csaf := CommandSwitchAccountFactory{}
|
|
common.Must(csaf.Marshal(sa, buffer))
|
|
|
|
Payload := buffer.Bytes()
|
|
|
|
cmd, err := csaf.Unmarshal(Payload[:len(Payload)-1])
|
|
assert.Error(t, err)
|
|
assert.Nil(t, cmd)
|
|
}
|