add full state in forced_mode

davidtazy · Dec 10, 2023 · de55f62 · de55f62
1 parent 51d9a11
commit de55f62
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/doc/soft_solat_state_machine.png b/doc/soft_solat_state_machine.png
diff --git a/soft_solar_router/application/state_machine.py b/soft_solar_router/application/state_machine.py
@@ -18,9 +18,11 @@ class SolarRouterStateMachine(StateMachine):
     event_no_importing = sunny_off.to(sunny_on)
 
     event_start_forced = idle.to(forced_on)
-    event_stop_forced = forced_on.to(idle)
+    event_stop_forced = forced_on.to(idle) | full.to(idle)
 
-    event_not_enought_consumption_when_switch_on = sunny_on.to(full)
+    event_not_enought_consumption_when_switch_on = sunny_on.to(full) | forced_on.to(
+        full
+    )
 
     def __init__(self):
         self.expected_switch_state = False