[backend] Simulate job killing by worker on tests

beat/web/backend/tests.py

@@ -1179,6 +1179,11 @@ class Scheduling(BaseBackendTestCase):
 
         xp.cancel()
 
+        # simulate worker cancelling
+        split.refresh_from_db()
+        self.assertEqual(split.status, Job.KILL)
+        split.end(None, Job.CANCELLED)
+
         self.assertEqual(
             [str(k) for k in xp.blocks.order_by('id').values_list('status', flat=True)],
             [Block.CANCELLED, Block.CANCELLED]
@@ -1630,6 +1635,12 @@ class Scheduling(BaseBackendTestCase):
 
         # cancels the blocking experiment - the blocked one must continue
         xp.cancel()
+
+        # simulate worker cancelling
+        split.refresh_from_db()
+        self.assertEqual(split.status, Job.KILL)
+        split.end(None, Job.CANCELLED)
+
         self.assertEqual(
             [str(k) for k in xp.blocks.order_by('id').values_list('status', flat=True)],
             [Block.CANCELLED, Block.CANCELLED]