Change VerifyIPsWithDig to expect one occurrence of each IP

Signed-off-by: Tom Pantelis <[email protected]>
submariner-io · Dec 15, 2023 · 25f89fe · 25f89fe
1 parent b1b38f7
commit 25f89fe
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/e2e/framework/framework.go b/test/e2e/framework/framework.go
@@ -718,13 +718,13 @@ func (f *Framework) VerifyIPsWithDig(cluster framework.ClusterIndex, service *v1
 			return false, fmt.Sprintf("expected execution result %q to be empty", result), nil
 		}
 		for _, ip := range ipList {
-			doesContain := strings.Contains(result.(string), ip)
-			if doesContain && !shouldContain {
+			count := strings.Count(result.(string), ip)
+			if count > 0 && !shouldContain {
 				return false, fmt.Sprintf("expected execution result %q not to contain %q", result, ip), nil
 			}
 
-			if !doesContain && shouldContain {
-				return false, fmt.Sprintf("expected execution result %q to contain %q", result, ip), nil
+			if count != 1 && shouldContain {
+				return false, fmt.Sprintf("expected execution result %q to contain one occurrence of %q", result, ip), nil
 			}
 		}