feat: create standalone native fnOS repository
This commit is contained in:
@@ -0,0 +1,53 @@
|
||||
package traffic
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/havline/havline/internal/proxy"
|
||||
)
|
||||
|
||||
func TestCollectorAttributesConnectionsByHostAndPort(t *testing.T) {
|
||||
c := NewCollector(nil, "", nil)
|
||||
now := time.Now()
|
||||
line6893 := "2026-09-17T08:30:00+08:00 xxx.com 6893 GET /login 200 0.010 218.88.23.20 192.168.8.3:6893 100 200"
|
||||
line8443 := "2026-09-17T08:30:01+08:00 xxx.com 8443 GET / 200 0.010 218.88.23.20 192.168.8.3:8443 100 200"
|
||||
|
||||
c.ingestLine(line6893, true)
|
||||
c.ingestLine(line8443, true)
|
||||
|
||||
rules := []proxy.Rule{
|
||||
{
|
||||
ID: 1,
|
||||
ListenPort: 6893,
|
||||
Hosts: []proxy.Host{{Hostname: "xxx.com"}},
|
||||
},
|
||||
{
|
||||
ID: 2,
|
||||
ListenPort: 8443,
|
||||
Hosts: []proxy.Host{{Hostname: "xxx.com"}},
|
||||
},
|
||||
}
|
||||
|
||||
stats := c.SnapshotForRules(rules)
|
||||
if len(stats) != 2 {
|
||||
t.Fatalf("expected 2 stats, got %d", len(stats))
|
||||
}
|
||||
byID := map[int64]RuleTraffic{}
|
||||
for _, s := range stats {
|
||||
byID[s.RuleID] = s
|
||||
}
|
||||
if byID[1].Connections != 1 {
|
||||
t.Fatalf("rule 6893 connections=%d want 1", byID[1].Connections)
|
||||
}
|
||||
if byID[2].Connections != 1 {
|
||||
t.Fatalf("rule 8443 connections=%d want 1", byID[2].Connections)
|
||||
}
|
||||
_ = now
|
||||
}
|
||||
|
||||
func TestEndpointKeyIncludesPort(t *testing.T) {
|
||||
if got := EndpointKey("XXX.com", 6893); got != "xxx.com:6893" {
|
||||
t.Fatalf("got %q", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user