mirror of
https://github.com/mikefarah/yq.git
synced 2025-01-12 19:25:37 +00:00
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/mikefarah/yq/v2/pkg/marshal"
|
|
"github.com/mikefarah/yq/v2/test"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TestMultilineString(t *testing.T) {
|
|
testString := `
|
|
abcd
|
|
efg`
|
|
formattedResult, _ := marshal.NewYamlConverter().YamlToString(testString, false)
|
|
test.AssertResult(t, testString, formattedResult)
|
|
}
|
|
|
|
func TestNewYaml(t *testing.T) {
|
|
result, _ := newYaml([]string{"b.c", "3"})
|
|
formattedResult := fmt.Sprintf("%v", result)
|
|
test.AssertResult(t,
|
|
"[{b [{c 3}]}]",
|
|
formattedResult)
|
|
}
|
|
|
|
func TestNewYamlArray(t *testing.T) {
|
|
result, _ := newYaml([]string{"[0].cat", "meow"})
|
|
formattedResult := fmt.Sprintf("%v", result)
|
|
test.AssertResult(t,
|
|
"[[{cat meow}]]",
|
|
formattedResult)
|
|
}
|
|
|
|
func TestNewYamlBigInt(t *testing.T) {
|
|
result, _ := newYaml([]string{"b", "1212121"})
|
|
formattedResult := fmt.Sprintf("%v", result)
|
|
test.AssertResult(t,
|
|
"[{b 1212121}]",
|
|
formattedResult)
|
|
}
|
|
|
|
func TestNewYaml_WithScript(t *testing.T) {
|
|
writeScript = "examples/instruction_sample.yaml"
|
|
expectedResult := `b:
|
|
c: cat
|
|
e:
|
|
- name: Mike Farah`
|
|
result, _ := newYaml([]string{""})
|
|
actualResult, _ := marshal.NewYamlConverter().YamlToString(result, true)
|
|
test.AssertResult(t, expectedResult, actualResult)
|
|
}
|
|
|
|
func TestNewYaml_WithUnknownScript(t *testing.T) {
|
|
writeScript = "fake-unknown"
|
|
_, err := newYaml([]string{""})
|
|
if err == nil {
|
|
t.Error("Expected error due to unknown file")
|
|
}
|
|
var expectedOutput string
|
|
if runtime.GOOS == "windows" {
|
|
expectedOutput = `open fake-unknown: The system cannot find the file specified.`
|
|
} else {
|
|
expectedOutput = `open fake-unknown: no such file or directory`
|
|
}
|
|
test.AssertResult(t, expectedOutput, err.Error())
|
|
}
|
|
|
|
func TestReadWithKeyOnly(t *testing.T) {
|
|
readCmd := createReadCmd()
|
|
expectedResult := `b
|
|
d
|
|
f
|
|
`
|
|
actualResult, err := executeTestCommand(readCmd, "test/fixture/keyonly.yaml", "a", "-k")
|
|
if err != nil {
|
|
t.Error(err.Error())
|
|
}
|
|
test.AssertResult(t, expectedResult, actualResult)
|
|
}
|
|
|
|
func executeTestCommand(command *cobra.Command, args ...string) (output string, err error) {
|
|
buf := new(bytes.Buffer)
|
|
|
|
command.SetOutput(buf)
|
|
command.SetArgs(args)
|
|
|
|
_, err = command.ExecuteC()
|
|
|
|
return buf.String(), err
|
|
}
|