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
|
// Copyright (C) 2015 Nippon Telegraph and Telephone Corporation.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
// implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package bmp
import (
"github.com/osrg/gobgp/packet/bgp"
"github.com/stretchr/testify/assert"
"reflect"
"testing"
)
func verify(t *testing.T, m1 *BMPMessage) {
buf1, _ := m1.Serialize()
m2, err := ParseBMPMessage(buf1)
if err != nil {
t.Error(err)
}
buf2, _ := m2.Serialize()
if reflect.DeepEqual(m1, m2) == true {
t.Log("OK")
} else {
t.Error("Something wrong")
t.Error(len(buf1), m1, buf1)
t.Error(len(buf2), m2, buf2)
}
}
func Test_Initiation(t *testing.T) {
verify(t, NewBMPInitiation(nil))
m := NewBMPInitiation([]BMPInfoTLVInterface{
NewBMPInfoTLVString(BMP_INIT_TLV_TYPE_STRING, "free-form UTF-8 string"),
NewBMPInfoTLVUnknown(0xff, []byte{0x01, 0x02, 0x03, 0x04}),
})
verify(t, m)
}
func Test_Termination(t *testing.T) {
verify(t, NewBMPTermination(nil))
m := NewBMPTermination([]BMPTermTLVInterface{
NewBMPTermTLVString(BMP_TERM_TLV_TYPE_STRING, "free-form UTF-8 string"),
NewBMPTermTLV16(BMP_TERM_TLV_TYPE_REASON, BMP_TERM_REASON_ADMIN),
NewBMPTermTLVUnknown(0xff, []byte{0x01, 0x02, 0x03, 0x04}),
})
verify(t, m)
}
func Test_PeerUpNotification(t *testing.T) {
m := bgp.NewTestBGPOpenMessage()
p0 := NewBMPPeerHeader(0, 0, 1000, "10.0.0.1", 70000, "10.0.0.2", 1)
verify(t, NewBMPPeerUpNotification(*p0, "10.0.0.3", 10, 100, m, m))
p1 := NewBMPPeerHeader(0, 0, 1000, "fe80::6e40:8ff:feab:2c2a", 70000, "10.0.0.2", 1)
verify(t, NewBMPPeerUpNotification(*p1, "fe80::6e40:8ff:feab:2c2a", 10, 100, m, m))
}
func Test_PeerDownNotification(t *testing.T) {
p0 := NewBMPPeerHeader(0, 0, 1000, "10.0.0.1", 70000, "10.0.0.2", 1)
verify(t, NewBMPPeerDownNotification(*p0, BMP_PEER_DOWN_REASON_UNKNOWN, nil, []byte{0x3, 0xb}))
m := bgp.NewBGPNotificationMessage(1, 2, nil)
verify(t, NewBMPPeerDownNotification(*p0, BMP_PEER_DOWN_REASON_LOCAL_BGP_NOTIFICATION, m, nil))
}
func Test_RouteMonitoring(t *testing.T) {
m := bgp.NewTestBGPUpdateMessage()
p0 := NewBMPPeerHeader(0, 0, 1000, "fe80::6e40:8ff:feab:2c2a", 70000, "10.0.0.2", 1)
verify(t, NewBMPRouteMonitoring(*p0, m))
}
func Test_StatisticsReport(t *testing.T) {
p0 := NewBMPPeerHeader(0, 0, 1000, "10.0.0.1", 70000, "10.0.0.2", 1)
s0 := NewBMPStatisticsReport(
*p0,
[]BMPStatsTLVInterface{
NewBMPStatsTLV32(BMP_STAT_TYPE_REJECTED, 100),
NewBMPStatsTLV64(BMP_STAT_TYPE_ADJ_RIB_IN, 200),
NewBMPStatsTLVPerAfiSafi64(BMP_STAT_TYPE_PER_AFI_SAFI_LOC_RIB, bgp.AFI_IP, bgp.SAFI_UNICAST, 300),
},
)
verify(t, s0)
}
func Test_BogusHeader(t *testing.T) {
h, err := ParseBMPMessage(make([]byte, 10))
assert.Nil(t, h)
assert.NotNil(t, err)
}
|
