Function TestRunCheckerSuccess

cmd/tempo-vulture/main_test.go:905–948

(t *testing.T)

}

func TestRunCheckerSuccess(t *testing.T) {
	seed := time.Date(2008, 1, 1, 12, 0, 0, 0, time.UTC)
	startTime := time.Date(2007, 1, 1, 12, 0, 0, 0, time.UTC)
	now := time.Date(2009, 1, 1, 12, 0, 0, 0, time.UTC) // Far in the future to ensure trace is ready

	config := vultureConfiguration{
		tempoOrgID:                    "orgID",
		tempoWriteBackoffDuration:     time.Second, // Small value to ensure trace is ready
		tempoLongWriteBackoffDuration: time.Second,
	}

	logger = zap.NewNop()

	tickChan := make(chan time.Time, 1)
	mockTicker := &time.Ticker{C: tickChan}

	// Send the time on the channel to trigger the ticker
	go func() {
		tickChan <- now
	}()

	// Checker function that signals completion
	mx := sync.Mutex{}
	checkerCalled := false
	var checkedInfo *util.TraceInfo
	checker := func(info *util.TraceInfo, _ *zap.Logger) {
		mx.Lock()
		defer mx.Unlock()
		checkerCalled = true
		checkedInfo = info
	}

	selectPastTimestamp := func(_ time.Time) (newStart, ts time.Time, skip bool) {
		return startTime, seed, false
	}
	runChecker(mockTicker, config, selectPastTimestamp, checker, logger)
	time.Sleep(10 * time.Millisecond)

	mx.Lock()
	defer mx.Unlock()
	assert.True(t, checkerCalled)
	require.NotNil(t, checkedInfo)
	assert.Equal(t, seed.Unix(), checkedInfo.Timestamp().Unix())
}

func TestCreateHttpClient(t *testing.T) {
	testCases := []struct {