fix(telemetry)_: fix flaky cache test

status-im · Sep 28, 2024 · 327be21 · 327be21
1 parent 7a23ac5
commit 327be21
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/telemetry/client_test.go b/telemetry/client_test.go
@@ -288,19 +288,27 @@ func TestRetryCacheCleanup(t *testing.T) {
  ctx := context.Background()
 
  client := createClient(t, "")
- client.Start(ctx)
 
  for i := 0; i < 6000; i++ {
- sendEnvelope(ctx, client)
+ go sendEnvelope(ctx, client)
+ telemetryRequest := <-client.telemetryCh
+ client.telemetryCache = append(client.telemetryCache, telemetryRequest)
  }
 
- time.Sleep(110 * time.Millisecond)
-
+ client.pushTelemetryRequest(client.telemetryCache)
+ client.telemetryCache = nil
  require.Equal(t, 6000, len(client.telemetryRetryCache))
 
- sendEnvelope(ctx, client)
+ go sendEnvelope(ctx, client)
+ telemetryRequest := <-client.telemetryCh
+ client.telemetryCache = append(client.telemetryCache, telemetryRequest)
+
+ client.pushTelemetryRequest(client.telemetryCache)
+ telemetryRequests := make([]TelemetryRequest, len(client.telemetryCache))
+ copy(telemetryRequests, client.telemetryCache)
+ client.telemetryCache = nil
 
- time.Sleep(210 * time.Millisecond)
+ client.pushTelemetryRequest(telemetryRequests)
 
  require.Equal(t, 5001, len(client.telemetryRetryCache))
 }