Make slow batch receiver even slower in egress test for drops

cloudfoundry · Feb 2, 2022 · 33db0ab · 33db0ab
1 parent 78c2dc8
commit 33db0ab
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/src/router/internal/server/v2/egress_server.go b/src/router/internal/server/v2/egress_server.go
@@ -1,7 +1,7 @@
 package v2
 
 import (
-	"log"
+	"fmt"
 	"time"
 
 	gendiode "code.cloudfoundry.org/go-diodes"
@@ -80,9 +80,13 @@ func (s *EgressServer) BatchedReceiver(
 	s.subscriptionsMetric.Increment(1.0)
 	defer s.subscriptionsMetric.Decrement(1.0)
 
-	d := diodes.NewOneToOneWaiterEnvelopeV2(1000, gendiode.AlertFunc(func(missed int) {
-		log.Printf("Dropped %d envelopes (v2 buffer) ShardID: %s", missed, req.ShardId)
-		s.Alert(missed)}),
+	d := diodes.NewOneToOneWaiterEnvelopeV2(
+		1000,
+		gendiode.AlertFunc(func(missed int) {
+			fmt.Printf("Dropped %d envelopes (v2 buffer) ShardID: %s\n", missed, req.ShardId)
+			s.Alert(missed)
+			fmt.Printf("Alerted successfully: %d missed\n", missed)
+		}),
 		gendiode.WithWaiterContext(sender.Context()),
 	)
 	cancel := s.subscriber.Subscribe(req, d)

diff --git a/src/router/internal/server/v2/egress_server_test.go b/src/router/internal/server/v2/egress_server_test.go
@@ -234,9 +234,11 @@ var _ = Describe("EgressServer", func() {
 
 			go server.BatchedReceiver(&loggregator_v2.EgressBatchRequest{}, br)
 
-			Eventually(
-				egressDropped.GetDelta,
-				40).Should(BeNumerically(">", 1))
+			Eventually(func() uint64 {
+				fmt.Println(egressDropped.GetDelta())
+				return egressDropped.GetDelta()
+			}, 10, 1).Should(BeNumerically(">", 1))
+			// Eventually(egressDropped.GetDelta, 10).Should(BeNumerically(">", 1))
 		})
 	})
 })