@@ -27,9 +27,9 @@ def current_time():
2727def campaign_machine ():
2828 "Define a new class for each test"
2929 from statemachine import State
30- from statemachine import StateMachine
30+ from statemachine import StateChart
3131
32- class CampaignMachine (StateMachine ):
32+ class CampaignMachine (StateChart ):
3333 "A workflow machine"
3434
3535 draft = State (initial = True )
@@ -47,11 +47,13 @@ class CampaignMachine(StateMachine):
4747def campaign_machine_with_validator ():
4848 "Define a new class for each test"
4949 from statemachine import State
50- from statemachine import StateMachine
50+ from statemachine import StateChart
5151
52- class CampaignMachine (StateMachine ):
52+ class CampaignMachine (StateChart ):
5353 "A workflow machine"
5454
55+ error_on_execution = False
56+
5557 draft = State (initial = True )
5658 producing = State ("Being produced" )
5759 closed = State (final = True )
@@ -71,9 +73,9 @@ def can_produce(*args, **kwargs):
7173def campaign_machine_with_final_state ():
7274 "Define a new class for each test"
7375 from statemachine import State
74- from statemachine import StateMachine
76+ from statemachine import StateChart
7577
76- class CampaignMachine (StateMachine ):
78+ class CampaignMachine (StateChart ):
7779 "A workflow machine"
7880
7981 draft = State (initial = True )
@@ -91,9 +93,9 @@ class CampaignMachine(StateMachine):
9193def campaign_machine_with_values ():
9294 "Define a new class for each test"
9395 from statemachine import State
94- from statemachine import StateMachine
96+ from statemachine import StateChart
9597
96- class CampaignMachineWithKeys (StateMachine ):
98+ class CampaignMachineWithKeys (StateChart ):
9799 "A workflow machine"
98100
99101 draft = State (initial = True , value = 1 )
@@ -131,9 +133,9 @@ def AllActionsMachine():
131133@pytest .fixture ()
132134def classic_traffic_light_machine (engine ):
133135 from statemachine import State
134- from statemachine import StateMachine
136+ from statemachine import StateChart
135137
136- class TrafficLightMachine (StateMachine ):
138+ class TrafficLightMachine (StateChart ):
137139 green = State (initial = True )
138140 yellow = State ()
139141 red = State ()
@@ -150,18 +152,16 @@ def _get_engine(self):
150152
151153@pytest .fixture ()
152154def classic_traffic_light_machine_allow_event (classic_traffic_light_machine ):
153- class TrafficLightMachineAllowingEventWithoutTransition (classic_traffic_light_machine ):
154- allow_event_without_transition = True
155-
156- return TrafficLightMachineAllowingEventWithoutTransition
155+ """Already allow_event_without_transition=True (StateChart default)."""
156+ return classic_traffic_light_machine
157157
158158
159159@pytest .fixture ()
160160def reverse_traffic_light_machine ():
161161 from statemachine import State
162- from statemachine import StateMachine
162+ from statemachine import StateChart
163163
164- class ReverseTrafficLightMachine (StateMachine ):
164+ class ReverseTrafficLightMachine (StateChart ):
165165 "A traffic light machine"
166166
167167 green = State (initial = True )
@@ -177,9 +177,9 @@ class ReverseTrafficLightMachine(StateMachine):
177177@pytest .fixture ()
178178def approval_machine (current_time ): # noqa: C901
179179 from statemachine import State
180- from statemachine import StateMachine
180+ from statemachine import StateChart
181181
182- class ApprovalMachine (StateMachine ):
182+ class ApprovalMachine (StateChart ):
183183 "A workflow machine"
184184
185185 requested = State (initial = True )
0 commit comments