@@ -42,7 +42,7 @@ func testMW(t *testing.T, config DBConfig) {
4242 lbPolicy := config .lbPolicy
4343
4444 // Skip extreme cases that are likely to cause load balancer distribution issues
45- if noOfPrimaries > 10 || noOfReplicas > 10 {
45+ if noOfPrimaries > 3 || noOfReplicas > 3 {
4646 t .Skipf ("skipping extreme case with %d primaries and %d replicas for test stability" , noOfPrimaries , noOfReplicas )
4747 return
4848 }
@@ -82,7 +82,7 @@ func testMW(t *testing.T, config DBConfig) {
8282 // Limit iterations to prevent excessive mock expectations during fuzzing
8383 maxIterations := 6
8484 if noOfPrimaries > 1 {
85- maxIterations = min (12 , 6 * min ( noOfPrimaries , 2 )) // Cap at 12 iterations, even for many primaries
85+ maxIterations = min (noOfPrimaries * 6 , 30 ) // Cap at 30 iterations
8686 }
8787
8888 for i := 0 ; i < maxIterations ; i ++ {
@@ -172,10 +172,7 @@ func testMW(t *testing.T, config DBConfig) {
172172 }
173173
174174 // Limit iterations to prevent excessive mock expectations during fuzzing
175- maxIterations := 5
176- if noOfReplicas > 1 {
177- maxIterations = min (8 , 4 * min (noOfReplicas , 2 )) // Cap at 8 iterations, even for many replicas
178- }
175+ maxIterations := min (noOfReplicas * 5 , 20 ) // Cap at 20 iterations
179176
180177 for i := 0 ; i < maxIterations ; i ++ {
181178 robin := resolver .loadBalancer .predict (noOfReplicas )
0 commit comments