Function TestUnaryOperation_extractConditions

pkg/traceql/ast_conditions_test.go:422–482

(t *testing.T)

}

func TestUnaryOperation_extractConditions(t *testing.T) {
	tests := []struct {
		query         string
		conditions    []Condition
		allConditions bool
	}{
		{
			query: `{ span.foo != nil }`,
			conditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "foo"), OpExists),
			},
			allConditions: true,
		},
		{
			query: `{ span.foo = nil }`,
			conditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "foo"), OpNotExists),
			},
			allConditions: true,
		},
		{
			query: `{ span.foo }`,
			conditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "foo"), OpNone),
			},
			allConditions: true,
		},
		{
			query: `{ !span.foo }`,
			conditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "foo"), OpNone),
			},
			allConditions: true,
		},
		{
			query: `{ !(span.a = "a" || span.a = "c") }`,
			conditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "a"), OpEqual, NewStaticString("a")),
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "a"), OpEqual, NewStaticString("c")),
			},
			allConditions: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.query, func(t *testing.T) {
			expr, err := Parse(tt.query)
			require.NoError(t, err)

			req := &FetchSpansRequest{
				Conditions:    []Condition{},
				AllConditions: true,
			}
			expr.extractConditions(req)

			assert.Equal(t, tt.conditions, req.Conditions)
			assert.Nil(t, req.SecondPassConditions)
			assert.Equal(t, tt.allConditions, req.AllConditions, "FetchSpansRequest.AllConditions")