mirror of
https://github.com/mikefarah/yq.git
synced 2024-12-19 20:19:04 +00:00
384 lines
11 KiB
Go
384 lines
11 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/mikefarah/yq/v4/test"
|
|
)
|
|
|
|
const csvSimple = `name,numberOfCats,likesApples,height
|
|
Gary,1,true,168.8
|
|
Samantha's Rabbit,2,false,-188.8
|
|
`
|
|
|
|
const csvSimpleWithObject = `name,numberOfCats,likesApples,height,facts
|
|
Gary,1,true,168.8,cool: true
|
|
Samantha's Rabbit,2,false,-188.8,tall: indeed
|
|
`
|
|
const csvMissing = `name,numberOfCats,likesApples,height
|
|
,null,,168.8
|
|
`
|
|
const expectedUpdatedSimpleCsv = `name,numberOfCats,likesApples,height
|
|
Gary,3,true,168.8
|
|
Samantha's Rabbit,2,false,-188.8
|
|
`
|
|
|
|
const csvSimpleShort = `Name,Number of Cats
|
|
Gary,1
|
|
Samantha's Rabbit,2
|
|
`
|
|
|
|
const tsvSimple = `name numberOfCats likesApples height
|
|
Gary 1 true 168.8
|
|
Samantha's Rabbit 2 false -188.8
|
|
`
|
|
|
|
const expectedYamlFromCSV = `- name: Gary
|
|
numberOfCats: 1
|
|
likesApples: true
|
|
height: 168.8
|
|
- name: Samantha's Rabbit
|
|
numberOfCats: 2
|
|
likesApples: false
|
|
height: -188.8
|
|
`
|
|
const expectedYamlFromCSVWithObject = `- name: Gary
|
|
numberOfCats: 1
|
|
likesApples: true
|
|
height: 168.8
|
|
facts:
|
|
cool: true
|
|
- name: Samantha's Rabbit
|
|
numberOfCats: 2
|
|
likesApples: false
|
|
height: -188.8
|
|
facts:
|
|
tall: indeed
|
|
`
|
|
|
|
const expectedYamlFromCSVNoParsing = `- name: Gary
|
|
numberOfCats: 1
|
|
likesApples: true
|
|
height: 168.8
|
|
facts: 'cool: true'
|
|
- name: Samantha's Rabbit
|
|
numberOfCats: 2
|
|
likesApples: false
|
|
height: -188.8
|
|
facts: 'tall: indeed'
|
|
`
|
|
|
|
const expectedYamlFromCSVMissingData = `- name: Gary
|
|
numberOfCats: 1
|
|
height: 168.8
|
|
- name: Samantha's Rabbit
|
|
height: -188.8
|
|
likesApples: false
|
|
`
|
|
|
|
const csvSimpleMissingData = `name,numberOfCats,height
|
|
Gary,1,168.8
|
|
Samantha's Rabbit,,-188.8
|
|
`
|
|
|
|
const csvTestSimpleYaml = `- [i, like, csv]
|
|
- [because, excel, is, cool]`
|
|
|
|
const expectedSimpleCsv = `i,like,csv
|
|
because,excel,is,cool
|
|
`
|
|
|
|
const tsvTestExpectedSimpleCsv = `i like csv
|
|
because excel is cool
|
|
`
|
|
|
|
var csvScenarios = []FormatScenario{
|
|
{
|
|
description: "Encode CSV simple",
|
|
input: csvTestSimpleYaml,
|
|
expected: expectedSimpleCsv,
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "Encode TSV simple",
|
|
input: csvTestSimpleYaml,
|
|
expected: tsvTestExpectedSimpleCsv,
|
|
scenarioType: "encode-tsv",
|
|
},
|
|
{
|
|
description: "Encode Empty",
|
|
skipDoc: true,
|
|
input: `[]`,
|
|
expected: "",
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "Comma in value",
|
|
skipDoc: true,
|
|
input: `["comma, in, value", things]`,
|
|
expected: "\"comma, in, value\",things\n",
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "Encode array of objects to csv",
|
|
input: expectedYamlFromCSV,
|
|
expected: csvSimple,
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "Encode array of objects to custom csv format",
|
|
subdescription: "Add the header row manually, then the we convert each object into an array of values - resulting in an array of arrays. Pick the columns and call the header whatever you like.",
|
|
input: expectedYamlFromCSV,
|
|
expected: csvSimpleShort,
|
|
expression: `[["Name", "Number of Cats"]] + [.[] | [.name, .numberOfCats ]]`,
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "Encode array of objects to csv - missing fields behaviour",
|
|
subdescription: "First entry is used to determine the headers, and it is missing 'likesApples', so it is not included in the csv. Second entry does not have 'numberOfCats' so that is blank",
|
|
input: expectedYamlFromCSVMissingData,
|
|
expected: csvSimpleMissingData,
|
|
scenarioType: "encode-csv",
|
|
},
|
|
{
|
|
description: "decode csv missing",
|
|
skipDoc: true,
|
|
input: csvMissing,
|
|
expected: csvMissing,
|
|
scenarioType: "roundtrip-csv",
|
|
},
|
|
{
|
|
description: "decode csv key",
|
|
skipDoc: true,
|
|
input: csvSimple,
|
|
expression: ".[0].name | key",
|
|
expected: "name\n",
|
|
scenarioType: "decode-csv",
|
|
},
|
|
{
|
|
description: "decode csv parent",
|
|
skipDoc: true,
|
|
input: csvSimple,
|
|
expression: ".[0].name | parent | .height",
|
|
expected: "168.8\n",
|
|
scenarioType: "decode-csv",
|
|
},
|
|
{
|
|
description: "Parse CSV into an array of objects",
|
|
subdescription: "First row is assumed to be the header row. By default, entries with YAML/JSON formatting will be parsed!",
|
|
input: csvSimpleWithObject,
|
|
expected: expectedYamlFromCSVWithObject,
|
|
scenarioType: "decode-csv",
|
|
},
|
|
{
|
|
description: "Decode CSV line breaks",
|
|
skipDoc: true,
|
|
input: "heading1\n\"some data\nwith a line break\"\n",
|
|
expected: "- heading1: |-\n some data\n with a line break\n",
|
|
scenarioType: "decode-csv",
|
|
},
|
|
{
|
|
description: "Parse CSV into an array of objects, no auto-parsing",
|
|
subdescription: "First row is assumed to be the header row. Entries with YAML/JSON will be left as strings.",
|
|
input: csvSimpleWithObject,
|
|
expected: expectedYamlFromCSVNoParsing,
|
|
scenarioType: "decode-csv-no-auto",
|
|
},
|
|
{
|
|
description: "Scalar roundtrip",
|
|
skipDoc: true,
|
|
input: "mike\ncat",
|
|
expression: ".[0].mike",
|
|
expected: "cat\n",
|
|
scenarioType: "roundtrip-csv",
|
|
},
|
|
{
|
|
description: "Parse TSV into an array of objects",
|
|
subdescription: "First row is assumed to be the header row.",
|
|
input: tsvSimple,
|
|
expected: expectedYamlFromCSV,
|
|
scenarioType: "decode-tsv-object",
|
|
},
|
|
{
|
|
description: "Round trip",
|
|
input: csvSimple,
|
|
expected: expectedUpdatedSimpleCsv,
|
|
expression: `(.[] | select(.name == "Gary") | .numberOfCats) = 3`,
|
|
scenarioType: "roundtrip-csv",
|
|
},
|
|
}
|
|
|
|
func testCSVScenario(t *testing.T, s FormatScenario) {
|
|
switch s.scenarioType {
|
|
case "encode-csv":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewYamlDecoder(ConfiguredYamlPreferences), NewCsvEncoder(ConfiguredCsvPreferences)), s.description)
|
|
case "encode-tsv":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewYamlDecoder(ConfiguredYamlPreferences), NewCsvEncoder(ConfiguredTsvPreferences)), s.description)
|
|
case "decode-csv":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewCSVObjectDecoder(ConfiguredCsvPreferences), NewYamlEncoder(ConfiguredYamlPreferences)), s.description)
|
|
case "decode-csv-no-auto":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewCSVObjectDecoder(CsvPreferences{Separator: ',', AutoParse: false}), NewYamlEncoder(ConfiguredYamlPreferences)), s.description)
|
|
case "decode-tsv-object":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewCSVObjectDecoder(ConfiguredTsvPreferences), NewYamlEncoder(ConfiguredYamlPreferences)), s.description)
|
|
case "roundtrip-csv":
|
|
test.AssertResultWithContext(t, s.expected, mustProcessFormatScenario(s, NewCSVObjectDecoder(ConfiguredCsvPreferences), NewCsvEncoder(ConfiguredCsvPreferences)), s.description)
|
|
default:
|
|
panic(fmt.Sprintf("unhandled scenario type %q", s.scenarioType))
|
|
}
|
|
}
|
|
|
|
func documentCSVDecodeObjectScenario(w *bufio.Writer, s FormatScenario, formatType string) {
|
|
writeOrPanic(w, fmt.Sprintf("## %v\n", s.description))
|
|
|
|
if s.subdescription != "" {
|
|
writeOrPanic(w, s.subdescription)
|
|
writeOrPanic(w, "\n\n")
|
|
}
|
|
|
|
writeOrPanic(w, fmt.Sprintf("Given a sample.%v file of:\n", formatType))
|
|
writeOrPanic(w, fmt.Sprintf("```%v\n%v\n```\n", formatType, s.input))
|
|
|
|
writeOrPanic(w, "then\n")
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -p=%v sample.%v\n```\n", formatType, formatType))
|
|
writeOrPanic(w, "will output\n")
|
|
|
|
separator := ','
|
|
if formatType == "tsv" {
|
|
separator = '\t'
|
|
}
|
|
|
|
writeOrPanic(w, fmt.Sprintf("```yaml\n%v```\n\n",
|
|
mustProcessFormatScenario(s, NewCSVObjectDecoder(CsvPreferences{Separator: separator, AutoParse: true}), NewYamlEncoder(ConfiguredYamlPreferences))),
|
|
)
|
|
}
|
|
|
|
func documentCSVDecodeObjectNoAutoScenario(w *bufio.Writer, s FormatScenario, formatType string) {
|
|
writeOrPanic(w, fmt.Sprintf("## %v\n", s.description))
|
|
|
|
if s.subdescription != "" {
|
|
writeOrPanic(w, s.subdescription)
|
|
writeOrPanic(w, "\n\n")
|
|
}
|
|
|
|
writeOrPanic(w, fmt.Sprintf("Given a sample.%v file of:\n", formatType))
|
|
writeOrPanic(w, fmt.Sprintf("```%v\n%v\n```\n", formatType, s.input))
|
|
|
|
writeOrPanic(w, "then\n")
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -p=%v --csv-auto-parse=f sample.%v\n```\n", formatType, formatType))
|
|
writeOrPanic(w, "will output\n")
|
|
|
|
separator := ','
|
|
if formatType == "tsv" {
|
|
separator = '\t'
|
|
}
|
|
|
|
writeOrPanic(w, fmt.Sprintf("```yaml\n%v```\n\n",
|
|
mustProcessFormatScenario(s, NewCSVObjectDecoder(CsvPreferences{Separator: separator, AutoParse: false}), NewYamlEncoder(ConfiguredYamlPreferences))),
|
|
)
|
|
}
|
|
|
|
func documentCSVEncodeScenario(w *bufio.Writer, s FormatScenario, formatType string) {
|
|
writeOrPanic(w, fmt.Sprintf("## %v\n", s.description))
|
|
|
|
if s.subdescription != "" {
|
|
writeOrPanic(w, s.subdescription)
|
|
writeOrPanic(w, "\n\n")
|
|
}
|
|
|
|
writeOrPanic(w, "Given a sample.yml file of:\n")
|
|
writeOrPanic(w, fmt.Sprintf("```yaml\n%v\n```\n", s.input))
|
|
|
|
writeOrPanic(w, "then\n")
|
|
|
|
expression := s.expression
|
|
|
|
if expression != "" {
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -o=%v '%v' sample.yml\n```\n", formatType, expression))
|
|
} else {
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -o=%v sample.yml\n```\n", formatType))
|
|
}
|
|
writeOrPanic(w, "will output\n")
|
|
|
|
separator := ','
|
|
if formatType == "tsv" {
|
|
separator = '\t'
|
|
}
|
|
csvPrefs := NewDefaultCsvPreferences()
|
|
csvPrefs.Separator = separator
|
|
writeOrPanic(w, fmt.Sprintf("```%v\n%v```\n\n", formatType,
|
|
mustProcessFormatScenario(s, NewYamlDecoder(ConfiguredYamlPreferences), NewCsvEncoder(csvPrefs))),
|
|
)
|
|
}
|
|
|
|
func documentCSVRoundTripScenario(w *bufio.Writer, s FormatScenario, formatType string) {
|
|
writeOrPanic(w, fmt.Sprintf("## %v\n", s.description))
|
|
|
|
if s.subdescription != "" {
|
|
writeOrPanic(w, s.subdescription)
|
|
writeOrPanic(w, "\n\n")
|
|
}
|
|
|
|
writeOrPanic(w, fmt.Sprintf("Given a sample.%v file of:\n", formatType))
|
|
writeOrPanic(w, fmt.Sprintf("```%v\n%v\n```\n", formatType, s.input))
|
|
|
|
writeOrPanic(w, "then\n")
|
|
|
|
expression := s.expression
|
|
|
|
if expression != "" {
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -p=%v -o=%v '%v' sample.%v\n```\n", formatType, formatType, expression, formatType))
|
|
} else {
|
|
writeOrPanic(w, fmt.Sprintf("```bash\nyq -p=%v -o=%v sample.%v\n```\n", formatType, formatType, formatType))
|
|
}
|
|
writeOrPanic(w, "will output\n")
|
|
|
|
separator := ','
|
|
if formatType == "tsv" {
|
|
separator = '\t'
|
|
}
|
|
|
|
csvPrefs := NewDefaultCsvPreferences()
|
|
csvPrefs.Separator = separator
|
|
|
|
writeOrPanic(w, fmt.Sprintf("```%v\n%v```\n\n", formatType,
|
|
mustProcessFormatScenario(s, NewCSVObjectDecoder(CsvPreferences{Separator: separator, AutoParse: true}), NewCsvEncoder(csvPrefs))),
|
|
)
|
|
}
|
|
|
|
func documentCSVScenario(_ *testing.T, w *bufio.Writer, i interface{}) {
|
|
s := i.(FormatScenario)
|
|
if s.skipDoc {
|
|
return
|
|
}
|
|
switch s.scenarioType {
|
|
case "encode-csv":
|
|
documentCSVEncodeScenario(w, s, "csv")
|
|
case "encode-tsv":
|
|
documentCSVEncodeScenario(w, s, "tsv")
|
|
case "decode-csv":
|
|
documentCSVDecodeObjectScenario(w, s, "csv")
|
|
case "decode-csv-no-auto":
|
|
documentCSVDecodeObjectNoAutoScenario(w, s, "csv")
|
|
case "decode-tsv-object":
|
|
documentCSVDecodeObjectScenario(w, s, "tsv")
|
|
case "roundtrip-csv":
|
|
documentCSVRoundTripScenario(w, s, "csv")
|
|
|
|
default:
|
|
panic(fmt.Sprintf("unhandled scenario type %q", s.scenarioType))
|
|
}
|
|
}
|
|
|
|
func TestCSVScenarios(t *testing.T) {
|
|
for _, tt := range csvScenarios {
|
|
testCSVScenario(t, tt)
|
|
}
|
|
genericScenarios := make([]interface{}, len(csvScenarios))
|
|
for i, s := range csvScenarios {
|
|
genericScenarios[i] = s
|
|
}
|
|
documentScenarios(t, "usage", "csv-tsv", genericScenarios, documentCSVScenario)
|
|
}
|