diff --git a/beat/web/backend/tests.py b/beat/web/backend/tests.py index 014cbff0e260ce73f26114270f2fafb3b770229e..ecfdb0da15c58926b5924284ec2caeb2f2647b46 100644 --- a/beat/web/backend/tests.py +++ b/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]