test coverage and linting

pkg/marshal/json_converter.go

@@ -12,7 +12,7 @@ type JsonConverter interface {
 	JsonToString(context interface{}) (string, error)
 }
 
-type jsonConverter struct {}
+type jsonConverter struct{}
 
 func NewJsonConverter() JsonConverter {
 	return &jsonConverter{}

pkg/marshal/json_converter_test.go

@@ -2,6 +2,7 @@ package marshal
 
 import (
 	"testing"
+
 	"github.com/mikefarah/yq/test"
 )
 

pkg/marshal/yaml_converter.go

@@ -1,16 +1,17 @@
 package marshal
 
 import (
+	"strings"
+
 	yaml "github.com/mikefarah/yaml/v2"
 	errors "github.com/pkg/errors"
-	"strings"
 )
 
 type YamlConverter interface {
 	YamlToString(context interface{}, trimOutput bool) (string, error)
 }
 
-type yamlConverter struct {}
+type yamlConverter struct{}
 
 func NewYamlConverter() YamlConverter {
 	return &yamlConverter{}
@@ -39,4 +40,4 @@ func (y *yamlConverter) marshalContext(context interface{}, trimOutput bool) (st
 		return strings.Trim(outStr, "\n "), nil
 	}
 	return outStr, nil
-}
+}

pkg/marshal/yaml_converter_test.go

@@ -0,0 +1,52 @@
+package marshal
+
+import (
+	"testing"
+
+	"github.com/mikefarah/yq/test"
+)
+
+func TestYamlToString(t *testing.T) {
+	var raw = `b:
+  c: 2
+`
+	var data = test.ParseData(raw)
+	got, _ := NewYamlConverter().YamlToString(data, false)
+	test.AssertResult(t, raw, got)
+}
+
+func TestYamlToString_withTrim(t *testing.T) {
+	var raw = `b:
+  c: 2`
+	var data = test.ParseData(raw)
+	got, _ := NewYamlConverter().YamlToString(data, true)
+	test.AssertResult(t, raw, got)
+}
+
+func TestYamlToString_withIntKey(t *testing.T) {
+	var raw = `b:
+  2: c
+`
+	var data = test.ParseData(raw)
+	got, _ := NewYamlConverter().YamlToString(data, false)
+	test.AssertResult(t, raw, got)
+}
+
+func TestYamlToString_withBoolKey(t *testing.T) {
+	var raw = `b:
+  false: c
+`
+	var data = test.ParseData(raw)
+	got, _ := NewYamlConverter().YamlToString(data, false)
+	test.AssertResult(t, raw, got)
+}
+
+func TestYamlToString_withArray(t *testing.T) {
+	var raw = `b:
+- item: one
+- item: two
+`
+	var data = test.ParseData(raw)
+	got, _ := NewYamlConverter().YamlToString(data, false)
+	test.AssertResult(t, raw, got)
+}