tornado: Use call_consume_in_tests for message_sender queue.

This increases test coverage of queue_processors.py significantly,
with no real cost, achieving valuable progress torwards #6542.
This commit is contained in:
Tim Abbott
2017-11-03 14:07:19 -07:00
parent f382875d5a
commit eade4d0052
2 changed files with 4 additions and 5 deletions

View File

@@ -220,7 +220,7 @@ class SocketConnection(sockjs.tornado.SockJSConnection):
return_queue="tornado_return",
log_data=log_data,
request_environ=dict(REMOTE_ADDR=self.session.conn_info.ip))),
fake_message_sender)
fake_message_sender, call_consume_in_tests=True)
def on_close(self) -> None:
log_data = dict(extra='[transport=%s]' % (self.session.transport_name,))

View File

@@ -33,7 +33,7 @@ from zerver.lib.send_email import send_future_email, send_email_from_dict, \
FromAddress, EmailNotDeliveredException
from zerver.lib.email_mirror import process_message as mirror_email
from zerver.decorator import JsonableError
from zerver.tornado.socket import req_redis_key
from zerver.tornado.socket import req_redis_key, respond_send_message
from confirmation.models import Confirmation, create_confirmation_link
from zerver.lib.db import reset_queries
from zerver.lib.redis_utils import get_redis_client
@@ -393,9 +393,8 @@ class MessageSenderWorker(QueueProcessingWorker):
self.redis_client.hmset(redis_key, {'status': 'complete',
'response': resp_content})
# Since this sends back to Tornado, we can't use
# call_consume_in_tests here.
queue_json_publish(server_meta['return_queue'], result, lambda e: None)
queue_json_publish(server_meta['return_queue'], result,
respond_send_message)
@assign_queue('digest_emails')
class DigestWorker(QueueProcessingWorker):