Also marshal Node by value

go-yaml · May 4, 2020 · a81ed60 · a81ed60
1 parent ae27a74
commit a81ed60
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/encode.go b/encode.go
@@ -119,6 +119,9 @@ func (e *encoder) marshal(tag string, in reflect.Value) {
 	case *Node:
 		e.nodev(in)
 		return
+	case Node:
+		e.nodev(in.Addr())
+		return
 	case time.Time:
 		e.timev(tag, in)
 		return

diff --git a/encode_test.go b/encode_test.go
@@ -428,6 +428,21 @@ var marshalTests = []struct {
 		map[string]string{"a": "\tB\n\tC\n"},
 		"a: |\n    \tB\n    \tC\n",
 	},
+
+	// yaml.Node
+	{
+		&struct {
+			Value yaml.Node
+		}{
+			yaml.Node{
+				Kind:  yaml.ScalarNode,
+				Tag:   "!!str",
+				Value: "foo",
+				Style: yaml.SingleQuotedStyle,
+			},
+		},
+		"value: 'foo'\n",
+	},
 }
 
 func (s *S) TestMarshal(c *C) {