mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-04 02:11:11 -05:00
Added some more tests
This commit is contained in:
@@ -1,28 +1,9 @@
|
||||
from app import db
|
||||
from app.dao.dao_utils import transactional
|
||||
from app.errors import InvalidRequest
|
||||
from app.models import ServiceEmailReplyTo
|
||||
|
||||
|
||||
def create_or_update_email_reply_to(service_id, email_address):
|
||||
reply_to = dao_get_reply_to_by_service_id(service_id)
|
||||
if len(reply_to) == 0:
|
||||
reply_to = ServiceEmailReplyTo(service_id=service_id, email_address=email_address)
|
||||
dao_create_reply_to_email_address(reply_to)
|
||||
elif len(reply_to) == 1:
|
||||
reply_to[0].email_address = email_address
|
||||
dao_update_reply_to_email(reply_to[0])
|
||||
else:
|
||||
raise InvalidRequest(
|
||||
"Multiple reply to email addresses were found, this method should not be used.",
|
||||
status_code=500
|
||||
)
|
||||
|
||||
|
||||
@transactional
|
||||
def dao_create_reply_to_email_address(reply_to_email):
|
||||
db.session.add(reply_to_email)
|
||||
|
||||
|
||||
def dao_get_reply_to_by_service_id(service_id):
|
||||
reply_to = db.session.query(
|
||||
ServiceEmailReplyTo
|
||||
@@ -32,6 +13,27 @@ def dao_get_reply_to_by_service_id(service_id):
|
||||
return reply_to
|
||||
|
||||
|
||||
def create_or_update_email_reply_to(service_id, email_address):
|
||||
reply_to = dao_get_reply_to_by_service_id(service_id)
|
||||
if len(reply_to) == 0:
|
||||
reply_to = ServiceEmailReplyTo(service_id=service_id, email_address=email_address)
|
||||
dao_create_reply_to_email_address(reply_to)
|
||||
elif len(reply_to) == 1:
|
||||
reply_to[0].email_address = email_address
|
||||
dao_update_reply_to_email(reply_to[0])
|
||||
else:
|
||||
# Once we move allowing multiple email address this methods will be removed
|
||||
raise InvalidRequest(
|
||||
"Multiple reply to email addresses were found, this method should not be used.",
|
||||
status_code=500
|
||||
)
|
||||
|
||||
|
||||
@transactional
|
||||
def dao_create_reply_to_email_address(reply_to_email):
|
||||
db.session.add(reply_to_email)
|
||||
|
||||
|
||||
@transactional
|
||||
def dao_update_reply_to_email(reply_to):
|
||||
db.session.add(reply_to)
|
||||
|
||||
Reference in New Issue
Block a user