Browse Source

Trying to get some $#™£¡™ log output from a test case

Ask Solem 11 years ago
parent
commit
497b92c7b3
1 changed files with 23 additions and 20 deletions
  1. 23 20
      celery/tests/worker/test_worker.py

+ 23 - 20
celery/tests/worker/test_worker.py

@@ -1,4 +1,4 @@
-from __future__ import absolute_import
+from __future__ import absolute_import, print_function
 
 import os
 import socket
@@ -516,49 +516,52 @@ class test_Consumer(AppCase):
         self.assertTrue(logger.critical.call_count)
 
     def test_receive_message_eta(self):
-        print('TEST RECEIVE MESSAGE ETA')
-        print('+CREATE MYKOMBUCONSUMER')
+        import sys
+        from functools import partial
+        pp = partial(print, file=sys.__stderr__)
+        pp('TEST RECEIVE MESSAGE ETA')
+        pp('+CREATE MYKOMBUCONSUMER')
         l = _MyKombuConsumer(self.buffer.put, timer=self.timer, app=self.app)
-        print('-CREATE MYKOMBUCONSUMER')
+        pp('-CREATE MYKOMBUCONSUMER')
         l.steps.pop()
         l.event_dispatcher = mock_event_dispatcher()
         backend = Mock()
-        print('+ CREATE MESSAGE')
+        pp('+ CREATE MESSAGE')
         m = create_message(
             backend, task=self.foo_task.name,
             args=[2, 4, 8], kwargs={},
             eta=(datetime.now() + timedelta(days=1)).isoformat(),
         )
-        print('- CREATE MESSAGE')
+        pp('- CREATE MESSAGE')
 
         try:
-            print('+ BLUEPRINT START 1')
+            pp('+ BLUEPRINT START 1')
             l.blueprint.start(l)
-            print('- BLUEPRINT START 1')
+            pp('- BLUEPRINT START 1')
             p = l.app.conf.BROKER_CONNECTION_RETRY
             l.app.conf.BROKER_CONNECTION_RETRY = False
-            print('+ BLUEPRINT START 2')
+            pp('+ BLUEPRINT START 2')
             l.blueprint.start(l)
-            print('- BLUEPRINT START 2')
+            pp('- BLUEPRINT START 2')
             l.app.conf.BROKER_CONNECTION_RETRY = p
-            print('+ BLUEPRINT RESTART')
+            pp('+ BLUEPRINT RESTART')
             l.blueprint.restart(l)
-            print('- BLUEPRINT RESTART')
+            pp('- BLUEPRINT RESTART')
             l.event_dispatcher = mock_event_dispatcher()
-            print('+ GET ON MESSAGE')
+            pp('+ GET ON MESSAGE')
             callback = self._get_on_message(l)
-            print('- GET ON MESSAGE')
-            print('+ CALLBACK')
+            pp('- GET ON MESSAGE')
+            pp('+ CALLBACK')
             callback(m.decode(), m)
-            print('- CALLBACK')
+            pp('- CALLBACK')
         finally:
-            print('+ STOP TIMER')
+            pp('+ STOP TIMER')
             l.timer.stop()
-            print('- STOP TIMER')
+            pp('- STOP TIMER')
             try:
-                print('+ JOIN TIMER')
+                pp('+ JOIN TIMER')
                 l.timer.join()
-                print('- JOIN TIMER')
+                pp('- JOIN TIMER')
             except RuntimeError:
                 pass