-
Notifications
You must be signed in to change notification settings - Fork 0
/
barn_test.go
97 lines (89 loc) · 3.01 KB
/
barn_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
package main
import (
"bytes"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"testing"
)
func LoadTestConfig() {
viper.SetConfigType("yaml") // or viper.SetConfigType("YAML")
// any approach to require this configuration into your program.
var yamlExample = []byte(`
monitors:
http:
remote:
name: "Some remote url"
description: "Some remote url description"
url: http://127.0.0.1/test
rule:
invert: true
pattern: "'^[A-Z]+\\.com$"
remote2:
name: "Some remote url2"
description: "Some remote url2 description"
url: http://127.0.0.2/test
file:
local:
name: "Local file"
description: "Some local file description"
path: /tmp/test
dummy:
fake:
is_safe: true
`)
viper.ReadConfig(bytes.NewBuffer(yamlExample))
}
func TestBarnServer_AddMonitorTest(t *testing.T) {
var barn = New()
sm := NewSafetyMonitorDummy("dummy", "name", "description", false)
barn.AddMonitor(sm)
assert.Equal(t, barn.monitors[sm.GetId()], sm, "List of monitors should contain monitor")
assert.Equal(t, "dummy", barn.GetMonitor(sm.GetId()).GetId(), "List of monitors should contain monitor")
}
func TestBarnServer_RemoveMonitorTest(t *testing.T) {
var barn = New()
sm := NewSafetyMonitorDummy("dummy", "name", "description", false)
barn.AddMonitor(sm)
barn.RemoveMonitor("dummy")
assert.Nil(t, barn.monitors["dummy"], "should be nil")
}
func TestBarnServer_GetMonitorByIndex(t *testing.T) {
var barn = New()
sm := NewSafetyMonitorDummy("dummy", "name", "description", false)
barn.AddMonitor(sm)
sm2 := NewSafetyMonitorDummy("aummy2", "name", "description", true)
barn.AddMonitor(sm2)
smr, _ := barn.GetMonitorByIndex(0)
assert.Equal(t, "dummy", smr.GetId(), "should be equal")
assert.Equal(t, false, smr.IsSafe(), "should be equal")
smr, _ = barn.GetMonitorByIndex(1)
assert.Equal(t, "aummy2", smr.GetId(), "should be equal")
_, err := barn.GetMonitorByIndex(2)
assert.Error(t, err, "should be error")
}
func TestBarnServer_LoadConfig(t *testing.T) {
LoadTestConfig()
var barn = New()
mCfg := viper.GetViper()
barn.LoadMonitorsFromConfig(mCfg)
assert.NotNil(t, barn.GetMonitor("remote"), "shouldn't be nil")
assert.NotNil(t, barn.GetMonitor("remote2"), "shouldn't be nil")
switch v := barn.GetMonitor("remote").(type) {
case *SafetyMonitorHttp:
assert.Equal(t, "http://127.0.0.1/test", v.url, "should be equal")
assert.Equal(t, "Some remote url", v.GetName(), "should be equal")
assert.Equal(t, "Some remote url description", v.GetDescription(), "should be equal")
assert.Equal(t, "'^[A-Z]+\\.com$", v.rule.pattern, "should be equal")
assert.Equal(t, false, v.rule.invert, "should be equal")
default:
assert.Fail(t, "Wrong type")
}
assert.NotNil(t, barn.GetMonitor("local"), "shouldn't be nil")
switch v := barn.GetMonitor("local").(type) {
case *SafetyMonitorFile:
assert.Equal(t, v.path, "/tmp/test", "should be equal")
default:
assert.Fail(t, "Wrong type")
}
assert.NotNil(t, barn.GetMonitor("fake"), "shouldn't be nil")
}