diff --git a/queue_services/events-listener/src/events_listener/worker.py b/queue_services/events-listener/src/events_listener/worker.py index 1b7b97fae9..5795c1a307 100644 --- a/queue_services/events-listener/src/events_listener/worker.py +++ b/queue_services/events-listener/src/events_listener/worker.py @@ -73,12 +73,14 @@ async def process_event(event_message, flask_app): if message_type == LOCK_ACCOUNT_MESSAGE_TYPE: org.status_code = OrgStatus.NSF_SUSPENDED.value org.suspended_on = datetime.now() + org.suspension_reason_code = data.get('suspensionReasonCode', None) data = { 'accountId': org_id, } await publish_mailer_events(LOCK_ACCOUNT_MESSAGE_TYPE, org_id, data) elif message_type == UNLOCK_ACCOUNT_MESSAGE_TYPE: org.status_code = OrgStatus.ACTIVE.value + org.suspension_reason_code = None await publish_mailer_events(UNLOCK_ACCOUNT_MESSAGE_TYPE, org_id, data) else: logger.error('Unknown Message Type : %s', message_type) diff --git a/queue_services/events-listener/tests/integration/utils.py b/queue_services/events-listener/tests/integration/utils.py index 9e8bcdfae2..70d1c70405 100644 --- a/queue_services/events-listener/tests/integration/utils.py +++ b/queue_services/events-listener/tests/integration/utils.py @@ -15,6 +15,7 @@ import json import stan +from auth_api.utils.enums import SuspensionReasonCode async def helper_add_event_to_queue(stan_client: stan.aio.client.Client, @@ -31,7 +32,8 @@ async def helper_add_event_to_queue(stan_client: stan.aio.client.Client, 'datacontenttype': 'application/json', 'data': { 'accountId': org_id, - 'accountName': 'DEV - PAD01' + 'accountName': 'DEV - PAD01', + 'suspensionReasonCode': SuspensionReasonCode.OVERDUE_EFT.name } }