Function TestServiceSampling

internal/telemetry/telemetry_test.go:496–614

(t *testing.T)

}

func TestServiceSampling(t *testing.T) {
	t.Run("sampleRate 0 sends all events", func(t *testing.T) {
		t.Cleanup(stubDeviceID("test-device"))

		var captured SendTelemetryPayload
		svc := newService(func(p SendTelemetryPayload) { captured = p }, nil)
		svc.sampleRate = 0
		svc.sampleBucket = 99

		svc.Record(ghtelemetry.Event{Type: "test"})
		svc.Flush()

		require.Len(t, captured.Events, 1)
	})

	t.Run("sampleRate 100 sends all events regardless of bucket", func(t *testing.T) {
		t.Cleanup(stubDeviceID("test-device"))

		var captured SendTelemetryPayload
		svc := newService(func(p SendTelemetryPayload) { captured = p }, nil)
		svc.sampleRate = 100
		svc.sampleBucket = 99

		svc.Record(ghtelemetry.Event{Type: "test"})
		svc.Flush()

		require.Len(t, captured.Events, 1)
	})

	t.Run("bucket below sampleRate sends events", func(t *testing.T) {
		t.Cleanup(stubDeviceID("test-device"))

		var captured SendTelemetryPayload
		svc := newService(func(p SendTelemetryPayload) { captured = p }, nil)
		svc.sampleRate = 50
		svc.sampleBucket = 49 // below rate, should be included

		svc.Record(ghtelemetry.Event{Type: "test"})
		svc.Flush()

		require.Len(t, captured.Events, 1)
	})

	t.Run("bucket at sampleRate drops events", func(t *testing.T) {
		t.Cleanup(stubDeviceID("test-device"))

		called := false
		svc := newService(func(SendTelemetryPayload) { called = true }, nil)
		svc.sampleRate = 50
		svc.sampleBucket = 50 // at rate boundary, should be excluded

		svc.Record(ghtelemetry.Event{Type: "test"})
		svc.Flush()

		assert.False(t, called, "flusher should not be called when bucket >= sampleRate")
	})

	t.Run("bucket above sampleRate drops events", func(t *testing.T) {