test: cover daemon telemetry publication
This commit is contained in:
+16
-9
@@ -255,15 +255,7 @@ func publishClientTelemetry(ctx context.Context, state *spool.Store, health *obs
|
||||
ticker := time.NewTicker(10 * time.Second)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
sampleContext, cancel := context.WithTimeout(ctx, time.Second)
|
||||
telemetry, err := state.Telemetry(sampleContext)
|
||||
cancel()
|
||||
if err != nil {
|
||||
health.Inc("telemetry_read_failure")
|
||||
} else {
|
||||
health.SetGauge("queue_depth", float64(telemetry.QueuedCommands))
|
||||
health.SetGauge("client_spool_bytes", float64(telemetry.ChargedBytes))
|
||||
}
|
||||
publishClientTelemetrySample(ctx, state.Telemetry, health)
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
@@ -272,6 +264,21 @@ func publishClientTelemetry(ctx context.Context, state *spool.Store, health *obs
|
||||
}
|
||||
}
|
||||
|
||||
// publishClientTelemetrySample performs one intentionally bounded, best-effort
|
||||
// telemetry read. An unavailable spool metric must not interrupt reconnect or
|
||||
// command supervision.
|
||||
func publishClientTelemetrySample(ctx context.Context, read func(context.Context) (spool.Telemetry, error), health *observability.Health) {
|
||||
sampleContext, cancel := context.WithTimeout(ctx, time.Second)
|
||||
telemetry, err := read(sampleContext)
|
||||
cancel()
|
||||
if err != nil {
|
||||
health.Inc("telemetry_read_failure")
|
||||
return
|
||||
}
|
||||
health.SetGauge("queue_depth", float64(telemetry.QueuedCommands))
|
||||
health.SetGauge("client_spool_bytes", float64(telemetry.ChargedBytes))
|
||||
}
|
||||
|
||||
func clientJobProfiles(profiles config.Profiles) map[string]clientwindows.JobProfile {
|
||||
return map[string]clientwindows.JobProfile{
|
||||
rvboxv1.ExecutionProfile_EXECUTION_PROFILE_LIGHT.String(): toJobProfile(profiles.Light),
|
||||
|
||||
@@ -2,11 +2,15 @@ package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/rvbox/rvbox/internal/client/spool"
|
||||
"github.com/rvbox/rvbox/internal/config"
|
||||
"github.com/rvbox/rvbox/internal/observability"
|
||||
)
|
||||
|
||||
func TestClientModeSelectionRequiresExactlyOneMode_HP_WINCLI_01(t *testing.T) {
|
||||
@@ -77,3 +81,27 @@ func TestNonWindowsServiceModesRemainExplicitlyUnsupported_BH_WINCLI_01(t *testi
|
||||
t.Fatal("non-Windows tray mode unexpectedly available")
|
||||
}
|
||||
}
|
||||
|
||||
func TestPublishClientTelemetrySample_HP_OPS_07(t *testing.T) {
|
||||
t.Parallel()
|
||||
health := observability.New()
|
||||
publishClientTelemetrySample(context.Background(), func(context.Context) (spool.Telemetry, error) {
|
||||
return spool.Telemetry{QueuedCommands: 2, ChargedBytes: 2048}, nil
|
||||
}, health)
|
||||
snapshot := health.MetricsSnapshot()
|
||||
if snapshot.Gauges["queue_depth"] != 2 || snapshot.Gauges["client_spool_bytes"] != 2048 {
|
||||
t.Fatalf("client telemetry gauges = %#v", snapshot.Gauges)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPublishClientTelemetrySampleFailure_BH_OPS_04(t *testing.T) {
|
||||
t.Parallel()
|
||||
health := observability.New()
|
||||
publishClientTelemetrySample(context.Background(), func(context.Context) (spool.Telemetry, error) {
|
||||
return spool.Telemetry{}, errors.New("spool unavailable")
|
||||
}, health)
|
||||
_, _, counters := health.Snapshot()
|
||||
if counters["telemetry_read_failure"] != 1 {
|
||||
t.Fatalf("telemetry failure counters = %#v", counters)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user