yq/pkg/yqlib/operator_pick_test.go

61 lines
1.9 KiB
Go
Raw Normal View History

2022-03-09 03:38:02 +00:00
package yqlib
import (
"testing"
)
var pickOperatorScenarios = []expressionScenario{
{
description: "Pick keys from map",
subdescription: "Note that the order of the keys matches the pick order and non existent keys are skipped.",
document: "myMap: {cat: meow, dog: bark, thing: hamster, hamster: squeak}\n",
2022-03-09 03:38:02 +00:00
expression: `.myMap |= pick(["hamster", "cat", "goat"])`,
expected: []string{
"D0, P[], (!!map)::myMap: {hamster: squeak, cat: meow}\n",
2022-03-09 03:38:02 +00:00
},
},
2022-07-29 00:26:50 +00:00
{
description: "Pick keys from map",
skipDoc: true,
document: "!things myMap: {cat: meow, dog: bark, thing: hamster, hamster: squeak}\n",
2022-07-29 00:26:50 +00:00
expression: `.myMap |= pick(["hamster", "cat", "goat"])`,
expected: []string{
"D0, P[], (!!map)::!things myMap: {hamster: squeak, cat: meow}\n",
2022-07-29 00:26:50 +00:00
},
},
{
description: "Pick keys from map with comments",
skipDoc: true,
document: "# abc\nmyMap: {cat: meow, dog: bark, thing: hamster, hamster: squeak}\n# xyz\n",
expression: `.myMap |= pick(["hamster", "cat", "goat"])`,
expected: []string{
"D0, P[], (!!map)::# abc\nmyMap: {hamster: squeak, cat: meow}\n# xyz\n",
},
},
2022-03-09 03:38:02 +00:00
{
description: "Pick indices from array",
subdescription: "Note that the order of the indices matches the pick order and non existent indices are skipped.",
2022-03-09 03:38:02 +00:00
document: `[cat, leopard, lion]`,
expression: `pick([2, 0, 734, -5])`,
expected: []string{
"D0, P[], (!!seq)::[lion, cat]\n",
},
},
{
description: "Pick indices from array with comments",
skipDoc: true,
document: "# abc\n[cat, leopard, lion]\n# xyz",
expression: `pick([2, 0, 734, -5])`,
expected: []string{
"D0, P[], (!!seq)::# abc\n[lion, cat]\n# xyz\n",
},
},
2022-03-09 03:38:02 +00:00
}
func TestPickOperatorScenarios(t *testing.T) {
for _, tt := range pickOperatorScenarios {
testScenario(t, &tt)
}
documentOperatorScenarios(t, "pick", pickOperatorScenarios)
}