Revert "ensure we're passing through api keys and key types from notifications"

This reverts commit 25d1777937.
This commit is contained in:
Martyn Inglis
2017-03-30 10:38:41 +01:00
parent bf99921cff
commit 385a73da2b
4 changed files with 20 additions and 85 deletions

View File

@@ -1,6 +1,6 @@
import uuid
from datetime import datetime
from unittest.mock import Mock, ANY
from unittest.mock import Mock
import pytest
from flask import current_app
@@ -65,22 +65,17 @@ def test_should_have_decorated_tasks_functions():
def email_job_with_placeholders(notify_db, notify_db_session, sample_email_template_with_placeholders):
return sample_job(notify_db, notify_db_session, template=sample_email_template_with_placeholders)
# --- send_notification_to_persist_queue tests -- #
@freeze_time("2016-01-01 11:09:00.061258")
def test_should_put_sms_messages_in_db_sms_queue(sample_service, sample_template, mocker):
mocker.patch('app.celery.tasks.send_sms.apply_async')
send_notification_to_persist_queue(
'notification-id', sample_service, sample_template.template_type, 'encrypted', 'api_key_id', 'key_type'
)
send_notification_to_persist_queue("notification-id", sample_service, sample_template.template_type, "encrypted")
tasks.send_sms.apply_async.assert_called_once_with(
(str(sample_service.id),
"notification-id",
"encrypted",
"2016-01-01T11:09:00.061258Z"),
kwargs=ANY,
queue="db-sms"
)
@@ -89,8 +84,7 @@ def test_should_put_sms_messages_in_db_sms_queue(sample_service, sample_template
def test_should_put_messages_in_priority_queue(sample_service, sample_template, mocker):
mocker.patch('app.celery.tasks.send_sms.apply_async')
send_notification_to_persist_queue(
'notification-id', sample_service, sample_template.template_type, 'encrypted', 'api_key_id', 'key_type',
priority=True
"notification-id", sample_service, sample_template.template_type, "encrypted", True
)
tasks.send_sms.apply_async.assert_called_once_with(
@@ -98,47 +92,15 @@ def test_should_put_messages_in_priority_queue(sample_service, sample_template,
"notification-id",
"encrypted",
"2016-01-01T11:09:00.061258Z"),
kwargs=ANY,
queue="notify"
)
@freeze_time("2016-01-01 11:09:00.061258")
def test_should_pass_through_api_key_info(sample_service, sample_template, mocker):
mocker.patch('app.celery.tasks.send_sms.apply_async')
api_key_id = 'foo'
key_type = 'bar'
send_notification_to_persist_queue(
"notification-id",
sample_service,
sample_template.template_type,
"encrypted",
api_key_id,
key_type
)
tasks.send_sms.apply_async.assert_called_once_with(
ANY,
kwargs={
'api_key_id': api_key_id,
'key_type': key_type
},
queue=ANY
)
@freeze_time("2016-01-01 11:09:00.061258")
def test_should_put_messages_in_research_mode_queue(sample_service, sample_template, mocker):
mocker.patch('app.celery.tasks.send_sms.apply_async')
send_notification_to_persist_queue(
"notification-id",
sample_service,
sample_template.template_type,
"encrypted",
'api_key_id',
'key_type',
priority=False,
research_mode=True
"notification-id", sample_service, sample_template.template_type, "encrypted", False, True
)
tasks.send_sms.apply_async.assert_called_once_with(
@@ -146,7 +108,6 @@ def test_should_put_messages_in_research_mode_queue(sample_service, sample_templ
"notification-id",
"encrypted",
"2016-01-01T11:09:00.061258Z"),
kwargs=ANY,
queue="research-mode"
)
@@ -155,14 +116,7 @@ def test_should_put_messages_in_research_mode_queue(sample_service, sample_templ
def test_should_put_messages_in_research_mode_queue_overriding_priority_mode(sample_service, sample_template, mocker):
mocker.patch('app.celery.tasks.send_sms.apply_async')
send_notification_to_persist_queue(
"notification-id",
sample_service,
sample_template.template_type,
"encrypted",
'api_key_id',
'key_type',
priority=True,
research_mode=True
"notification-id", sample_service, sample_template.template_type, "encrypted", True, True
)
tasks.send_sms.apply_async.assert_called_once_with(
@@ -170,7 +124,6 @@ def test_should_put_messages_in_research_mode_queue_overriding_priority_mode(sam
"notification-id",
"encrypted",
"2016-01-01T11:09:00.061258Z"),
kwargs=ANY,
queue="research-mode"
)
@@ -179,7 +132,7 @@ def test_should_put_messages_in_research_mode_queue_overriding_priority_mode(sam
def test_should_put_email_messages_in_db_email_queue(sample_service, sample_email_template, mocker):
mocker.patch('app.celery.tasks.send_email.apply_async')
send_notification_to_persist_queue(
'notification-id', sample_service, sample_email_template.template_type, 'encrypted', 'api_key_id', 'key_type'
"notification-id", sample_service, sample_email_template.template_type, "encrypted"
)
tasks.send_email.apply_async.assert_called_once_with(
@@ -187,7 +140,6 @@ def test_should_put_email_messages_in_db_email_queue(sample_service, sample_emai
"notification-id",
"encrypted",
"2016-01-01T11:09:00.061258Z"),
kwargs=ANY,
queue="db-email"
)
@@ -502,8 +454,6 @@ def test_process_row_sends_letter_task(template_type, research_mode, expected_fu
),
queue=expected_queue
)
# -------- send_sms and send_email tests -------- #