fix: differing inbount/outbound modalities for Hocs

sine-fdn · Sep 16, 2024 · f6a2327 · f6a2327
1 parent c693ca6
commit f6a2327
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/gen/src/arbitrary_impls.rs b/gen/src/arbitrary_impls.rs
@@ -69,13 +69,13 @@ impl Arbitrary for Hoc {
             g: &mut quickcheck::Gen,
         ) -> (Option<TransportMode>, Option<TransportMode>) {
             let inbound = Some(TransportMode::arbitrary(g));
-            let outbound = Some(TransportMode::arbitrary(g));
+            let mut outbound = Some(TransportMode::arbitrary(g));
 
-            if inbound == outbound {
-                (inbound, Some(TransportMode::arbitrary(g)))
-            } else {
-                (inbound, outbound)
+            while inbound == outbound {
+                outbound = Some(TransportMode::arbitrary(g));
             }
+
+            (inbound, outbound)
         }
 
         let hub_type = HubType::arbitrary(g);