1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
| package mux_test
|
| import (
| "testing"
|
| . "github.com/v2fly/v2ray-core/v5/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")
| }
| }
|
|
