Ver Fonte

Fixes hub tests

Ask Solem há 11 anos atrás
pai
commit
96eb2fa004
1 ficheiros alterados com 5 adições e 1 exclusões
  1. 5 1
      celery/tests/worker/test_hub.py

+ 5 - 1
celery/tests/worker/test_hub.py

@@ -53,8 +53,12 @@ class test_LaxBoundedSemaphore(Case):
         self.assertEqual(x.value, 0)
         self.assertFalse(c3.called)
 
+        x.release()
+        self.assertEqual(x.value, 0)
         x.release()
         self.assertEqual(x.value, 1)
+        x.release()
+        self.assertEqual(x.value, 2)
         c3.assert_called_with(3)
 
     def test_bounded(self):
@@ -83,7 +87,7 @@ class test_LaxBoundedSemaphore(Case):
         x.grow(2)
         cb2.assert_called_with(2)
         cb3.assert_called_with(3)
-        self.assertEqual(x.value, 3)
+        self.assertEqual(x.value, 2)
         self.assertEqual(x.initial_value, 3)
 
         self.assertFalse(x._waiting)