sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-09-24 15:49:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

update test to validate event aggregation over tcp, udp, icmp, and icmpv6

Browse Source 
Signed-off-by: Dmitri <dmitri.external@netbird.io>
This commit is contained in:
Dmitri
2026-06-11 15:34:03 +02:00
parent 598558c77e
commit 98ce097ecb
1 changed files with 18 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client/internal/netflow/store/tcp_aggregation_test.go
+9 -12
View File
@@ -18,7 +18,8 @@ var pregeneratedUUIDs = func() []uuid.UUID {
return toret return toret
}() }()
func TestTcpAggregation(t *testing.T) { func TestFlowAggregation(t *testing.T) {
var protocols = []types.Protocol{types.ICMP, types.ICMPv6, types.TCP, types.UDP}
var tests = []struct { var tests = []struct {
description string description string
events []*types.Event events []*types.Event
@@ -35,7 +36,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -55,7 +55,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeEnd, Type: types.TypeEnd,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -77,7 +76,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -105,7 +103,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -125,7 +122,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeDrop, Type: types.TypeDrop,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -147,7 +143,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -175,7 +170,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -197,7 +191,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeStart, Type: types.TypeStart,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -225,7 +218,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeDrop, Type: types.TypeDrop,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -247,7 +239,6 @@ func TestTcpAggregation(t *testing.T) {
Type: types.TypeDrop, Type: types.TypeDrop,
RuleID: []byte("rule-id-1"), RuleID: []byte("rule-id-1"),
Direction: types.Egress, Direction: types.Egress,
Protocol: types.TCP,
SourceIP: ipAddr("1.1.1.1"), SourceIP: ipAddr("1.1.1.1"),
SourcePort: 1234, SourcePort: 1234,
DestIP: ipAddr("2.2.2.2"), DestIP: ipAddr("2.2.2.2"),
@@ -265,18 +256,24 @@ func TestTcpAggregation(t *testing.T) {
}, },
}} }}
for _, protocol := range protocols {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.description, func(t *testing.T) { t.Run(tt.description+" "+protocol.String(), func(t *testing.T) {
store := NewAggregatingMemoryStore() store := NewAggregatingMemoryStore()
for _, e := range tt.events { for _, e := range tt.events {
e.Protocol = protocol
store.StoreEvent(e) store.StoreEvent(e)
} }
for _, e := range tt.expected {
e.Protocol = protocol
}
events := store.GetAggregatedEvents() events := store.GetAggregatedEvents()
assert.Len(t, events, len(tt.expected)) assert.Len(t, events, len(tt.expected))
assert.ElementsMatch(t, events, tt.expected) assert.ElementsMatch(t, events, tt.expected)
}) })
} }
} }
}
func ipAddr(a string) netip.Addr { func ipAddr(a string) netip.Addr {
addr, _ := netip.ParseAddr(a) addr, _ := netip.ParseAddr(a)