Function testSearchCombinesResults

modules/frontend/combiner/search_test.go:196–247

(t *testing.T, marshalingFormat api.MarshallingFormat)

}

func testSearchCombinesResults(t *testing.T, marshalingFormat api.MarshallingFormat) {
	for _, keepMostRecent := range []bool{true, false} {
		start := time.Date(1, 2, 3, 4, 5, 6, 7, time.UTC)
		traceID := "traceID"

		c := NewSearch(10, keepMostRecent, marshalingFormat, false)
		sr := toHTTPResponseWithFormat(t, &tempopb.SearchResponse{
			Traces: []*tempopb.TraceSearchMetadata{
				{
					TraceID:           traceID,
					StartTimeUnixNano: uint64(start.Add(time.Second).UnixNano()),
					DurationMs:        uint32(time.Second.Milliseconds()),
				}, // 1 second after start and shorter duration
				{
					TraceID:           traceID,
					StartTimeUnixNano: uint64(start.UnixNano()),
					DurationMs:        uint32(time.Hour.Milliseconds()),
				}, // earliest start time and longer duration
				{
					TraceID:           traceID,
					StartTimeUnixNano: uint64(start.Add(time.Hour).UnixNano()),
					DurationMs:        uint32(time.Millisecond.Milliseconds()),
				}, // 1 hour after start and shorter duration
			},
			Metrics: &tempopb.SearchMetrics{},
		}, 200, nil, marshalingFormat)
		err := c.AddResponse(sr)
		require.NoError(t, err)

		expected := &tempopb.SearchResponse{
			Traces: []*tempopb.TraceSearchMetadata{
				{
					TraceID:           traceID,
					StartTimeUnixNano: uint64(start.UnixNano()),
					DurationMs:        uint32(time.Hour.Milliseconds()),
					RootServiceName:   search.RootSpanNotYetReceivedText,
				},
			},
			Metrics: &tempopb.SearchMetrics{
				CompletedJobs: 1,
			},
		}

		resp, err := c.HTTPFinal()
		require.NoError(t, err)

		actual := &tempopb.SearchResponse{}
		fromHTTPResponse(t, resp, actual)

		require.Equal(t, expected, actual)
	}
}

func TestSearchResponseCombinerJSON(t *testing.T) {
	testSearchResponseCombiner(t, api.MarshallingFormatJSON)