mirror of
https://github.com/akvorado/akvorado.git
synced 2025-12-11 22:14:02 +01:00
114 lines
2.7 KiB
Go
114 lines
2.7 KiB
Go
// SPDX-FileCopyrightText: 2022 Free Mobile
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
package bmp_test
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
"testing"
|
|
"time"
|
|
|
|
gobmp "github.com/osrg/gobgp/v4/pkg/packet/bmp"
|
|
|
|
"akvorado/common/daemon"
|
|
"akvorado/common/helpers"
|
|
"akvorado/common/reporter"
|
|
"akvorado/demoexporter/bmp"
|
|
)
|
|
|
|
func TestClient(t *testing.T) {
|
|
listener, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatalf("Listen() error:\n%+v", err)
|
|
}
|
|
defer listener.Close()
|
|
|
|
config := bmp.DefaultConfiguration()
|
|
config.Target = listener.Addr().String()
|
|
config.RetryAfter = 0
|
|
config.StatsDelay = 10 * time.Millisecond
|
|
config.Routes = []bmp.RouteConfiguration{
|
|
{
|
|
Prefixes: []netip.Prefix{netip.MustParsePrefix("2001:db8::/64")},
|
|
ASPath: []uint32{65001, 65002, 65002},
|
|
Communities: []bmp.Community{500, 600, 700},
|
|
}, {
|
|
Prefixes: []netip.Prefix{
|
|
netip.MustParsePrefix("192.0.2.0/24"),
|
|
netip.MustParsePrefix("203.0.113.0/24"),
|
|
},
|
|
ASPath: []uint32{12322, 1299},
|
|
}, {
|
|
Prefixes: []netip.Prefix{
|
|
netip.MustParsePrefix("192.0.2.0/24"),
|
|
netip.MustParsePrefix("2001:db8::/64"),
|
|
},
|
|
ASPath: []uint32{65001, 65002},
|
|
LargeCommunities: []bmp.LargeCommunity{
|
|
{ASN: 65000, LocalData1: 100, LocalData2: 200},
|
|
{ASN: 65000, LocalData1: 300, LocalData2: 2000},
|
|
},
|
|
},
|
|
}
|
|
r := reporter.NewMock(t)
|
|
c, err := bmp.New(r, config, bmp.Dependencies{
|
|
Daemon: daemon.NewMock(t),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("New() error:\n%+v", err)
|
|
}
|
|
helpers.StartStop(t, c)
|
|
|
|
// Test we get a reconnect
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
t.Fatalf("Accept() error:\n%+v", err)
|
|
}
|
|
time.Sleep(20 * time.Millisecond)
|
|
conn.Close()
|
|
conn, err = listener.Accept()
|
|
if err != nil {
|
|
t.Fatalf("Accept() error:\n%+v", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
got := make([]byte, 5000)
|
|
n, err := conn.Read(got)
|
|
if err != nil {
|
|
t.Fatalf("Read() error:\n%+v", err)
|
|
}
|
|
got = got[:n]
|
|
|
|
msgs := []*gobmp.BMPMessage{}
|
|
for {
|
|
advance, token, err := gobmp.SplitBMP(got, len(got) > 0)
|
|
if err != nil {
|
|
t.Fatalf("SplitBMP() error:\n%+v", err)
|
|
}
|
|
if token == nil {
|
|
break
|
|
}
|
|
t.Logf("BMP message len: %d", len(token))
|
|
msg, err := gobmp.ParseBMPMessage(token)
|
|
if err != nil {
|
|
t.Fatalf("ParseBMPMessage() error:\n%+v", err)
|
|
}
|
|
msgs = append(msgs, msg)
|
|
got = got[advance:]
|
|
}
|
|
|
|
// Assume we got what we want.
|
|
_ = msgs
|
|
|
|
time.Sleep(20 * time.Millisecond)
|
|
gotMetrics := r.GetMetrics("akvorado_demoexporter_")
|
|
expectedMetrics := map[string]string{
|
|
`bmp_connections_total`: "2",
|
|
`bmp_errors_total{error="EOF"}`: "1",
|
|
}
|
|
if diff := helpers.Diff(gotMetrics, expectedMetrics); diff != "" {
|
|
t.Fatalf("Metrics (-got, +want):\n%s", diff)
|
|
}
|
|
}
|