mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-04 18:31:13 -05:00
Only return non-archived email reply_to addresses
Updated the DAO methods which return a single email reply_to address and all reply_to addresses to only return the non-archived addresses. Changed the type of error that gets raised when using the Admin interface to be BadRequestError instead of a SQLAlchemyError.
This commit is contained in:
@@ -10,7 +10,8 @@ def dao_get_reply_to_by_service_id(service_id):
|
|||||||
reply_to = db.session.query(
|
reply_to = db.session.query(
|
||||||
ServiceEmailReplyTo
|
ServiceEmailReplyTo
|
||||||
).filter(
|
).filter(
|
||||||
ServiceEmailReplyTo.service_id == service_id
|
ServiceEmailReplyTo.service_id == service_id,
|
||||||
|
ServiceEmailReplyTo.archived == False # noqa
|
||||||
).order_by(desc(ServiceEmailReplyTo.is_default), desc(ServiceEmailReplyTo.created_at)).all()
|
).order_by(desc(ServiceEmailReplyTo.is_default), desc(ServiceEmailReplyTo.created_at)).all()
|
||||||
return reply_to
|
return reply_to
|
||||||
|
|
||||||
@@ -20,7 +21,8 @@ def dao_get_reply_to_by_id(service_id, reply_to_id):
|
|||||||
ServiceEmailReplyTo
|
ServiceEmailReplyTo
|
||||||
).filter(
|
).filter(
|
||||||
ServiceEmailReplyTo.service_id == service_id,
|
ServiceEmailReplyTo.service_id == service_id,
|
||||||
ServiceEmailReplyTo.id == reply_to_id
|
ServiceEmailReplyTo.id == reply_to_id,
|
||||||
|
ServiceEmailReplyTo.archived == False # noqa
|
||||||
).order_by(ServiceEmailReplyTo.created_at).one()
|
).order_by(ServiceEmailReplyTo.created_at).one()
|
||||||
return reply_to
|
return reply_to
|
||||||
|
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
|
from sqlalchemy.orm.exc import NoResultFound
|
||||||
|
|
||||||
from app.config import QueueNames
|
from app.config import QueueNames
|
||||||
from app.dao.service_email_reply_to_dao import dao_get_reply_to_by_id
|
from app.dao.service_email_reply_to_dao import dao_get_reply_to_by_id
|
||||||
from app.dao.service_sms_sender_dao import dao_get_service_sms_senders_by_id
|
from app.dao.service_sms_sender_dao import dao_get_service_sms_senders_by_id
|
||||||
@@ -89,7 +91,11 @@ def get_reply_to_text(notification_type, sender_id, service, template):
|
|||||||
reply_to = None
|
reply_to = None
|
||||||
if sender_id:
|
if sender_id:
|
||||||
if notification_type == EMAIL_TYPE:
|
if notification_type == EMAIL_TYPE:
|
||||||
reply_to = dao_get_reply_to_by_id(service.id, sender_id).email_address
|
try:
|
||||||
|
reply_to = dao_get_reply_to_by_id(service.id, sender_id).email_address
|
||||||
|
except NoResultFound:
|
||||||
|
message = 'Reply to email address not found'
|
||||||
|
raise BadRequestError(message=message)
|
||||||
elif notification_type == SMS_TYPE:
|
elif notification_type == SMS_TYPE:
|
||||||
reply_to = dao_get_service_sms_senders_by_id(service.id, sender_id).get_reply_to_text()
|
reply_to = dao_get_service_sms_senders_by_id(service.id, sender_id).get_reply_to_text()
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -25,6 +25,23 @@ def test_dao_get_reply_to_by_service_id(notify_db_session):
|
|||||||
assert second_reply_to == results[2]
|
assert second_reply_to == results[2]
|
||||||
|
|
||||||
|
|
||||||
|
def test_dao_get_reply_to_by_service_id_does_not_return_archived_reply_tos(notify_db_session):
|
||||||
|
service = create_service()
|
||||||
|
create_reply_to_email(service=service, email_address='something@email.com')
|
||||||
|
create_reply_to_email(service=service, email_address='another@email.com', is_default=False)
|
||||||
|
archived_reply_to = create_reply_to_email(
|
||||||
|
service=service,
|
||||||
|
email_address='second@email.com',
|
||||||
|
is_default=False,
|
||||||
|
archived=True
|
||||||
|
)
|
||||||
|
|
||||||
|
results = dao_get_reply_to_by_service_id(service_id=service.id)
|
||||||
|
|
||||||
|
assert len(results) == 2
|
||||||
|
assert archived_reply_to not in results
|
||||||
|
|
||||||
|
|
||||||
def test_add_reply_to_email_address_for_service_creates_first_email_for_service(notify_db_session):
|
def test_add_reply_to_email_address_for_service_creates_first_email_for_service(notify_db_session):
|
||||||
service = create_service()
|
service = create_service()
|
||||||
add_reply_to_email_address_for_service(service_id=service.id,
|
add_reply_to_email_address_for_service(service_id=service.id,
|
||||||
@@ -163,6 +180,18 @@ def test_dao_get_reply_to_by_id_raises_sqlalchemy_error_when_reply_to_does_not_e
|
|||||||
dao_get_reply_to_by_id(service_id=sample_service.id, reply_to_id=uuid.uuid4())
|
dao_get_reply_to_by_id(service_id=sample_service.id, reply_to_id=uuid.uuid4())
|
||||||
|
|
||||||
|
|
||||||
|
def test_dao_get_reply_to_by_id_raises_sqlalchemy_error_when_reply_to_is_archived(sample_service):
|
||||||
|
create_reply_to_email(service=sample_service, email_address='email@address.com')
|
||||||
|
archived_reply_to = create_reply_to_email(
|
||||||
|
service=sample_service,
|
||||||
|
email_address='email_two@address.com',
|
||||||
|
is_default=False,
|
||||||
|
archived=True)
|
||||||
|
|
||||||
|
with pytest.raises(SQLAlchemyError):
|
||||||
|
dao_get_reply_to_by_id(service_id=sample_service.id, reply_to_id=archived_reply_to.id)
|
||||||
|
|
||||||
|
|
||||||
def test_dao_get_reply_to_by_id_raises_sqlalchemy_error_when_service_does_not_exist(sample_service):
|
def test_dao_get_reply_to_by_id_raises_sqlalchemy_error_when_service_does_not_exist(sample_service):
|
||||||
reply_to = create_reply_to_email(service=sample_service, email_address='email@address.com')
|
reply_to = create_reply_to_email(service=sample_service, email_address='email@address.com')
|
||||||
with pytest.raises(SQLAlchemyError):
|
with pytest.raises(SQLAlchemyError):
|
||||||
|
|||||||
@@ -398,14 +398,16 @@ def create_monthly_billing_entry(
|
|||||||
|
|
||||||
|
|
||||||
def create_reply_to_email(
|
def create_reply_to_email(
|
||||||
service,
|
service,
|
||||||
email_address,
|
email_address,
|
||||||
is_default=True
|
is_default=True,
|
||||||
|
archived=False
|
||||||
):
|
):
|
||||||
data = {
|
data = {
|
||||||
'service': service,
|
'service': service,
|
||||||
'email_address': email_address,
|
'email_address': email_address,
|
||||||
'is_default': is_default,
|
'is_default': is_default,
|
||||||
|
'archived': archived,
|
||||||
}
|
}
|
||||||
reply_to = ServiceEmailReplyTo(**data)
|
reply_to = ServiceEmailReplyTo(**data)
|
||||||
|
|
||||||
|
|||||||
@@ -319,8 +319,9 @@ def test_send_one_off_notification_should_throw_exception_if_reply_to_id_doesnot
|
|||||||
'created_by': str(sample_email_template.service.created_by_id)
|
'created_by': str(sample_email_template.service.created_by_id)
|
||||||
}
|
}
|
||||||
|
|
||||||
with pytest.raises(expected_exception=SQLAlchemyError):
|
with pytest.raises(expected_exception=BadRequestError) as e:
|
||||||
send_one_off_notification(service_id=sample_email_template.service.id, post_data=data)
|
send_one_off_notification(service_id=sample_email_template.service.id, post_data=data)
|
||||||
|
assert e.value.message == 'Reply to email address not found'
|
||||||
|
|
||||||
|
|
||||||
def test_send_one_off_notification_should_throw_exception_if_sms_sender_id_doesnot_exist(
|
def test_send_one_off_notification_should_throw_exception_if_sms_sender_id_doesnot_exist(
|
||||||
|
|||||||
@@ -700,6 +700,30 @@ def test_post_email_notification_with_invalid_reply_to_id_returns_400(client, sa
|
|||||||
assert 'BadRequestError' in resp_json['errors'][0]['error']
|
assert 'BadRequestError' in resp_json['errors'][0]['error']
|
||||||
|
|
||||||
|
|
||||||
|
def test_post_email_notification_with_archived_reply_to_id_returns_400(client, sample_email_template, mocker):
|
||||||
|
archived_reply_to = create_reply_to_email(
|
||||||
|
sample_email_template.service,
|
||||||
|
'reply_to@test.com',
|
||||||
|
is_default=False,
|
||||||
|
archived=True)
|
||||||
|
mocker.patch('app.celery.provider_tasks.deliver_email.apply_async')
|
||||||
|
data = {
|
||||||
|
"email_address": 'test@test.com',
|
||||||
|
"template_id": sample_email_template.id,
|
||||||
|
'email_reply_to_id': archived_reply_to.id
|
||||||
|
}
|
||||||
|
auth_header = create_authorization_header(service_id=sample_email_template.service_id)
|
||||||
|
response = client.post(
|
||||||
|
path="v2/notifications/email",
|
||||||
|
data=json.dumps(data),
|
||||||
|
headers=[('Content-Type', 'application/json'), auth_header])
|
||||||
|
assert response.status_code == 400
|
||||||
|
resp_json = json.loads(response.get_data(as_text=True))
|
||||||
|
assert 'email_reply_to_id {} does not exist in database for service id {}'. \
|
||||||
|
format(archived_reply_to.id, sample_email_template.service_id) in resp_json['errors'][0]['message']
|
||||||
|
assert 'BadRequestError' in resp_json['errors'][0]['error']
|
||||||
|
|
||||||
|
|
||||||
def test_post_notification_with_document_upload(client, notify_db, notify_db_session, mocker):
|
def test_post_notification_with_document_upload(client, notify_db, notify_db_session, mocker):
|
||||||
service = sample_service(notify_db, notify_db_session, permissions=[EMAIL_TYPE, UPLOAD_DOCUMENT])
|
service = sample_service(notify_db, notify_db_session, permissions=[EMAIL_TYPE, UPLOAD_DOCUMENT])
|
||||||
template = create_sample_template(
|
template = create_sample_template(
|
||||||
|
|||||||
Reference in New Issue
Block a user