-
Notifications
You must be signed in to change notification settings - Fork 9
/
alert_test.go
66 lines (57 loc) · 1.25 KB
/
alert_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
package deepalert_test
import (
"testing"
"github.com/stretchr/testify/assert"
da "github.com/deepalert/deepalert"
)
func TestLookupAttribute(t *testing.T) {
alert := da.Alert{
Attributes: []da.Attribute{
{
Key: "myaddr",
Type: da.TypeIPAddr,
Context: []da.AttrContext{
da.CtxRemote,
da.CtxSubject,
},
},
},
}
attrs1 := alert.FindAttributes("myaddr")
assert.Equal(t, 1, len(attrs1))
assert.True(t, attrs1[0].Match(da.CtxRemote, da.TypeIPAddr))
assert.False(t, attrs1[0].Match(da.CtxLocal, da.TypeIPAddr))
assert.False(t, attrs1[0].Match(da.CtxSubject, da.TypeUserName))
attrs2 := alert.FindAttributes("invalid key")
assert.Equal(t, 0, len(attrs2))
}
func TestAttributeHash(t *testing.T) {
a1 := da.Attribute{
Key: "myaddr",
Type: da.TypeIPAddr,
Context: []da.AttrContext{
da.CtxRemote,
da.CtxSubject,
},
}
a2 := da.Attribute{
Key: "hoge",
Type: da.TypeIPAddr,
Context: []da.AttrContext{
da.CtxRemote,
da.CtxSubject,
},
}
a3 := da.Attribute{
Key: "myaddr",
Type: da.TypeIPAddr,
Context: []da.AttrContext{
// Reversed
da.CtxSubject,
da.CtxRemote,
},
}
assert.NotEqual(t, a1.Hash(), "")
assert.NotEqual(t, a1.Hash(), a2.Hash())
assert.Equal(t, a1.Hash(), a3.Hash())
}