[proptest-state-machine] Introduce Sequential::new ctor (#506)

proptest-rs · Sep 22, 2024 · 52d3a38 · 52d3a38
1 parent a3cc4a5
commit 52d3a38
Showing 1 changed file with 27 additions and 7 deletions.
diff --git a/proptest-state-machine/src/strategy.rs b/proptest-state-machine/src/strategy.rs
@@ -102,13 +102,13 @@ pub trait ReferenceStateMachine {
         BoxedStrategy<Self::State>,
         BoxedStrategy<Self::Transition>,
     > {
-        Sequential {
-            size: size.into(),
-            init_state: Self::init_state,
-            preconditions: Self::preconditions,
-            transitions: Self::transitions,
-            next: Self::apply,
-        }
+        Sequential::new(
+            size.into(),
+            Self::init_state,
+            Self::preconditions,
+            Self::transitions,
+            Self::apply,
+        )
     }
 }
 
@@ -146,6 +146,26 @@ pub struct Sequential<State, Transition, StateStrategy, TransitionStrategy> {
     next: fn(state: State, transition: &Transition) -> State,
 }
 
+impl<State, Transition, StateStrategy, TransitionStrategy>
+    Sequential<State, Transition, StateStrategy, TransitionStrategy>
+{
+    pub fn new(
+        size: SizeRange,
+        init_state: fn() -> StateStrategy,
+        preconditions: fn(state: &State, transition: &Transition) -> bool,
+        transitions: fn(state: &State) -> TransitionStrategy,
+        next: fn(state: State, transition: &Transition) -> State,
+    ) -> Self {
+        Self {
+            size,
+            init_state,
+            preconditions,
+            transitions,
+            next,
+        }
+    }
+}
+
 impl<State, Transition, StateStrategy, TransitionStrategy> Debug
     for Sequential<State, Transition, StateStrategy, TransitionStrategy>
 {