Function TestQueryMetrics

cmd/tempo-vulture/main_test.go:620–797

(t *testing.T)

}

func TestQueryMetrics(t *testing.T) {
	seed := time.Date(2008, 1, 1, 12, 0, 0, 0, time.UTC)

	config := vultureConfiguration{
		tempoOrgID:                 "orgID",
		tempoWriteBackoffDuration:  time.Second,
		tempoSearchBackoffDuration: time.Second,
	}

	info := util.NewTraceInfo(seed, config.tempoOrgID)
	hexID := info.HexID()

	successMetricsResponse := &tempopb.QueryRangeResponse{
		Series: []*tempopb.TimeSeries{
			{
				Samples: []tempopb.Sample{
					{
						TimestampMs: seed.UnixMilli(),
						Value:       2.0,
					},
				},
			},
		},
	}

	tests := []struct {
		name            string
		response        *tempopb.QueryRangeResponse
		searchResponse  []*tempopb.TraceSearchMetadata
		err             error
		expectedMetrics traceMetrics
		expectedError   string
	}{
		{
			name:     "successful metrics query: 1 trace, 2 spans",
			response: successMetricsResponse,
			searchResponse: []*tempopb.TraceSearchMetadata{
				{
					SpanSets: []*tempopb.SpanSet{
						{
							Matched: 2,
						},
					},
				},
			},
			expectedMetrics: traceMetrics{
				requested: 1,
			},
		},
		{
			name:     "successful metrics query: 2 traces, 1 span each",
			response: successMetricsResponse,
			searchResponse: []*tempopb.TraceSearchMetadata{
				{
					SpanSets: []*tempopb.SpanSet{
						{
							Matched: 1,
						},