v2ray-core/common/mux/session_test.go

package mux_test

import (
	"testing"

	. "v2ray.com/core/common/mux"
	. "v2ray.com/ext/assert"
)

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

	m := NewSessionManager()

	s := m.Allocate()
	assert(s.ID, Equals, uint16(1))
	assert(m.Size(), Equals, 1)

	s = m.Allocate()
	assert(s.ID, Equals, uint16(2))
	assert(m.Size(), Equals, 2)

	s = &Session{
		ID: 4,
	}
	m.Add(s)
	assert(s.ID, Equals, uint16(4))
}

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

	m := NewSessionManager()
	s := m.Allocate()

	assert(m.CloseIfNoSession(), IsFalse)
	m.Remove(s.ID)
	assert(m.CloseIfNoSession(), IsTrue)
}