Function Test_ReportLoop

pkg/usagestats/reporter_test.go:129–210

(t *testing.T)

}

func Test_ReportLoop(t *testing.T) {
	origReportCheckInterval := reportCheckInterval
	origReportInterval := reportInterval
	origStabilityCheckInterval := stabilityCheckInterval
	origStabilityMinimumRequired := stabilityMinimumRequired
	origUsageStatsURL := usageStatsURL
	defer func() {
		reportCheckInterval = origReportCheckInterval
		reportInterval = origReportInterval
		stabilityCheckInterval = origStabilityCheckInterval
		stabilityMinimumRequired = origStabilityMinimumRequired
		usageStatsURL = origUsageStatsURL
	}()

	reportCheckInterval = 10 * time.Millisecond
	reportInterval = 50 * time.Millisecond
	stabilityCheckInterval = 10 * time.Millisecond
	stabilityMinimumRequired = 1

	const targetReports = 5
	var (
		mtx         sync.Mutex
		totalReport int
		clusterIDs  []string
	)
	reportsDone := make(chan struct{})
	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		var received Report
		require.NoError(t, json.NewDecoder(r.Body).Decode(&received))

		mtx.Lock()
		totalReport++
		clusterIDs = append(clusterIDs, received.ClusterID)
		shouldClose := totalReport >= targetReports
		mtx.Unlock()

		if shouldClose {
			select {
			case <-reportsDone:
			default:
				close(reportsDone)
			}
		}

		rw.WriteHeader(http.StatusOK)
	}))
	defer server.Close()
	usageStatsURL = server.URL

	objectClient, err := local.NewBackend(&local.Config{
		Path: t.TempDir(),
	})
	require.NoError(t, err)

	r, err := NewReporter(Config{Leader: true, Enabled: true}, kv.Config{
		Store: "inmemory",
	}, objectClient, objectClient, log.NewNopLogger(), prometheus.NewPedanticRegistry())
	require.NoError(t, err)