def test_remove_service_safelist_does_not_commit(notify_db, sample_service_safelist):
dao_remove_service_safelist(sample_service_safelist.service_id)
# since dao_remove_service_safelist doesn't commit, we can still rollback its changes
notify_db.session.rollback()
assert ServiceSafelist.query.count() == 1
def update_safelist(service_id):
# doesn't commit so if there are any errors, we preserve old values in db
dao_remove_service_safelist(service_id)
try:
safelist_objs = get_safelist_objects(service_id, request.get_json())
except ValueError as e:
current_app.logger.exception(e)
dao_rollback()
msg = '{} is not a valid email address or phone number'.format(str(e))
raise InvalidRequest(msg, 400)
else:
dao_add_and_commit_safelisted_contacts(safelist_objs)
return '', 204
def test_remove_service_safelist_only_removes_for_my_service(notify_db, notify_db_session):
service_1 = create_service(notify_db, notify_db_session, service_name="service 1")
service_2 = create_service(notify_db, notify_db_session, service_name="service 2")
dao_add_and_commit_safelisted_contacts(
[
ServiceSafelist.from_string(service_1.id, EMAIL_TYPE, "*****@*****.**"),
ServiceSafelist.from_string(service_2.id, EMAIL_TYPE, "*****@*****.**"),
]
)
dao_remove_service_safelist(service_1.id)
assert service_1.safelist == []
assert len(service_2.safelist) == 1
