git add

Hunter0x7c7

2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece

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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package stats_test 
 
import ( 
    "context" 
    "testing" 
    "time" 
 
    . "github.com/v2fly/v2ray-core/v5/app/stats" 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/features/stats" 
) 
 
func TestInterface(t *testing.T) { 
    _ = (stats.Manager)(new(Manager)) 
} 
 
func TestStatsChannelRunnable(t *testing.T) { 
    raw, err := common.CreateObject(context.Background(), &Config{}) 
    common.Must(err) 
 
    m := raw.(stats.Manager) 
 
    ch1, err := m.RegisterChannel("test.channel.1") 
    c1 := ch1.(*Channel) 
    common.Must(err) 
 
    if c1.Running() { 
        t.Fatalf("unexpected running channel: test.channel.%d", 1) 
    } 
 
    common.Must(m.Start()) 
 
    if !c1.Running() { 
        t.Fatalf("unexpected non-running channel: test.channel.%d", 1) 
    } 
 
    ch2, err := m.RegisterChannel("test.channel.2") 
    c2 := ch2.(*Channel) 
    common.Must(err) 
 
    if !c2.Running() { 
        t.Fatalf("unexpected non-running channel: test.channel.%d", 2) 
    } 
 
    s1, err := c1.Subscribe() 
    common.Must(err) 
    common.Must(c1.Close()) 
 
    if c1.Running() { 
        t.Fatalf("unexpected running channel: test.channel.%d", 1) 
    } 
 
    select { // Check all subscribers in closed channel are closed 
    case _, ok := <-s1: 
        if ok { 
            t.Fatalf("unexpected non-closed subscriber in channel: test.channel.%d", 1) 
        } 
    case <-time.After(500 * time.Millisecond): 
        t.Fatalf("unexpected non-closed subscriber in channel: test.channel.%d", 1) 
    } 
 
    if len(c1.Subscribers()) != 0 { // Check subscribers in closed channel are emptied 
        t.Fatalf("unexpected non-empty subscribers in channel: test.channel.%d", 1) 
    } 
 
    common.Must(m.Close()) 
 
    if c2.Running() { 
        t.Fatalf("unexpected running channel: test.channel.%d", 2) 
    } 
 
    ch3, err := m.RegisterChannel("test.channel.3") 
    c3 := ch3.(*Channel) 
    common.Must(err) 
 
    if c3.Running() { 
        t.Fatalf("unexpected running channel: test.channel.%d", 3) 
    } 
 
    common.Must(c3.Start()) 
    common.Must(m.UnregisterChannel("test.channel.3")) 
 
    if c3.Running() { // Test that unregistering will close the channel. 
        t.Fatalf("unexpected running channel: test.channel.%d", 3) 
    } 
}