Function TestWriteValidationTrace

cmd/tempo-vulture/validation_test.go:16–79

(t *testing.T)

)

func TestWriteValidationTrace(t *testing.T) {
	tests := []struct {
		name           string
		config         ValidationConfig
		expectError    bool
		emitBatchError error
	}{
		{
			name: "successful writes",
			config: ValidationConfig{
				TempoOrgID: "test-org",
			},
			expectError: false,
		},
		{
			name: "emit batch failure",
			config: ValidationConfig{
				TempoOrgID: "test-org",
			},
			expectError:    true,
			emitBatchError: errors.New("network error"),
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mockReporter := &MockReporter{
				err: tt.emitBatchError, // Set error if test expects failure
			}

			mockClock := &MockClock{
				now: time.Date(2025, 1, 1, 12, 0, 0, 0, time.UTC),
			}

			vs := &ValidationService{
				config: tt.config,
				clock:  mockClock,
				logger: zap.NewNop(),
			}

			// Call the function
			traceInfo, actualTrace, err := vs.writeValidationTrace(context.Background(), mockReporter)

			// Assertions
			if tt.expectError {
				assert.Error(t, err)
				assert.Nil(t, traceInfo, "trace info should be nil")
				assert.Nil(t, actualTrace, "actual trace should be nil")
			} else {
				assert.NoError(t, err)
				assert.NotNil(t, traceInfo, "traceInfo should not be nil")
				assert.NotNil(t, actualTrace, "actualTrace should not be nil")

				// verify traceInfo is populated correctly
				assert.NotEmpty(t, traceInfo.HexID(), "traceInfo.HexID should not be empty")
				assert.NotEmpty(t, actualTrace.ResourceSpans, "actualTrace.ResourceSpans should not be empty")

				// Verify batches were actually emitted