yq/pkg/yqlib/operator_length_test.go

package yqlib

import (
	"testing"
)

var lengthOperatorScenarios = []expressionScenario{
	{
		description:    "String length",
		subdescription: "returns length of string",
		document:       `{a: cat}`,
		expression:     `.a | length`,
		expected: []string{
			"D0, P[a], (!!int)::3\n",
		},
	},
	{
		description: "null length",
		document:    `{a: null}`,
		expression:  `.a | length`,
		expected: []string{
			"D0, P[a], (!!int)::0\n",
		},
	},
	{
		skipDoc:    true,
		document:   `{a: ~}`,
		expression: `.a | length`,
		expected: []string{
			"D0, P[a], (!!int)::0\n",
		},
	},
	{
		skipDoc:    true,
		document:   "# abc\n{a: key no exist}",
		expression: `.b | length`,
		expected: []string{
			"D0, P[b], (!!int)::0\n",
		},
	},
	{
		description:    "Map length",
		subdescription: "returns number of entries",
		document:       `{a: cat, c: dog}`,
		expression:     `length`,
		expected: []string{
			"D0, P[], (!!int)::2\n",
		},
	},
	{
		description:    "Array length",
		subdescription: "returns number of elements",
		document:       `[2,4,6,8]`,
		expression:     `length`,
		expected: []string{
			"D0, P[], (!!int)::4\n",
		},
	},
}

func TestLengthOperatorScenarios(t *testing.T) {
	for _, tt := range lengthOperatorScenarios {
		testScenario(t, &tt)
	}
	documentOperatorScenarios(t, "length", lengthOperatorScenarios)
}