Function TestBinaryAndUnaryOperationsRewrites

pkg/traceql/parse_test.go:1051–1083

TestBinaryAndUnaryOperationsRewrites tests code in the newBinaryOperation and newUnaryOperation functions that attempts to simplify combinations of static values where possible.

(t *testing.T)

// TestBinaryAndUnaryOperationsRewrites tests code in the newBinaryOperation and newUnaryOperation functions
// that attempts to simplify combinations of static values where possible.
func TestBinaryAndUnaryOperationsRewrites(t *testing.T) {
	tests := []struct {
		in       string
		expected FieldExpression
	}{
		// collapse to statics
		{in: "{ duration > 1 + 2}", expected: newBinaryOperation(OpGreater, NewIntrinsic(IntrinsicDuration), NewStaticInt(3))},
		{in: "{ -1 }", expected: NewStaticInt(-1)},
		{in: "{ 1 + 1 > 1 }", expected: NewStaticBool(true)},
		{in: "{ `foo` = `bar` }", expected: NewStaticBool(false)},
		{in: "{ 1 = 1. }", expected: NewStaticBool(true)}, // this is an interesting case, it returns true even though { span.foo = 1 } would be false if span.foo had the float value 1.0
		{in: "{ .1 + 1 }", expected: NewStaticFloat(1.1)},
		{in: "{ 1 * -1 = -1 }", expected: NewStaticBool(true)},
		{in: "{ .foo * -1. = -1 }", expected: newBinaryOperation(OpEqual, newBinaryOperation(OpMult, NewAttribute("foo"), NewStaticFloat(-1)), NewStaticInt(-1))},
		// rewrite != nil to existence
		{in: "{ .foo != nil }", expected: newUnaryOperation(OpExists, NewAttribute("foo"))},
		{in: "{ .foo = nil }", expected: newUnaryOperation(OpNotExists, NewAttribute("foo"))},
		{in: "{ nil != .foo }", expected: newUnaryOperation(OpExists, NewAttribute("foo"))},
		{in: "{ nil = .foo }", expected: newUnaryOperation(OpNotExists, NewAttribute("foo"))},
	}

	test := func(t *testing.T, q string, expected FieldExpression) {
		actual, err := Parse(q)
		require.NoError(t, err, q)
		require.Equal(t, newRootExpr(newPipeline(newSpansetFilter(expected))), actual, q)
	}

	for _, tc := range tests {
		t.Run(tc.in, func(t *testing.T) {
			test(t, tc.in, tc.expected)
		})
	}
}

func TestAttributes(t *testing.T) {
	tests := []struct {