Merge pull request #57 from alphatwirl/dev

Update state transitions
alphatwirl · Jun 30, 2024 · 9c24ae2 · 9c24ae2
2 parents b03b7d7 + 724d948
commit 9c24ae2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/atpbar/machine.py b/atpbar/machine.py
@@ -138,7 +138,7 @@ def flush(self) -> 'State':
         return self
 
     def shutdown(self) -> 'State':
-        return self
+        return Initial(callback=self._callback)
 
 
 class Initial(State):

diff --git a/tests/test_state.py b/tests/test_state.py
@@ -140,12 +140,11 @@ def shutdown(self) -> None:
         prev = self.state
         self.callback.reset_mock()
         self.state = prev.shutdown()
+        assert isinstance(self.state, Initial)
         match prev:
             case Active():
-                assert isinstance(self.state, Initial)
                 assert self.callback.mock_calls == [call.shutdown_in_active()]
             case _:
-                assert self.state is prev
                 assert self.callback.mock_calls == []