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
|
package bonafide
import (
"reflect"
"sort"
"testing"
)
const (
eipGwTestPath = "testdata/eip-service3.json"
)
func TestGatewayPool(t *testing.T) {
b := Bonafide{client: mockClient{eipGwTestPath, geoPath}}
err := b.fetchEipJSON()
if err != nil {
t.Fatal("fetchEipJSON returned an error: ", err)
}
g := gatewayPool{available: b.eip.getGateways()}
if len(g.available) != 7 {
/* just to check that the dataset has not changed */
t.Fatal("Expected 7 initial gateways, got", len(g.available))
}
/* now we initialize a pool the proper way */
pool := newGatewayPool(b.eip)
if len(pool.available) != 7 {
t.Fatal("Expected 7 initial gateways, got", len(g.available))
}
expectedLabels := []string{"a", "b", "c"}
sort.Strings(expectedLabels)
labels := pool.getLocations()
sort.Strings(labels)
if !reflect.DeepEqual(expectedLabels, labels) {
t.Fatal("gatewayPool labels not what expected. Got:", labels)
}
if pool.userChoice != "" {
t.Fatal("userChoice should be empty by default")
}
err = pool.setUserChoice("foo")
if err == nil {
t.Fatal("gatewayPool should not let you set a foo gateway")
}
err = pool.setUserChoice("a")
if err != nil {
t.Fatal("location 'a' should be a valid label")
}
err = pool.setUserChoice("c")
if err != nil {
t.Fatal("location 'c' should be a valid label")
}
if string(pool.userChoice) != "c" {
t.Fatal("userChoice should be c")
}
pool.setAutomaticChoice()
if string(pool.userChoice) != "" {
t.Fatal("userChoice should be empty after auto selection")
}
gw, err := pool.getRandomGatewayByLocation("foo", "openvpn")
if err == nil {
t.Fatal("should get an error with invalid label")
}
gw, err = pool.getRandomGatewayByLocation("a", "openvpn")
if gw.IPAddress != "1.1.1.1" {
t.Fatal("expected to get gw 1.1.1.1 with label a")
}
gw, err = pool.getGatewayByIP("1.1.1.1")
if err != nil {
t.Fatal("expected to get gw a with ip 1.1.1.1")
}
if gw.Host != "1.example.com" {
t.Fatal("expected to get gw 1.example.com with ip 1.1.1.1")
}
// TODO test getBest
}
|
