mirror of
https://github.com/mikefarah/yq.git
synced 2024-12-19 20:19:04 +00:00
Added encoder tests
This commit is contained in:
parent
7696aeb560
commit
d0c3da1262
47
pkg/yqlib/encoder_csv_test.go
Normal file
47
pkg/yqlib/encoder_csv_test.go
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
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(writer, separator)
|
||||||
|
inputs, err := readDocuments(strings.NewReader(sampleYaml), "sample.yml", 0)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
node := inputs.Front().Value.(*CandidateNode).Node
|
||||||
|
err = jsonEncoder.Encode(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"]`
|
||||||
|
|
||||||
|
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 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)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user