Ask Solem 10 lat temu
rodzic
commit
46e4ebfd4e

+ 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'])