Reset cycle count on kill

Nilswaap-dot · Jun 16, 2021 · bb9db3d · bb9db3d
1 parent e231583
commit bb9db3d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/rogue/rogue.py b/src/rogue/rogue.py
@@ -213,6 +213,7 @@ def exec(self):
 
     def kill(self):
         self._daemon.kill()
+        self._cycle_count = 0
 
     def process_errors(self):
         self._daemon.process_errors()
diff --git a/tests/test_rogue.py b/tests/test_rogue.py
@@ -153,9 +153,9 @@ def loop(client):
     def test_restart(self):
         server = rogue.Server()
         server.exec()
-        time.sleep(0.01)
+        time.sleep(0.09)
         server.kill()
-        time.sleep(0.01)
+        assert server._cycle_count == 0
         server.exec()
         time.sleep(0.01)
         server.process_errors()