Function TestCombineProtoTotals

pkg/model/trace/combine_test.go:19–65

(t *testing.T)

)

func TestCombineProtoTotals(t *testing.T) {
	methods := []func(a, b *tempopb.Trace) (*tempopb.Trace, int){
		func(a, b *tempopb.Trace) (*tempopb.Trace, int) {
			c := NewCombiner(0, false)
			_, err := c.Consume(a)
			require.NoError(t, err)
			_, err = c.Consume(b)
			require.NoError(t, err)
			return c.Result()
		},
	}

	sameTrace := test.MakeTraceWithSpanCount(10, 10, []byte{0x01, 0x03})
	tests := []struct {
		traceA        *tempopb.Trace
		traceB        *tempopb.Trace
		expectedTotal int
	}{
		{
			traceA:        nil,
			traceB:        test.MakeTraceWithSpanCount(10, 10, []byte{0x01, 0x03}),
			expectedTotal: -1,
		},
		{
			traceA:        test.MakeTraceWithSpanCount(10, 10, []byte{0x01, 0x03}),
			traceB:        nil,
			expectedTotal: -1,
		},
		{
			traceA:        test.MakeTraceWithSpanCount(10, 10, []byte{0x01, 0x03}),
			traceB:        test.MakeTraceWithSpanCount(10, 10, []byte{0x01, 0x01}),
			expectedTotal: 200,
		},
		{
			traceA:        sameTrace,
			traceB:        sameTrace,
			expectedTotal: 100,
		},
	}

	for _, tt := range tests {
		for _, m := range methods {
			_, actualTotal := m(tt.traceA, tt.traceB)
			assert.Equal(t, tt.expectedTotal, actualTotal)
		}
	}
}

func TestCombinerChecksMaxBytes(t *testing.T) {
	// Ensure that the combiner checks max bytes when consuming a trace.