operator_filter_test.go

package yqlib

import (
	"testing"
)

var filterOperatorScenarios = []expressionScenario{
	{
		description: "Filter array",
		document:   `[1,2,3]`,
		expression: `filter(. < 3)`,
		expected: []string{
			"D0, P[], (!!seq)::- 1\n",
			"D0, P[], (!!seq)::- 2\n",
		},
	},
	{
		skipDoc: true,
		document:    `[1,2,3]`,
		expression:  `filter(. > 1)`,
		expected: []string{
			"D0, P[], (!!seq)::- 2\n",
			"D0, P[], (!!seq)::- 3\n",
		},
	},
	{
		skipDoc: true,
		description: "Filter array to empty",
		document:    `[1,2,3]`,
		expression:  `filter(. > 4)`,
		expected: []string{
		},
	},
	{
		skipDoc: true,
		description: "Filter empty array",
		document:    `[]`,
		expression:  `filter(. > 1)`,
		expected: []string{
		},
	},
}

func TestFilterOperatorScenarios(t *testing.T) {
	for _, tt := range filterOperatorScenarios {
		testScenario(t, &tt)
	}
	documentOperatorScenarios(t, "filter", filterOperatorScenarios)
}