v1.0.0

common/mux/session_test.go

@@ -0,0 +1,51 @@
+package mux_test
+
+import (
+	"testing"
+
+	. "github.com/xtls/xray-core/v1/common/mux"
+)
+
+func TestSessionManagerAdd(t *testing.T) {
+	m := NewSessionManager()
+
+	s := m.Allocate()
+	if s.ID != 1 {
+		t.Error("id: ", s.ID)
+	}
+	if m.Size() != 1 {
+		t.Error("size: ", m.Size())
+	}
+
+	s = m.Allocate()
+	if s.ID != 2 {
+		t.Error("id: ", s.ID)
+	}
+	if m.Size() != 2 {
+		t.Error("size: ", m.Size())
+	}
+
+	s = &Session{
+		ID: 4,
+	}
+	m.Add(s)
+	if s.ID != 4 {
+		t.Error("id: ", s.ID)
+	}
+	if m.Size() != 3 {
+		t.Error("size: ", m.Size())
+	}
+}
+
+func TestSessionManagerClose(t *testing.T) {
+	m := NewSessionManager()
+	s := m.Allocate()
+
+	if m.CloseIfNoSession() {
+		t.Error("able to close")
+	}
+	m.Remove(s.ID)
+	if !m.CloseIfNoSession() {
+		t.Error("not able to close")
+	}
+}