From c5a515fc01fa044ca3a344fac9e8b85e99f22984 Mon Sep 17 00:00:00 2001
From: sivchari <shibuuuu5@gmail.com>
Date: Sun, 17 Mar 2024 12:24:09 +0900
Subject: [PATCH] add with quotes case

Signed-off-by: sivchari <shibuuuu5@gmail.com>
---
 flow/process_test.go | 58 +++++++++++++++++++++++++++++++++++++-------
 1 file changed, 49 insertions(+), 9 deletions(-)

diff --git a/flow/process_test.go b/flow/process_test.go
index ce5f203..a22478f 100644
--- a/flow/process_test.go
+++ b/flow/process_test.go
@@ -231,9 +231,17 @@ func TestRegexTemplate(t *testing.T) {
 }
 
 func TestVersionREwriteRegex(t *testing.T) {
-	change := "version: xyz"
 
-	original := `
+	testcases := []struct {
+		name     string
+		change   func(m regexp2.Match) string
+		original string
+		expected string
+	}{
+		{
+			name:   "no quotes",
+			change: func(m regexp2.Match) string { return "version: xyz" },
+			original: `
 version: abc
 version: 123 # do-not-rewrite
 test:
@@ -242,8 +250,8 @@ test:
       version: abc
 	foo:
       version: aiueo # no-rewrite
-`
-	expected := `
+`,
+			expected: `
 version: xyz
 version: 123 # do-not-rewrite
 test:
@@ -252,10 +260,42 @@ test:
       version: xyz
 	foo:
       version: aiueo # no-rewrite
-`
-	re := regexp2.MustCompile(versionRewriteRegex, 0)
-	result, err := re.Replace(original, change, 0, -1)
+`,
+		},
+		{
+			name:   "with quotes",
+			change: func(m regexp2.Match) string { return `version: "xyz"` },
 
-	assert.Nil(t, err)
-	assert.Equal(t, expected, result)
+			original: `
+version: abc
+version: 123 # do-not-rewrite
+test:
+  foo:
+    bar:
+      version: abc
+	foo:
+      version: aiueo # no-rewrite
+`,
+			expected: `
+version: "xyz"
+version: 123 # do-not-rewrite
+test:
+  foo:
+    bar:
+      version: "xyz"
+	foo:
+      version: aiueo # no-rewrite
+`,
+		},
+	}
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+
+			re := regexp2.MustCompile(versionRewriteRegex, 0)
+			result, err := re.ReplaceFunc(tc.original, tc.change, 0, -1)
+
+			assert.Nil(t, err)
+			assert.Equal(t, tc.expected, result)
+		})
+	}
 }