gurix/yoloyolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(11-01): update hardcoded count assertions to reflect 35 traffic classes

Browse Source 
- Rename TestNewBankHas14Layers to TestNewBankHasAllLayers, use len(classify.AllClasses())
- Update config_test.go: replace 4 hardcoded 14 counts with len(classify.AllClasses())
- Update TestPrintConfigContainsAllClasses: replace hardcoded list with AllClasses() loop
- Update classifier_test.go: change TestAllClassesCount want from 32 to 35
This commit is contained in:
gurix
2026-03-27 16:31:26 +01:00
parent 1556ce83e9
commit 35d8915e41
3 changed files with 17 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/config_test.go
+12 -17
View File
@@ -117,8 +117,8 @@ func TestLoadNoConfig(t *testing.T) {
t.Fatalf("Load with no config: %v", err)
}
cfgs := result.FreqCfgs
if len(cfgs) != 14 {
t.Errorf("result map size: got %d, want 14", len(cfgs))
if len(cfgs) != len(classify.AllClasses()) {
t.Errorf("result map size: got %d, want %d", len(cfgs), len(classify.AllClasses()))
}
// ICMP should be at its default BaseHz (65.0)
if cfgs[classify.ClassICMP].BaseHz != 65.0 {
@@ -146,8 +146,8 @@ func TestLoadUnknownClass(t *testing.T) {
t.Fatalf("Load with unknown class: %v", err)
}
cfgs := result.FreqCfgs
if len(cfgs) != 14 {
t.Errorf("result map size: got %d, want 14 (BOGUS should not appear)", len(cfgs))
if len(cfgs) != len(classify.AllClasses()) {
t.Errorf("result map size: got %d, want %d (BOGUS should not appear)", len(cfgs), len(classify.AllClasses()))
}
// Confirm BOGUS is NOT in the map
if _, ok := cfgs["BOGUS"]; ok {
@@ -177,8 +177,8 @@ func TestLoadAllDefaultsPresent(t *testing.T) {
t.Fatalf("Load: %v", err)
}
cfgs := result.FreqCfgs
if len(cfgs) != 14 {
t.Errorf("result map size: got %d, want 14", len(cfgs))
if len(cfgs) != len(classify.AllClasses()) {
t.Errorf("result map size: got %d, want %d", len(cfgs), len(classify.AllClasses()))
}
for _, class := range classify.AllClasses() {
if _, ok := cfgs[class]; !ok {
@@ -396,7 +396,7 @@ class = "MyDeterministicClass"
// --- PrintConfig tests ---
// TestPrintConfigContainsAllClasses: defaults LoadResult produces output with all 14 class names.
// TestPrintConfigContainsAllClasses: defaults LoadResult produces output with all class names.
func TestPrintConfigContainsAllClasses(t *testing.T) {
t.Chdir(t.TempDir())
result, err := config.Load("")
@@ -404,14 +404,9 @@ func TestPrintConfigContainsAllClasses(t *testing.T) {
t.Fatalf("Load: %v", err)
}
output := config.PrintConfig(result)
classNames := []string{
"ICMP", "DNS", "HTTPS", "HTTP", "SSH", "SMTP",
"NTP", "DHCP", "other-TCP", "other-UDP",
"unknown-1", "unknown-2", "unknown-3", "unknown-4",
}
for _, name := range classNames {
if !strings.Contains(output, name) {
t.Errorf("PrintConfig output missing class %q", name)
for _, cls := range classify.AllClasses() {
if !strings.Contains(output, string(cls)) {
t.Errorf("PrintConfig output missing class %q", cls)
}
}
}
@@ -614,8 +609,8 @@ func TestLoadNoConfigReturnsLoadResult(t *testing.T) {
if err != nil {
t.Fatalf("Load: %v", err)
}
if len(result.FreqCfgs) != 14 {
t.Errorf("FreqCfgs len: got %d, want 14", len(result.FreqCfgs))
if len(result.FreqCfgs) != len(classify.AllClasses()) {
t.Errorf("FreqCfgs len: got %d, want %d", len(result.FreqCfgs), len(classify.AllClasses()))
}
if len(result.UserRules) != 0 {
t.Errorf("UserRules len: got %d, want 0", len(result.UserRules))