fix test

Author: jotak

cmd/display_loop_test.go

@@ -7,6 +7,7 @@ import (
 )
 
 func TestDisplayLoop(t *testing.T) {
+	enrichment.current = 0
 	assert.Equal(t, []string{"None"}, enrichment.getNames())
 	assert.Equal(t, []string{"SrcAddr", "SrcPort", "DstAddr", "DstPort"}, enrichment.getCols())
 

cmd/flow_capture.go

@@ -35,7 +35,7 @@ var (
 	lastFlows   = []config.GenericMap{}
 
 	features = displayLoop{
-		current: 6,
+		current: featureDefaultIndex,
 		all: []displayLoopItem{
 			{name: "Raw"},
 			{name: "Standard", columns: []string{"Dir", "Interfaces", "Proto", "Dscp", "Bytes", "Packets"}},
@@ -48,7 +48,7 @@ var (
 	}
 
 	enrichment = displayLoop{
-		current: 4,
+		current: enrichmentDefaultIndex,
 		all: []displayLoopItem{
 			{name: "None", columns: []string{"SrcAddr", "SrcPort", "DstAddr", "DstPort"}},
 			{name: "Zone", columns: []string{"SrcZone", "DstZone"}},
@@ -60,6 +60,11 @@ var (
 	}
 )
 
+const (
+	featureDefaultIndex    = 6 // All feats.
+	enrichmentDefaultIndex = 4 // Resources
+)
+
 func runFlowCapture(_ *cobra.Command, _ []string) {
 	go func() {
 		scanner()

cmd/flow_capture_test.go

@@ -54,6 +54,7 @@ func TestFlowTableMultipleFlows(t *testing.T) {
 
 	// set display to standard without enrichment
 	features.current = 1
+	enrichment.current = 0
 
 	// set time and bytes per flow
 	flowTime := 1704063600000

cmd/root_test.go

@@ -83,6 +83,8 @@ func setup() {
 	// clear filters and previous flows
 	regexes = []string{}
 	lastFlows = []config.GenericMap{}
+	features.current = featureDefaultIndex
+	enrichment.current = enrichmentDefaultIndex
 }
 
 func resetTime() {