mirror of https://github.com/mikefarah/yq.git
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mikefarah/yq/v4/test"
|
|
)
|
|
|
|
func yamlToCsv(sampleYaml string, separator rune) string {
|
|
var output bytes.Buffer
|
|
writer := bufio.NewWriter(&output)
|
|
|
|
var jsonEncoder = NewCsvEncoder(separator)
|
|
inputs, err := readDocuments(strings.NewReader(sampleYaml), "sample.yml", 0, NewYamlDecoder())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
node := inputs.Front().Value.(*CandidateNode).Node
|
|
err = jsonEncoder.Encode(writer, node)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
writer.Flush()
|
|
|
|
return strings.TrimSuffix(output.String(), "\n")
|
|
}
|
|
|
|
var sampleYaml = `["apple", apple2, "comma, in, value", "new
|
|
line", 3, 3.40, true, "tab here"]`
|
|
|
|
var sampleYamlArray = "[" + sampleYaml + ", [bob, cat, meow, puss]]"
|
|
|
|
func TestCsvEncoderEmptyArray(t *testing.T) {
|
|
var actualCsv = yamlToCsv(`[]`, ',')
|
|
test.AssertResult(t, "", actualCsv)
|
|
}
|
|
|
|
func TestCsvEncoder(t *testing.T) {
|
|
var expectedCsv = `apple,apple2,"comma, in, value",new line,3,3.40,true,tab here`
|
|
|
|
var actualCsv = yamlToCsv(sampleYaml, ',')
|
|
test.AssertResult(t, expectedCsv, actualCsv)
|
|
}
|
|
|
|
func TestCsvEncoderArrayOfArrays(t *testing.T) {
|
|
var actualCsv = yamlToCsv(sampleYamlArray, ',')
|
|
var expectedCsv = "apple,apple2,\"comma, in, value\",new line,3,3.40,true,tab here\nbob,cat,meow,puss"
|
|
test.AssertResult(t, expectedCsv, actualCsv)
|
|
}
|
|
|
|
func TestTsvEncoder(t *testing.T) {
|
|
|
|
var expectedCsv = `apple apple2 comma, in, value new line 3 3.40 true "tab here"`
|
|
|
|
var actualCsv = yamlToCsv(sampleYaml, '\t')
|
|
test.AssertResult(t, expectedCsv, actualCsv)
|
|
}
|