Function TestFindTraceByID_ExternalMode

modules/querier/querier_test.go:113–181

(t *testing.T)

}

func TestFindTraceByID_ExternalMode(t *testing.T) {
	traceID := []byte{0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10}
	userID := "test-tenant"

	externalTrace := &tempopb.Trace{
		ResourceSpans: []*v1_trace.ResourceSpans{
			{
				ScopeSpans: []*v1_trace.ScopeSpans{
					{
						Spans: []*v1_trace.Span{
							{
								TraceId: traceID,
								SpanId:  []byte{0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08},
								Name:    "external-span",
							},
						},
					},
				},
			},
		},
	}

	startTime := int64(1000)
	endTime := int64(2000)
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// Verify that start and end query parameters are present
		require.Equal(t, strconv.FormatInt(startTime, 10), r.URL.Query().Get("start"))
		require.Equal(t, strconv.FormatInt(endTime, 10), r.URL.Query().Get("end"))

		traceBytes, err := externalTrace.Marshal()
		require.NoError(t, err)

		w.Header().Set("Content-Type", api.HeaderAcceptProtobuf)
		w.WriteHeader(http.StatusOK)
		_, err = w.Write(traceBytes)
		require.NoError(t, err)
	}))
	defer server.Close()

	cfg := Config{
		TraceByID: TraceByIDConfig{
			External: ExternalConfig{
				Endpoint: server.URL,
				Timeout:  10 * time.Second,
			},
		},
	}

	o, err := overrides.NewOverrides(overrides.Config{}, nil, prometheus.DefaultRegisterer)
	require.NoError(t, err)

	q, err := New(cfg, nil, livestore_client.Config{}, nil, true, nil, o)
	require.NoError(t, err)

	ctx := user.InjectOrgID(context.Background(), userID)

	resp, err := q.FindTraceByID(ctx, &tempopb.TraceByIDRequest{
		TraceID:   traceID,