Function TestSelect_extractConditions

pkg/traceql/ast_conditions_test.go:254–312

(t *testing.T)

}

func TestSelect_extractConditions(t *testing.T) {
	tests := []struct {
		query                string
		conditions           []Condition
		secondPassConditions []Condition
		allConditions        bool
	}{
		{
			query: `{ .foo = "a" } | select(resource.service.name)`,
			conditions: []Condition{
				newCondition(NewAttribute("foo"), OpEqual, NewStaticString("a")),
			},
			secondPassConditions: []Condition{
				newCondition(NewScopedAttribute(AttributeScopeResource, false, "service.name"), OpNone),
			},
			allConditions: true,
		},
		{
			query: `{ } | select(.name,name)`,
			conditions: []Condition{
				newCondition(NewIntrinsic(IntrinsicSpanStartTime), OpNone),
			},
			secondPassConditions: []Condition{
				newCondition(NewAttribute("name"), OpNone),
				newCondition(NewIntrinsic(IntrinsicName), OpNone),
			},
			allConditions: true,
		},
		{
			// Pipleline elements after a select are always directed to the second pass
			query: `{ } | select(span.foo) | { span.foo = "a" && span.bar = "b"}`,
			conditions: []Condition{
				newCondition(NewIntrinsic(IntrinsicSpanStartTime), OpNone),
			},
			secondPassConditions: []Condition{
				// span.foo=a has no effect because it's already covered by the select statement
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "foo"), OpNone),
				newCondition(NewScopedAttribute(AttributeScopeSpan, false, "bar"), OpEqual, NewStaticString("b")),
			},
			allConditions: true,
		},
	}
	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.Pipeline.extractConditions(req)

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