Ask Solem 10 anni fa
parent
commit
dea4eac28a
1 ha cambiato i file con 4 aggiunte e 1 eliminazioni
  1. 4 1
      celery/tests/worker/test_consumer.py

+ 4 - 1
celery/tests/worker/test_consumer.py

@@ -83,10 +83,13 @@ class test_Consumer(AppCase):
         with patch('celery.worker.consumer.task_reserved') as reserved:
             bucket.can_consume.return_value = False
             bucket.expected_time.return_value = 3.33
+            limit_order = c._limit_order
             c._limit_task(request, bucket, 4)
+            self.assertEqual(c._limit_order, limit_order + 1)
             bucket.can_consume.assert_called_with(4)
             c.timer.call_after.assert_called_with(
-                3.33, c._limit_task, (request, bucket, 4),
+                3.33, c._limit_move_to_pool, (request, ),
+                priority=c._limit_order,
             )
             bucket.expected_time.assert_called_with(4)
             self.assertFalse(reserved.called)