Added encoder tests

pkg/yqlib/encoder_csv.go

@@ -18,16 +18,10 @@ func NewCsvEncoder(destination io.Writer, separator rune) Encoder {
 	return &csvEncoder{csvWriter}
 }
 
-func (e *csvEncoder) Encode(originalNode *yaml.Node) error {
-	// node must be a sequence
-	node := unwrapDoc(originalNode)
-	if node.Kind != yaml.SequenceNode {
-		return fmt.Errorf("csv encoding only works for arrays of scalars (string/numbers/booleans), got: %v", node.Tag)
-	}
+func (e *csvEncoder) encodeRow(contents []*yaml.Node) error {
+	stringValues := make([]string, len(contents))
 
-	stringValues := make([]string, len(node.Content))
-
-	for i, child := range node.Content {
+	for i, child := range contents {
 
 		if child.Kind != yaml.ScalarNode {
 			return fmt.Errorf("csv encoding only works for arrays of scalars (string/numbers/booleans), child[%v] is a %v", i, child.Tag)
@@ -36,3 +30,28 @@ func (e *csvEncoder) Encode(originalNode *yaml.Node) error {
 	}
 	return e.destination.Write(stringValues)
 }
+
+func (e *csvEncoder) Encode(originalNode *yaml.Node) error {
+	// node must be a sequence
+	node := unwrapDoc(originalNode)
+	if node.Kind != yaml.SequenceNode {
+		return fmt.Errorf("csv encoding only works for arrays, got: %v", node.Tag)
+	} else if len(node.Content) == 0 {
+		return nil
+	}
+	if node.Content[0].Kind == yaml.ScalarNode {
+		return e.encodeRow(node.Content)
+	}
+
+	for i, child := range node.Content {
+
+		if child.Kind != yaml.SequenceNode {
+			return fmt.Errorf("csv encoding only works for arrays of scalars (string/numbers/booleans), child[%v] is a %v", i, child.Tag)
+		}
+		err := e.encodeRow(child.Content)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}