Make statistics check eventuallly

e2e/connection_test.go

@@ -158,7 +158,9 @@ func consumerConnectFlow(t *testing.T, tequilapi *tequilapi_client.Client, consu
 	assert.Equal(t, "New", se.Status)
 
 	// Wait some time for session to collect stats.
-	time.Sleep(4 * time.Second)
+	if serviceType != "noop" {
+		assert.Eventually(t, sessionStatsReceived(tequilapi), 10*time.Second, 1*time.Second)
+	}
 
 	err = tequilapi.ConnectionDestroy()
 	assert.NoError(t, err)
@@ -181,6 +183,17 @@ func consumerConnectFlow(t *testing.T, tequilapi *tequilapi_client.Client, consu
 	serviceTypeAssertionMap[serviceType](t, se)
 }
 
+func sessionStatsReceived(tequilapi *tequilapi_client.Client) func() bool {
+	return func() bool {
+		stats, err := tequilapi.ConnectionStatistics()
+		if err != nil {
+			return false
+		}
+		log.Info().Msgf("sessions stats: %v", stats)
+		return stats.BytesReceived > 0 && stats.BytesSent > 0
+	}
+}
+
 type sessionAsserter func(t *testing.T, session tequilapi_client.ConnectionSessionDTO)
 
 var serviceTypeAssertionMap = map[string]sessionAsserter{