mirror of https://github.com/mikefarah/yq.git
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
var flattenOperatorScenarios = []expressionScenario{
|
|
{
|
|
description: "Flatten",
|
|
subdescription: "Recursively flattens all arrays",
|
|
document: `[1, [2], [[3]]]`,
|
|
expression: `flatten`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::[1, 2, 3]\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Flatten splat",
|
|
skipDoc: true,
|
|
document: `[1, [2], [[3]]]`,
|
|
expression: `flatten[]`,
|
|
expected: []string{
|
|
"D0, P[0], (!!int)::1\n",
|
|
"D0, P[0], (!!int)::2\n",
|
|
"D0, P[0], (!!int)::3\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Flatten with depth of one",
|
|
document: `[1, [2], [[3]]]`,
|
|
expression: `flatten(1)`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::[1, 2, [3]]\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Flatten with depth and splat",
|
|
skipDoc: true,
|
|
document: `[1, [2], [[3]]]`,
|
|
expression: `flatten(1)[]`,
|
|
expected: []string{
|
|
"D0, P[0], (!!int)::1\n",
|
|
"D0, P[0], (!!int)::2\n",
|
|
"D0, P[0], (!!seq)::[3]\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Flatten empty array",
|
|
document: `[[]]`,
|
|
expression: `flatten`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::[]\n",
|
|
},
|
|
},
|
|
{
|
|
description: "Flatten array of objects",
|
|
document: `[{foo: bar}, [{foo: baz}]]`,
|
|
expression: `flatten`,
|
|
expected: []string{
|
|
"D0, P[], (!!seq)::[{foo: bar}, {foo: baz}]\n",
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestFlattenOperatorScenarios(t *testing.T) {
|
|
for _, tt := range flattenOperatorScenarios {
|
|
testScenario(t, &tt)
|
|
}
|
|
documentOperatorScenarios(t, "flatten", flattenOperatorScenarios)
|
|
}
|