Fix scheduler.once() running multiple times (#231)

mmpy_bot/scheduler.py

@@ -18,8 +18,20 @@ def set_next_run(self, next_time: datetime):
             raise AssertionError("The next_time parameter should be a datetime object.")
         self.at_time = next_time
         self.next_run = next_time
+        self.should_run = True
+
+    @property
+    def should_run(self):
+        return self._keep_running and super().should_run
+
+    @should_run.setter
+    def should_run(self, value):
+        self._keep_running = value
 
     def run(self):
+        # This prevents the job from running more than once
+        self.should_run = False
+
         super().run()
         return schedule.CancelJob()
 

tests/unit_tests/scheduler_test.py

@@ -3,6 +3,7 @@
 from datetime import datetime
 from pathlib import Path
 from typing import Dict
+from unittest.mock import Mock
 
 import pytest
 
@@ -31,6 +32,19 @@ def job(path: str):
         assert float(file.readline()) - 2 == pytest.approx(start, abs=0.3)
 
 
+def test_once_single_call():
+    mock = Mock()
+    mock.side_effect = lambda: time.sleep(0.2)
+
+    schedule.once().do(mock)
+
+    for _ in range(10):
+        schedule.run_pending()
+        time.sleep(0.05)
+
+    mock.assert_called_once()
+
+
 def test_recurring_thread():
     def job(modifiable_arg: Dict):
         # Modify the variable, which should be shared with the main thread.