[Tests] Escaped characters in YAML keys break encoding/decoding

decode_test.go

@@ -2890,3 +2890,117 @@ func TestSameNameInineStruct(t *testing.T) {
 		t.Fatalf("failed to decode")
 	}
 }
+
+// Test newline in key as encoded by alternative YAML tools:
+func TestUnmarshalNewlineInKey(t *testing.T) {
+	tests := []struct {
+		name string
+		spec string
+	}{
+		{
+			name: "multiline-key-with-newline-01",
+			/* As rendered by this library from this JSON:
+			{
+				"a": "a",
+				"c\nc": "cc",
+				"d": "d"
+			}
+			*/
+			spec: `a: a
+|-
+  c
+  c: cc
+d: d
+`,
+		},
+		{
+			name: "quoted-key-with-newline",
+			spec: `a: a
+"c\nc": "cc"
+d: d
+`,
+		},
+		{
+			name: "multiline-key-with-newline-02",
+			// As encoded by alternative online YAML tools (e.g. https://www.json2yaml.com/)
+			spec: `a: a
+? |-
+  c
+  c
+: cc
+d: d
+`,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			decoder := yaml.NewDecoder(strings.NewReader(test.spec))
+			var out map[string]interface{}
+			for i := 0; ; i++ {
+				err := decoder.Decode(&out)
+				if err == io.EOF {
+					break
+				}
+				if err != nil {
+					t.Fatalf("Error on iteration %d: %v", i, err)
+				}
+			}
+		})
+	}
+}
+
+func TestUnmarshalBackslashInKey(t *testing.T) {
+	spec := `outer:
+  "b\\b": b
+  d : d
+`
+
+	decoder := yaml.NewDecoder(strings.NewReader(spec))
+	var out map[string]interface{}
+	for i := 0; ; i++ {
+		err := decoder.Decode(&out)
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			t.Fatalf("Error on iteration %d: %v", i, err)
+		}
+	}
+}
+
+func TestUnmarshalQuotesInKey(t *testing.T) {
+	spec := `outer:
+  "a\"b\"c": a
+  "d\"e\"f": d
+`
+	decoder := yaml.NewDecoder(strings.NewReader(spec))
+	var out map[string]interface{}
+	for i := 0; ; i++ {
+		err := decoder.Decode(&out)
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			t.Fatalf("Error on iteration %d: %v", i, err)
+		}
+	}
+}
+
+func TestUnmarshalCollectedEscapesInKey(t *testing.T) {
+	spec := `outer:
+  "a\"b\"c\\d\ne": a
+  "d\"e\"f": d
+	`
+	decoder := yaml.NewDecoder(strings.NewReader(spec))
+	var out map[string]interface{}
+	for i := 0; ; i++ {
+		err := decoder.Decode(&out)
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			t.Fatalf("Error on iteration %d: %v", i, err)
+		}
+	}
+}

encode_test.go

@@ -7,6 +7,7 @@ import (
 	"math"
 	"reflect"
 	"strconv"
+	"strings"
 	"testing"
 	"time"
 
@@ -1630,3 +1631,35 @@ b:
 		t.Fatalf("failed to encode. expected %s but got %s", expected, got)
 	}
 }
+
+func TestMarshalNewlineInKey(t *testing.T) {
+	spec := map[string]string{
+		"a":    "a",
+		"c\nc": "cc",
+		"d":    "d",
+	}
+	expected := []string{
+		`a: a
+"c\nc": "cc"
+d: d
+`,
+		`---
+a: a
+? |-
+  c
+  c
+: cc
+d: d
+`}
+	y, err := yaml.Marshal(spec)
+	if err != nil {
+		t.Fatalf("Error marshalling YAML: %v", err)
+	}
+	for _, exp := range expected {
+		if string(y) == exp {
+			return
+		}
+	}
+
+	t.Fatalf("testNewlineEncode FAILED\nExpected: '%s'\nGot: '%s'", strings.Join(expected, "\nor\n"), string(y))
+}