mirror of
https://github.com/netbirdio/netbird.git
synced 2025-07-21 08:22:20 +02:00
c02e236196
adds NetFlow functionality to track and log network traffic information between peers, with features including: - Flow logging for TCP, UDP, and ICMP traffic - Integration with connection tracking system - Resource ID tracking in NetFlow events - DNS and exit node collection configuration - Flow API and Redis cache in management - Memory-based flow storage implementation - Kernel conntrack counters and userspace counters - TCP state machine improvements for more accurate tracking - Migration from net.IP to netip.Addr in the userspace firewall
69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
package logger_test
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/netflow/logger"
|
|
"github.com/netbirdio/netbird/client/internal/netflow/types"
|
|
)
|
|
|
|
func TestStore(t *testing.T) {
|
|
logger := logger.New(context.Background(), nil, net.IPNet{})
|
|
logger.Enable()
|
|
|
|
event := types.EventFields{
|
|
FlowID: uuid.New(),
|
|
Type: types.TypeStart,
|
|
Direction: types.Ingress,
|
|
Protocol: 6,
|
|
}
|
|
|
|
wait := func() { time.Sleep(time.Millisecond) }
|
|
wait()
|
|
logger.StoreEvent(event)
|
|
wait()
|
|
|
|
allEvents := logger.GetEvents()
|
|
matched := false
|
|
for _, e := range allEvents {
|
|
if e.EventFields.FlowID == event.FlowID {
|
|
matched = true
|
|
}
|
|
}
|
|
if !matched {
|
|
t.Errorf("didn't match any event")
|
|
}
|
|
|
|
// test disable
|
|
logger.Disable()
|
|
wait()
|
|
logger.StoreEvent(event)
|
|
wait()
|
|
allEvents = logger.GetEvents()
|
|
if len(allEvents) != 0 {
|
|
t.Errorf("expected 0 events, got %d", len(allEvents))
|
|
}
|
|
|
|
// test re-enable
|
|
logger.Enable()
|
|
wait()
|
|
logger.StoreEvent(event)
|
|
wait()
|
|
|
|
allEvents = logger.GetEvents()
|
|
matched = false
|
|
for _, e := range allEvents {
|
|
if e.EventFields.FlowID == event.FlowID {
|
|
matched = true
|
|
}
|
|
}
|
|
if !matched {
|
|
t.Errorf("didn't match any event")
|
|
}
|
|
}
|