Function TestFlattenExprToOperations

pkg/traceql/lenient_extract_test.go:224–297

(t *testing.T)

}

func TestFlattenExprToOperations(t *testing.T) {
	type testOperation struct {
		opType Operator
		cond   []int
	}

	testCases := []struct {
		name, query string
		expected    []testOperation
	}{
		{
			name:     "three single condition ORs",
			query:    `{ .attr = "123" || .service = "b" || .env = "staging" }`,
			expected: []testOperation{{opType: OpOr, cond: []int{1, 1, 1}}},
		},
		{
			name:     "one double condition ORs with two single condition ORs",
			query:    `{ (.attr = "123" && .foo = "bar") || .service = "b" || .env = "staging" }`,
			expected: []testOperation{{opType: OpOr, cond: []int{2, 1, 1}}},
		},
		{
			name:     "two duplicated OR conditions",
			query:    `{ (.attr = "123" && .foo = "bar") || .service = "b" || .env = "staging" || (.attr = "123" && .foo = "bar") || .service = "b" }`,
			expected: []testOperation{{opType: OpOr, cond: []int{2, 1, 1}}},
		},
		{
			name:     "one single condition AND three single condition ORs",
			query:    `{ name = "abc" && (.attr = "123" || .service = "b" || .env = "staging") }`,
			expected: []testOperation{{opType: OpAnd, cond: []int{1}}, {opType: OpOr, cond: []int{1, 1, 1}}},
		},
		{
			name:     "one AND one double condition OR with two single condition ORs",
			query:    `{ name = "abc" && ( (.attr = "123" && .foo = "bar") || .service = "b" || .env = "staging") }`,
			expected: []testOperation{{opType: OpAnd, cond: []int{1}}, {opType: OpOr, cond: []int{2, 1, 1}}},
		},
		{
			name:     "two ANDs one double condition OR with two single condition ORs",
			query:    `{ name = "abc" && .attr = "abc" && ( (.attr = "123" && .foo = "bar") || .service = "b" || .env = "staging") }`,
			expected: []testOperation{{opType: OpAnd, cond: []int{1}}, {opType: OpAnd, cond: []int{1}}, {opType: OpOr, cond: []int{2, 1, 1}}},
		},
		{
			name:     "two AND with single condition OR in between",
			query:    `{ .attr = "123" && (.service = "b" || .service = "a" ) && .env = "staging" }`,
			expected: []testOperation{{opType: OpAnd, cond: []int{1}}, {opType: OpOr, cond: []int{1, 1}}, {opType: OpAnd, cond: []int{1}}},
		},
		{
			name:     "two ANDs of single condition ORs",
			query:    `{ ( .attr = "123" || .service = "b" ) && ( .service = "a" || .env = "staging" ) }`,
			expected: []testOperation{{opType: OpOr, cond: []int{1, 1}}, {opType: OpOr, cond: []int{1, 1}}},
		},
		{
			name:     "two ANDs of multiple conditions ORs",
			query:    `{ ( .attr = "123" || .service = "b" ) && ( .service = "a" || ( .env = "staging" && .foo = "bar" ) ) }`,
			expected: []testOperation{{opType: OpOr, cond: []int{1, 1}}, {opType: OpOr, cond: []int{1, 2}}},
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {