yq/pkg/yqlib/operator_entries_test.go

108 lines
2.8 KiB
Go
Raw Normal View History

2021-05-09 03:59:23 +00:00
package yqlib
import (
"testing"
)
var entriesOperatorScenarios = []expressionScenario{
{
description: "to_entries Map",
2021-05-09 05:12:50 +00:00
document: `{a: 1, b: 2}`,
expression: `to_entries`,
2021-05-09 03:59:23 +00:00
expected: []string{
"D0, P[], (!!seq)::- key: a\n value: 1\n- key: b\n value: 2\n",
},
},
{
description: "to_entries Array",
2021-05-09 05:12:50 +00:00
document: `[a, b]`,
expression: `to_entries`,
2021-05-09 03:59:23 +00:00
expected: []string{
"D0, P[], (!!seq)::- key: 0\n value: a\n- key: 1\n value: b\n",
},
},
2021-05-10 00:42:43 +00:00
{
description: "to_entries null",
document: `null`,
expression: `to_entries`,
2021-05-14 05:01:44 +00:00
expected: []string{},
2021-05-10 00:42:43 +00:00
},
2021-05-09 04:18:25 +00:00
{
description: "from_entries map",
2021-05-09 05:12:50 +00:00
document: `{a: 1, b: 2}`,
expression: `to_entries | from_entries`,
2021-05-09 04:18:25 +00:00
expected: []string{
"D0, P[], (!!map)::a: 1\nb: 2\n",
},
},
{
description: "from_entries with numeric key indices",
2021-05-09 04:18:25 +00:00
subdescription: "from_entries always creates a map, even for numeric keys",
2021-05-09 05:12:50 +00:00
document: `[a,b]`,
expression: `to_entries | from_entries`,
2021-05-09 04:18:25 +00:00
expected: []string{
"D0, P[], (!!map)::0: a\n1: b\n",
},
},
2021-05-09 05:12:50 +00:00
{
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",
},
},
2022-10-17 04:03:47 +00:00
{
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",
},
},
{
2021-06-08 11:50:14 +00:00
description: "Use with_entries to filter the map",
document: `{a: { b: bird }, c: { d: dog }}`,
expression: `with_entries(select(.value | has("b")))`,
expected: []string{
2021-06-08 11:50:14 +00:00
"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",
},
},
2021-05-09 03:59:23 +00:00
}
func TestEntriesOperatorScenarios(t *testing.T) {
for _, tt := range entriesOperatorScenarios {
testScenario(t, &tt)
}
2021-12-21 04:02:07 +00:00
documentOperatorScenarios(t, "entries", entriesOperatorScenarios)
2021-05-09 03:59:23 +00:00
}