darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-25 00:33:41 -04:00
Code Issues Packages Projects Releases Wiki Activity

add tests for get_earlier_events_for_broadcast_event

Browse Source
This commit is contained in:
Leo Hemsted
2020-07-27 15:33:29 +01:00
parent 3dd15841a5
commit bdf221a421
3 changed files with 74 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/app/db.py
View File

@@ -62,6 +62,7 @@ from app.models import (
ServiceContactList,
BroadcastMessage,
BroadcastStatusType,
BroadcastEvent
)
@@ -1021,3 +1022,29 @@ def create_broadcast_message(
db.session.add(broadcast_message)
db.session.commit()
return broadcast_message
def create_broadcast_event(
broadcast_message,
sent_at=None,
message_type='alert',
transmitted_content=None,
transmitted_areas=None,
transmitted_sender=None,
transmitted_starts_at=None,
transmitted_finishes_at=None,
):
b_e = BroadcastEvent(
service=broadcast_message.service,
broadcast_message=broadcast_message,
sent_at=sent_at or datetime.utcnow(),
message_type=message_type,
transmitted_content=transmitted_content or {'body': 'this is an emergency broadcast message'},
transmitted_areas=transmitted_areas or ['london'],
transmitted_sender=transmitted_sender or 'www.notifications.service.gov.uk',
transmitted_starts_at=transmitted_starts_at,
transmitted_finishes_at=transmitted_finishes_at,
)
db.session.add(b_e)
db.session.commit()
return b_e