From e104ad2887b77c2187425c2c09bdb3348ed3cfdd Mon Sep 17 00:00:00 2001
From: Valentin Lab <valentin.lab@kalysto.org>
Date: Sun, 22 Oct 2023 16:39:50 +0200
Subject: [PATCH] add tests for root node unwrap bug in previous versions

---
 acceptance_tests/output-format.sh |  9 +++++++++
 pkg/yqlib/printer_test.go         | 20 ++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/acceptance_tests/output-format.sh b/acceptance_tests/output-format.sh
index 4e9ad325..1e7bbfc2 100755
--- a/acceptance_tests/output-format.sh
+++ b/acceptance_tests/output-format.sh
@@ -72,6 +72,15 @@ EOL
   assertEquals "\"cat\"" "$X"
 }
 
+testOutputYamlRawOnRoot() {
+  cat >test.yml <<EOL
+'a'
+EOL
+
+  X=$(./yq e -r '.' test.yml)
+  assertEquals "a" "$X"
+}
+
 testOutputJsonRaw() {
   cat >test.yml <<EOL
 a: cat
diff --git a/pkg/yqlib/printer_test.go b/pkg/yqlib/printer_test.go
index 75b6e5ff..a22a2e70 100644
--- a/pkg/yqlib/printer_test.go
+++ b/pkg/yqlib/printer_test.go
@@ -390,3 +390,23 @@ func TestPrinterNulSeparatorWithJson(t *testing.T) {
 	writer.Flush()
 	test.AssertResult(t, expected, output.String())
 }
+
+func TestPrinterRootUnwrap(t *testing.T) {
+	var output bytes.Buffer
+	var writer = bufio.NewWriter(&output)
+	printer := NewSimpleYamlPrinter(writer, YamlOutputFormat, true, false, 2, false)
+	node, err := getExpressionParser().ParseExpression(".")
+	if err != nil {
+		panic(err)
+	}
+	streamEvaluator := NewStreamEvaluator()
+	_, err = streamEvaluator.Evaluate("sample", strings.NewReader("'a'"), node, printer, NewYamlDecoder(ConfiguredYamlPreferences))
+	if err != nil {
+		panic(err)
+	}
+
+	writer.Flush()
+    expected := `a
+`
+	test.AssertResult(t, expected, output.String())
+}