mirror of https://github.com/mikefarah/yq.git
127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
var entriesOperatorScenarios = []expressionScenario{
|
|
{
|
|
description: "to_entries splat",
|
|
skipDoc: true,
|
|
document: `{a: 1, b: 2}`,
|
|
expression: `to_entries[]`,
|
|
expected: []string{
|
|
"D0, P[0], (!!map)::key: a\nvalue: 1\n",
|
|
"D0, P[1], (!!map)::key: b\nvalue: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "to_entries, delete key",
|
|
skipDoc: true,
|
|
document: `{a: 1, b: 2}`,
|
|
expression: `to_entries | map(del(.key))`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::- value: 1\n- value: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "to_entries Map",
|
|
document: `{a: 1, b: 2}`,
|
|
expression: `to_entries`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::- key: a\n value: 1\n- key: b\n value: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "to_entries Array",
|
|
document: `[a, b]`,
|
|
expression: `to_entries`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::- key: 0\n value: a\n- key: 1\n value: b\n",
|
|
},
|
|
},
|
|
{
|
|
description: "to_entries null",
|
|
document: `null`,
|
|
expression: `to_entries`,
|
|
expected: []string{},
|
|
},
|
|
{
|
|
description: "from_entries map",
|
|
document: `{a: 1, b: 2}`,
|
|
expression: `to_entries | from_entries`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::a: 1\nb: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "from_entries with numeric key indices",
|
|
subdescription: "from_entries always creates a map, even for numeric keys",
|
|
document: `[a,b]`,
|
|
expression: `to_entries | from_entries`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::0: a\n1: b\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Use with_entries to update keys",
|
|
document: `{a: 1, b: 2}`,
|
|
expression: `with_entries(.key |= "KEY_" + .)`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::KEY_a: 1\nKEY_b: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Custom sort map keys",
|
|
subdescription: "Use to_entries to convert to an array of key/value pairs, sort the array using sort/sort_by/etc, and convert it back.",
|
|
document: `{a: 1, c: 3, b: 2}`,
|
|
expression: `to_entries | sort_by(.key) | reverse | from_entries`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::c: 3\nb: 2\na: 1\n",
|
|
},
|
|
},
|
|
{
|
|
skipDoc: true,
|
|
document: `{a: 1, b: 2}`,
|
|
document2: `{c: 1, d: 2}`,
|
|
expression: `with_entries(.key |= "KEY_" + .)`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::KEY_a: 1\nKEY_b: 2\n",
|
|
"D0, P[], (!!map)::KEY_c: 1\nKEY_d: 2\n",
|
|
},
|
|
},
|
|
{
|
|
skipDoc: true,
|
|
document: `[{a: 1, b: 2}, {c: 1, d: 2}]`,
|
|
expression: `.[] | with_entries(.key |= "KEY_" + .)`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::KEY_a: 1\nKEY_b: 2\n",
|
|
"D0, P[], (!!map)::KEY_c: 1\nKEY_d: 2\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Use with_entries to filter the map",
|
|
document: `{a: { b: bird }, c: { d: dog }}`,
|
|
expression: `with_entries(select(.value | has("b")))`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::a: {b: bird}\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Use with_entries to filter the map; head comment",
|
|
skipDoc: true,
|
|
document: "# abc\n{a: { b: bird }, c: { d: dog }}\n# xyz",
|
|
expression: `with_entries(select(.value | has("b")))`,
|
|
expected: []string{
|
|
"D0, P[], (!!map)::# abc\na: {b: bird}\n# xyz\n",
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestEntriesOperatorScenarios(t *testing.T) {
|
|
for _, tt := range entriesOperatorScenarios {
|
|
testScenario(t, &tt)
|
|
}
|
|
documentOperatorScenarios(t, "entries", entriesOperatorScenarios)
|
|
}
|