Function TestValidateTraceRetrieval

cmd/tempo-vulture/validation_test.go:81–164

(t *testing.T)

}

func TestValidateTraceRetrieval(t *testing.T) {
	tests := []struct {
		name          string
		queryError    error
		traceResponse *tempopb.Trace
		expectError   bool
	}{
		{
			name: "successful retrieval",
			// We'll create matching trace ID dynamically
			traceResponse: &tempopb.Trace{
				ResourceSpans: []*v1.ResourceSpans{
					{
						ScopeSpans: []*v1.ScopeSpans{
							{
								Spans: []*v1.Span{
									{
										// TraceId will be set dynamically to match
										Name: "test-span",
									},
								},
							},
						},
					},
				},
			},
			expectError: false,
		},
		{
			name:        "query error",
			queryError:  errors.New("query failed"),
			expectError: true,
		},
		{
			name: "empty trace spans",
			traceResponse: &tempopb.Trace{
				ResourceSpans: []*v1.ResourceSpans{}, // Empty!
			},
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Create a TraceInfo
			mockClock := &MockClock{
				now: time.Date(2025, 1, 1, 12, 0, 0, 0, time.UTC),
			}

			vs := &ValidationService{
				config: ValidationConfig{TempoOrgID: "test-org"},
				clock:  mockClock,
				logger: zap.NewNop(),
			}

			traceInfo := util.NewTraceInfo(mockClock.Now(), "test-org")

			// Set matching trace ID in the mock response