diff --git a/rust/connlib/tunnel/src/tests/transition.rs b/rust/connlib/tunnel/src/tests/transition.rs index ab47fde28..3d6742118 100644 --- a/rust/connlib/tunnel/src/tests/transition.rs +++ b/rust/connlib/tunnel/src/tests/transition.rs @@ -222,7 +222,7 @@ where src.prop_map(Into::into), dst.prop_map(Into::into), any::(), - any::(), + non_dns_ports(), any::(), any::(), ) @@ -250,7 +250,7 @@ where src.prop_map(Into::into), dst.prop_map(Into::into), any::(), - any::(), + non_dns_ports(), any::(), any::(), ) @@ -265,6 +265,13 @@ where }) } +fn non_dns_ports() -> impl Strategy { + any::().prop_filter( + "avoid using port 53 for non-dns queries for simplicity", + |p| *p != 53, + ) +} + /// Samples up to 5 DNS queries that will be sent concurrently into connlib. pub(crate) fn dns_queries( domain: impl Strategy,