SwitchAccount command

proxy/vmess/command/accounts_test.go

@@ -0,0 +1,35 @@
+package command_test
+
+import (
+	"bytes"
+	"testing"
+	"time"
+
+	"github.com/v2ray/v2ray-core/common/uuid"
+	. "github.com/v2ray/v2ray-core/proxy/vmess/command"
+	v2testing "github.com/v2ray/v2ray-core/testing"
+	"github.com/v2ray/v2ray-core/testing/assert"
+)
+
+func TestSwitchAccount(t *testing.T) {
+	v2testing.Current(t)
+
+	sa := &SwitchAccount{
+		ID:         uuid.New(),
+		ValidUntil: time.Now(),
+	}
+
+	cmd, err := CreateResponseCommand(1)
+	assert.Error(err).IsNil()
+
+	buffer := bytes.NewBuffer(make([]byte, 0, 1024))
+	nBytes, err := sa.Marshal(buffer)
+	assert.Error(err).IsNil()
+	assert.Int(nBytes).Equals(buffer.Len())
+
+	cmd.Unmarshal(buffer.Bytes())
+	sa2, ok := cmd.(*SwitchAccount)
+	assert.Bool(ok).IsTrue()
+	assert.String(sa.ID).Equals(sa2.ID.String())
+	assert.Int64(sa.ValidUntil.Unix()).Equals(sa2.ValidUntil.Unix())
+}