Browse Source

Fixed broken tests

Ask Solem 14 năm trước cách đây
mục cha
commit
37d6001e34
1 tập tin đã thay đổi với 7 bổ sung3 xóa
  1. 7 3
      celery/tests/test_bin/test_celeryd.py

+ 7 - 3
celery/tests/test_bin/test_celeryd.py

@@ -9,6 +9,7 @@ from StringIO import StringIO
 from celery import conf
 from celery import platforms
 from celery import signals
+from celery import log
 from celery.apps import worker as cd
 from celery.bin.celeryd import WorkerCommand, main as celeryd_main
 from celery.exceptions import ImproperlyConfigured
@@ -39,7 +40,7 @@ def disable_stdouts(fun):
 class _WorkController(object):
 
     def __init__(self, *args, **kwargs):
-        pass
+        self.logger = log.get_default_logger()
 
     def start(self):
         pass
@@ -156,15 +157,18 @@ class test_funs(unittest.TestCase):
 
     @disable_stdouts
     def test_set_process_status(self):
+        worker = Worker(hostname="xyzza")
         prev1, sys.argv = sys.argv, ["Arg0"]
         try:
-            st = cd.set_process_status("Running")
+            st = worker.set_process_status("Running")
             self.assertIn("celeryd", st)
+            self.assertIn("xyzza", st)
             self.assertIn("Running", st)
             prev2, sys.argv = sys.argv, ["Arg0", "Arg1"]
             try:
-                st = cd.set_process_status("Running")
+                st = worker.set_process_status("Running")
                 self.assertIn("celeryd", st)
+                self.assertIn("xyzza", st)
                 self.assertIn("Running", st)
                 self.assertIn("Arg1", st)
             finally: