handle update_scheduled_time

Author: nsaje

tasktiger/task.py

@@ -443,6 +443,8 @@ def update_scheduled_time(
                 )
             )
 
+        self._data["scheduled_at"] = ts
+        tiger.connection.set(tiger._key("task", self.id), json.dumps(self._data))
         self._ts = ts
 
     def __repr__(self) -> str:

tests/test_task.py

@@ -87,3 +87,16 @@ def test_scheduled_at_persists_after_reload(self, tiger):
     def test_scheduled_at_none_for_unqueued_task(self, tiger):
         task = Task(tiger, simple_task)
         assert task.scheduled_at is None
+
+    def test_update_scheduled_time_updates_scheduled_at(self, tiger):
+        future = datetime.timedelta(minutes=5)
+        later = datetime.timedelta(minutes=10)
+        with FreezeTime(self.FROZEN_NOW):
+            task = tiger.delay(simple_task, when=future)
+        assert task.scheduled_at == self.FROZEN_NOW + future
+
+        new_when = self.FROZEN_NOW + later
+        task.update_scheduled_time(when=new_when)
+
+        reloaded = Task.from_id(tiger, task.queue, "scheduled", task.id)
+        assert reloaded.scheduled_at == new_when