From 994aba5dd66ad8f399db935470105abf562cd95f Mon Sep 17 00:00:00 2001
From: Vihas Makwana <vihas.makwana@elastic.co>
Date: Fri, 7 Jun 2024 18:54:38 +0530
Subject: [PATCH] filestream: modify test cases

---
 filebeat/input/filestream/input_test.go | 47 +++++++++++++++++++------
 1 file changed, 36 insertions(+), 11 deletions(-)

diff --git a/filebeat/input/filestream/input_test.go b/filebeat/input/filestream/input_test.go
index e895dd6ab16..b267fce2740 100644
--- a/filebeat/input/filestream/input_test.go
+++ b/filebeat/input/filestream/input_test.go
@@ -35,6 +35,7 @@ import (
 	"github.com/elastic/beats/v7/libbeat/statestore/storetest"
 	conf "github.com/elastic/elastic-agent-libs/config"
 	"github.com/elastic/elastic-agent-libs/logp"
+	"github.com/elastic/elastic-agent-libs/mapstr"
 )
 
 func BenchmarkFilestream(b *testing.B) {
@@ -116,21 +117,45 @@ paths:
 }
 
 func TestTakeOverTags(t *testing.T) {
-	filename := generateFile(t, t.TempDir(), 5)
 
-	cfg := `
+	testCases := []struct {
+		name     string
+		takeOver bool
+		testFunc func(event beat.Event)
+	}{
+		{
+			name:     "test-take_over-true",
+			takeOver: true,
+			testFunc: func(event beat.Event) {
+				tags, err := event.GetValue("tags")
+				require.NoError(t, err)
+				require.Contains(t, tags, "take_over")
+			},
+		},
+		{
+			name:     "test-take_over-false",
+			takeOver: false,
+			testFunc: func(event beat.Event) {
+				_, err := event.GetValue("tags")
+				require.ErrorIs(t, err, mapstr.ErrKeyNotFound)
+			},
+		},
+	}
+	for _, testCase := range testCases {
+		t.Run(testCase.name, func(t *testing.T) {
+			filename := generateFile(t, t.TempDir(), 5)
+			cfg := fmt.Sprintf(`
 type: filestream
 prospector.scanner.check_interval: 1s
-take_over: true
+take_over: %t
 paths:
-  - ` + filename + `
-`
-	runner := createFilestreamTestRunner(context.Background(), t, "test-take_over-tag", cfg, 5, true)
-	events := runner(t)
-	for _, event := range events {
-		tags, err := event.GetValue("tags")
-		require.NoError(t, err)
-		require.Contains(t, tags, "take_over")
+    - %s`, testCase.takeOver, filename)
+			runner := createFilestreamTestRunner(context.Background(), t, testCase.name, cfg, 5, true)
+			events := runner(t)
+			for _, event := range events {
+				testCase.testFunc(event)
+			}
+		})
 	}
 }