diff --git a/rust/connlib/tunnel/proptest-regressions/tests.txt b/rust/connlib/tunnel/proptest-regressions/tests.txt index 00192877c..1282b3a6b 100644 --- a/rust/connlib/tunnel/proptest-regressions/tests.txt +++ b/rust/connlib/tunnel/proptest-regressions/tests.txt @@ -176,3 +176,4 @@ cc 7ab081a00991a3265b2ca82f2203284759bc50ef2805e5514baa0c24c966a580 cc 9cac073e45583d9940fd8813b93c4cadea91c5d304c454ab8d050b44ba49dc13 cc 608f3ed9392aa067bc730538d75f3692edf2ad5c3fa98beb3e95b166e04f7b5f cc 57c9d6263fdae8b6bb51fbb7108372c7d695d1186163fcfcdce010a6666c3db5 +cc f90e2fe4827f91aba42bf4806b53d1a1e7df7d1fd912d3a2cf32774eb0006f8a diff --git a/rust/connlib/tunnel/src/tests/strategies.rs b/rust/connlib/tunnel/src/tests/strategies.rs index 0f4612372..3bb111e7b 100644 --- a/rust/connlib/tunnel/src/tests/strategies.rs +++ b/rust/connlib/tunnel/src/tests/strategies.rs @@ -14,6 +14,7 @@ use itertools::Itertools; use prop::sample; use proptest::{collection, prelude::*}; use std::iter; +use std::num::NonZeroU16; use std::{ collections::{BTreeMap, BTreeSet}, net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr}, @@ -164,7 +165,7 @@ pub(crate) fn tcp_resources( let all_domains = dns_records.domains_iter().collect::>(); collection::btree_set( - (sample::select(all_domains.clone()), any::()), + (sample::select(all_domains.clone()), any::()), 1..=all_domains.len(), ) .prop_map(move |domains| { @@ -181,7 +182,7 @@ pub(crate) fn tcp_resources( move |(domain, port)| { let addresses = dns_records .domain_ips_iter(&domain) - .map(|address| SocketAddr::new(address, port)) + .map(|address| SocketAddr::new(address, port.get())) .collect::>(); (domain, addresses)