yq/pkg/yqlib/operator_anchors_aliases_test.go

package yqlib

import (
	"testing"
)

var specDocument = `- &CENTER { x: 1, y: 2 }
- &LEFT { x: 0, y: 2 }
- &BIG { r: 10 }
- &SMALL { r: 1 }
`

var expectedSpecResult = "D0, P[4], (!!map)::x: 1\ny: 2\nr: 10\n"

var simpleArrayRef = `item_value: &item_value
  value: true

thingOne:
  name: item_1
  <<: *item_value

thingTwo:
  name: item_2
  <<: *item_value
`

var expectedUpdatedArrayRef = `D0, P[], (doc)::item_value: &item_value
    value: true
thingOne:
    name: item_1
    value: false
thingTwo:
    name: item_2
    !!merge <<: *item_value
`

var anchorOperatorScenarios = []expressionScenario{
	// 	{
	// 		skipDoc:       true,
	// 		description:   "merge anchor not map",
	// 		document:      "a: &a\n  - 0\nc:\n  <<: [*a]\n",
	// 		expectedError: "merge anchor only supports maps, got !!seq instead",
	// 		expression:    "explode(.)",
	// 	},
	// 	{
	// 		description:    "Merge one map",
	// 		subdescription: "see https://yaml.org/type/merge.html",
	// 		document:       specDocument + "- << : *CENTER\n  r: 10\n",
	// 		expression:     ".[4] | explode(.)",
	// 		expected:       []string{expectedSpecResult},
	// 	},
	// 	{
	// 		description:    "Merge multiple maps",
	// 		subdescription: "see https://yaml.org/type/merge.html",
	// 		document:       specDocument + "- << : [ *CENTER, *BIG ]\n",
	// 		expression:     ".[4] | explode(.)",
	// 		expected:       []string{"D0, P[4], (!!map)::r: 10\nx: 1\ny: 2\n"},
	// 	},
	// 	{
	// 		description:    "Override",
	// 		subdescription: "see https://yaml.org/type/merge.html",
	// 		document:       specDocument + "- << : [ *BIG, *LEFT, *SMALL ]\n  x: 1\n",
	// 		expression:     ".[4] | explode(.)",
	// 		expected:       []string{"D0, P[4], (!!map)::r: 10\nx: 1\ny: 2\n"},
	// 	},
	// 	{
	// 		description: "Get anchor",
	// 		document:    `a: &billyBob cat`,
	// 		expression:  `.a | anchor`,
	// 		expected: []string{
	// 			"D0, P[a], (!!str)::billyBob\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Set anchor",
	// 		document:    `a: cat`,
	// 		expression:  `.a anchor = "foobar"`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::a: &foobar cat\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Set anchor relatively using assign-update",
	// 		document:    `a: {b: cat}`,
	// 		expression:  `.a anchor |= .b`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::a: &cat {b: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   `a: {c: cat}`,
	// 		expression: `.a anchor |= .b`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::a: {c: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   `a: {c: cat}`,
	// 		expression: `.a anchor = .b`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::a: {c: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Get alias",
	// 		document:    `{b: &billyBob meow, a: *billyBob}`,
	// 		expression:  `.a | alias`,
	// 		expected: []string{
	// 			"D0, P[a], (!!str)::billyBob\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Set alias",
	// 		document:    `{b: &meow purr, a: cat}`,
	// 		expression:  `.a alias = "meow"`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{b: &meow purr, a: *meow}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Set alias to blank does nothing",
	// 		document:    `{b: &meow purr, a: cat}`,
	// 		expression:  `.a alias = ""`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{b: &meow purr, a: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   `{b: &meow purr, a: cat}`,
	// 		expression: `.a alias = .c`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{b: &meow purr, a: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   `{b: &meow purr, a: cat}`,
	// 		expression: `.a alias |= .c`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{b: &meow purr, a: cat}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Set alias relatively using assign-update",
	// 		document:    `{b: &meow purr, a: {f: meow}}`,
	// 		expression:  `.a alias |= .f`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{b: &meow purr, a: *meow}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Explode alias and anchor",
	// 		document:    `{f : {a: &a cat, b: *a}}`,
	// 		expression:  `explode(.f)`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{f: {a: cat, b: cat}}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Explode with no aliases or anchors",
	// 		document:    `a: mike`,
	// 		expression:  `explode(.a)`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::a: mike\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Explode with alias keys",
	// 		document:    `{f : {a: &a cat, *a: b}}`,
	// 		expression:  `explode(.f)`,
	// 		expected: []string{
	// 			"D0, P[], (doc)::{f: {a: cat, cat: b}}\n",
	// 		},
	// 	},
	// 	{
	// 		description: "Explode with merge anchors",
	// 		document:    mergeDocSample,
	// 		expression:  `explode(.)`,
	// 		expected: []string{`D0, P[], (doc)::foo:
	//     a: foo_a
	//     thing: foo_thing
	//     c: foo_c
	// bar:
	//     b: bar_b
	//     thing: bar_thing
	//     c: bar_c
	// foobarList:
	//     b: bar_b
	//     thing: foo_thing
	//     c: foobarList_c
	//     a: foo_a
	// foobar:
	//     c: foo_c
	//     a: foo_a
	//     thing: foobar_thing
	// `},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   mergeDocSample,
	// 		expression: `.foo* | explode(.) | (. style="flow")`,
	// 		expected: []string{
	// 			"D0, P[foo], (!!map)::{a: foo_a, thing: foo_thing, c: foo_c}\n",
	// 			"D0, P[foobarList], (!!map)::{b: bar_b, thing: foo_thing, c: foobarList_c, a: foo_a}\n",
	// 			"D0, P[foobar], (!!map)::{c: foo_c, a: foo_a, thing: foobar_thing}\n",
	// 		},
	// 	},
	// 	{
	// 		skipDoc:    true,
	// 		document:   mergeDocSample,
	// 		expression: `.foo* | explode(explode(.)) | (. style="flow")`,
	// 		expected: []string{
	// 			"D0, P[foo], (!!map)::{a: foo_a, thing: foo_thing, c: foo_c}\n",
	// 			"D0, P[foobarList], (!!map)::{b: bar_b, thing: foo_thing, c: foobarList_c, a: foo_a}\n",
	// 			"D0, P[foobar], (!!map)::{c: foo_c, a: foo_a, thing: foobar_thing}\n",
	// 		},
	// 	},
	// {
	// 	skipDoc:    true,
	// 	document:   `{f : {a: &a cat, b: &b {foo: *a}, *a: *b}}`,
	// 	expression: `explode(.f)`,
	// 	expected: []string{
	// 		"D0, P[], (doc)::{f: {a: cat, b: {foo: cat}, cat: {foo: cat}}}\n",
	// 	},
	// },
	{
		description:    "Dereference and update a field",
		subdescription: "Use explode with multiply to dereference an object",
		document:       simpleArrayRef,
		expression:     `.thingOne |= explode(.) * {"value": false}`,
		expected:       []string{expectedUpdatedArrayRef},
	},
}

func TestAnchorAliasOperatorScenarios(t *testing.T) {
	for _, tt := range anchorOperatorScenarios {
		testScenario(t, &tt)
	}
	documentOperatorScenarios(t, "anchor-and-alias-operators", anchorOperatorScenarios)
}