Function TestTraceIDHandlerV2

modules/frontend/traceid_handlers_test.go:275–438

(t *testing.T)

}

func TestTraceIDHandlerV2(t *testing.T) {
	// create and split a splitTrace
	splitTrace := test.MakeTrace(2, []byte{0x01, 0x02})
	trace1 := &tempopb.Trace{}
	trace2 := &tempopb.Trace{}

	for i, b := range splitTrace.ResourceSpans {
		if i%2 == 0 {
			trace1.ResourceSpans = append(trace1.ResourceSpans, b)
		} else {
			trace2.ResourceSpans = append(trace2.ResourceSpans, b)
		}
	}

	tests := []struct {
		name           string
		status1        int
		status2        int
		trace1         *tempopb.Trace
		trace2         *tempopb.Trace
		err1           error
		err2           error
		expectedStatus int
		expectedTrace  *tempopb.Trace
	}{
		{
			name:           "empty returns",
			status1:        200,
			status2:        200,
			expectedStatus: 200,
			expectedTrace:  &tempopb.Trace{},
		},
		{
			name:           "404",
			status1:        404,
			status2:        404,
			expectedStatus: 404,
		},
		{
			name:           "400",
			status1:        400,
			status2:        400,
			expectedStatus: 500,
		},
		{
			name:           "500+200",
			status1:        500,
			status2:        200,
			trace2:         trace2,
			expectedStatus: 500,
		},
		{
			name:           "200+500",
			status1:        200,
			trace1:         trace1,
			status2:        500,
			expectedStatus: 500,
		},