Make BGP prefix checking less flaky (#423)

* Make BGP prefix checking less flaky * Increase timeout * Revert timeout back to 1min, 3min is overkill
openconfig · Jun 6, 2024 · 41b73cf · 41b73cf
1 parent c6c7f5e
commit 41b73cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/integration_tests/onedut_oneotg_tests/bgp_route_propagation/bgp_route_propagation_test.go b/integration_tests/onedut_oneotg_tests/bgp_route_propagation/bgp_route_propagation_test.go
@@ -277,7 +277,7 @@ func checkOTGBGP4Prefix(t *testing.T, otg *otg.OTG, config gosnappi.Config, expe
 	t.Helper()
 	_, ok := gnmi.WatchAll(t,
 		otg,
-		gnmi.OTG().BgpPeer(expectedOTGBGPPrefix.PeerName).UnicastIpv4PrefixAny().State(),
+		gnmi.OTG().BgpPeer(expectedOTGBGPPrefix.PeerName).UnicastIpv4PrefixAny().WithAddress(expectedOTGBGPPrefix.Address).State(),
 		time.Minute,
 		func(v *ygnmi.Value[*otgtelemetry.BgpPeer_UnicastIpv4Prefix]) bool {
 			_, present := v.Val()
@@ -302,7 +302,7 @@ func checkOTGBGP6Prefix(t *testing.T, otg *otg.OTG, config gosnappi.Config, expe
 	t.Helper()
 	_, ok := gnmi.WatchAll(t,
 		otg,
-		gnmi.OTG().BgpPeer(expectedOTGBGPPrefix.PeerName).UnicastIpv6PrefixAny().State(),
+		gnmi.OTG().BgpPeer(expectedOTGBGPPrefix.PeerName).UnicastIpv6PrefixAny().WithAddress(expectedOTGBGPPrefix.Address).State(),
 		time.Minute,
 		func(v *ygnmi.Value[*otgtelemetry.BgpPeer_UnicastIpv6Prefix]) bool {
 			_, present := v.Val()