Newer
Older
Christoph Petrausch
committed
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 locRIB
import (
"testing"
"github.com/bio-routing/bio-rd/net"
"github.com/bio-routing/bio-rd/route"
"github.com/stretchr/testify/assert"
)
type pfxPath struct {
pfx net.Prefix
path *route.Path
}
type containsPfxPathTestcase struct {
in []pfxPath
check pfxPath
expected bool
}
func TestContainsPfxPath(t *testing.T) {
testCases := []containsPfxPathTestcase{
{
in: []pfxPath{},
check: pfxPath{
pfx: net.NewPfx(1, 32),
path: nil,
},
expected: false,
},
// Not equal path
{
in: []pfxPath{
{
pfx: net.NewPfx(1, 32),
path: &route.Path{
Type: route.StaticPathType,
StaticPath: &route.StaticPath{
NextHop: 2,
},
},
},
},
check: pfxPath{
pfx: net.NewPfx(1, 32),
path: nil,
},
expected: false,
},
// Equal
{
in: []pfxPath{
{
pfx: net.NewPfx(1, 32),
path: &route.Path{
Type: route.StaticPathType,
StaticPath: &route.StaticPath{
NextHop: 2,
},
},
},
},
check: pfxPath{
pfx: net.NewPfx(1, 32),
path: &route.Path{
Type: route.StaticPathType,
StaticPath: &route.StaticPath{
NextHop: 2,
},
},
},
expected: true,
},
}
for i, tc := range testCases {
rib := New()
for _, p := range tc.in {
err := rib.AddPath(p.pfx, p.path)
assert.Nil(t, err, "could not fill rib in testcase %v", i)
}
contains := rib.ContainsPfxPath(tc.check.pfx, tc.check.path)
assert.Equal(t, tc.expected, contains, "mismatch in testcase %v", i)
}
}