def test_handle_api_request_v0_send(mocker):
from iris.bin.sender import message_send_enqueue
from iris.sender.rpc import handle_api_request, send_funcs
from iris.sender.shared import per_mode_send_queues
send_funcs['message_send_enqueue'] = message_send_enqueue
send_queue = per_mode_send_queues.setdefault('email', gevent.queue.Queue())
# support expanding target
mocker.patch('iris.sender.cache.targets_for_role', lambda role, target: [target])
mocker.patch('iris.bin.sender.db')
mocker.patch('iris.metrics.stats')
mocker.patch('iris.bin.sender.set_target_contact').return_value = True
mock_address = mocker.MagicMock()
mock_socket = mocker.MagicMock()
mock_socket.recv.return_value = msgpack.packb({
'endpoint': 'v0/send',
'data': fake_notification,
})
while send_queue.qsize() > 0:
send_queue.get()
handle_api_request(mock_socket, mock_address)
assert send_queue.qsize() == 1
m = send_queue.get()
assert m['subject'] == '[%s] %s' % (fake_notification['application'],
fake_notification['subject'])
def test_handle_api_request_v0_send_with_mode(mocker):
from iris.sender.rpc import handle_api_request
from iris.sender.shared import send_queue
# support expanding target
mocker.patch('iris.sender.cache.RoleTargets.__call__', lambda _, role, target: [target])
mocker.patch('iris.bin.sender.set_target_contact')
fake_mode_notification = {}
fake_mode_notification.update(fake_notification)
fake_mode_notification['mode'] = 'sms'
mock_address = mocker.MagicMock()
mock_socket = mocker.MagicMock()
mock_socket.recv.return_value = msgpack.packb({
'endpoint': 'v0/send',
'data': fake_mode_notification,
})
while send_queue.qsize() > 0:
send_queue.get()
handle_api_request(mock_socket, mock_address)
assert send_queue.qsize() == 1
m = send_queue.get()
assert m['subject'] == '[%s] %s' % (fake_mode_notification['application'],
fake_mode_notification['subject'])
