-
Notifications
You must be signed in to change notification settings - Fork 2
/
random_test.go
119 lines (107 loc) · 2.18 KB
/
random_test.go
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package balancer
import (
"strings"
"sync"
"sync/atomic"
"testing"
)
func TestRandom(t *testing.T) {
lb := NewRandom()
item := lb.Select()
if item != "" {
t.Fatalf("r expected empty, actual %s", item)
}
lb.Add("A")
item = lb.Select()
if item != "A" {
t.Fatalf("r expected A, actual %s", item)
}
nodes := []string{"A", "B", "C", "D"}
lb = NewRandom(nodes)
count := make(map[string]int)
for i := 0; i < 2000; i++ {
item := lb.Select()
count[item]++
}
if count["A"] <= 300 || count["B"] <= 300 || count["C"] <= 300 || count["D"] <= 300 {
t.Fatal("r wrong")
}
if count["A"]+count["B"]+count["C"]+count["D"] != 2000 {
t.Fatal("r wrong")
}
lb.Add("A")
lb.Add("A")
lb.Add("C")
lb.Add("A")
all := lb.All().([]string)
if strings.Join(all, "") != "ABCDAACA" {
t.Fatal("r all() wrong")
}
lb.Remove("C")
all = lb.All().([]string)
if strings.Join(all, "") != "ABDAACA" {
t.Fatal("r all() wrong")
}
lb.Remove("A", true)
all = lb.All().([]string)
if strings.Join(all, "") != "BDC" {
t.Fatal("r all() wrong")
}
lb.RemoveAll()
lb.Add("F", 1)
all, ok := lb.All().([]string)
if !ok || len(all) != 1 {
t.Fatal("r all() wrong")
}
nodes = []string{"X", "Y"}
ok = lb.Update(nodes)
if ok != true {
t.Fatal("r update wrong")
}
item = lb.Select()
if item != "X" && item != "Y" {
t.Fatal("r update wrong")
}
}
func TestRandom_C(t *testing.T) {
var (
a, b, c, d int64
)
nodes := []string{"A", "B", "C", "D"}
lb := NewRandom(nodes)
var wg sync.WaitGroup
for i := 0; i < 500; i++ {
wg.Add(1)
go func() {
defer wg.Done()
for j := 0; j < 2000; j++ {
switch lb.Select() {
case "A":
atomic.AddInt64(&a, 1)
case "B":
atomic.AddInt64(&b, 1)
case "C":
atomic.AddInt64(&c, 1)
case "D":
atomic.AddInt64(&d, 1)
}
}
}()
}
wg.Wait()
if atomic.LoadInt64(&a) <= 200000 {
t.Fatal("r wrong: a")
}
if atomic.LoadInt64(&b) <= 200000 {
t.Fatal("r wrong: b")
}
if atomic.LoadInt64(&c) <= 200000 {
t.Fatal("r wrong: c")
}
if atomic.LoadInt64(&d) <= 200000 {
t.Fatal("r wrong: d")
}
if atomic.LoadInt64(&a)+atomic.LoadInt64(&b)+atomic.LoadInt64(&c)+atomic.LoadInt64(&d) != 1000000 {
t.Fatal("r wrong: sum")
}
}