Ask Solem преди 10 години
родител
ревизия
46e4ebfd4e
променени са 2 файла, в които са добавени 8 реда и са изтрити 1 реда
  1. 5 1
      celery/tests/app/test_loaders.py
  2. 3 0
      celery/tests/worker/test_control.py

+ 5 - 1
celery/tests/app/test_loaders.py

@@ -263,7 +263,11 @@ class test_autodiscovery(Case):
                 imp.return_value.__path__ = 'foo'
                 base.find_related_module(base, 'tasks')
 
-                imp.side_effect = AttributeError()
+
+                def se1(val):
+                    imp.side_effect = AttributeError()
+
+                imp.side_effect = se1
                 base.find_related_module(base, 'tasks')
                 imp.side_effect = None
 

+ 3 - 0
celery/tests/worker/test_control.py

@@ -140,6 +140,9 @@ class test_ControlPanel(AppCase):
         evd = consumer.event_dispatcher
         evd.groups = set()
         panel.handle('enable_events')
+        self.assertFalse(evd.groups)
+        evd.groups = set(['worker'])
+        panel.handle('enable_events')
         self.assertIn('task', evd.groups)
         evd.groups = {'task'}
         self.assertIn('already enabled', panel.handle('enable_events')['ok'])