refac: Using get/set state for pickling

Author: fgmacedo

statemachine/engines/base.py

@@ -16,27 +16,13 @@
 
 class BaseEngine:
     def __init__(self, sm: "StateMachine", rtc: bool = True):
+        self.sm: StateMachine = proxy(sm)
+        self._external_queue: Queue = PriorityQueue()
         self._sentinel = object()
         self._rtc = rtc
         self._running = True
-        self._init(sm)
-
-    def _init(self, sm: "StateMachine"):
-        self.sm: StateMachine = proxy(sm)
-        self._external_queue: Queue = PriorityQueue()
         self._processing = Lock()
 
-    def __getstate__(self) -> dict:
-        state = self.__dict__.copy()
-        del state["_external_queue"]
-        del state["_processing"]
-        del state["sm"]
-        return state
-
-    def __setstate__(self, state: dict) -> None:
-        for attr, value in state.items():
-            setattr(self, attr, value)
-
     def empty(self):
         return self._external_queue.qsize() == 0
 

tests/test_copy.py

@@ -124,15 +124,15 @@ def assertions(sm, _reference):
         if not sm._listeners:
             pytest.fail("did not found any observer")
 
-        for listener in sm._listeners:
+        for listener in sm._listeners.values():
             listener.let_me_be_visible = False
 
         with pytest.raises(TransitionNotAllowed):
             sm.send("publish")
 
         sm.model.let_me_be_visible = True
 
-        for listener in sm._listeners:
+        for listener in sm._listeners.values():
             with pytest.raises(TransitionNotAllowed):
                 sm.send("publish")