Recognise that broadcast messages include content

The content attribute needs to be added to the model so we can use it
later.
This commit is contained in:
Chris Hill-Scott
2020-10-08 14:06:33 +01:00
parent 30cafce022
commit 850571cb8c
3 changed files with 14 additions and 0 deletions

View File

@@ -24,6 +24,7 @@ class BroadcastMessage(JSONModel):
'template_id',
'template_name',
'template_version',
'content',
'service_id',
'created_by',
'personalisation',

View File

@@ -650,6 +650,7 @@ def broadcast_message_json(
approved_by_id=None,
cancelled_by_id=None,
areas=None,
content=None,
):
return {
'id': id_,
@@ -659,6 +660,7 @@ def broadcast_message_json(
'template_id': template_id,
'template_version': 123,
'template_name': 'Example template',
'content': 'This is a test',
'personalisation': {},
'areas': areas or [

View File

@@ -34,3 +34,14 @@ def test_simple_polygons(fake_uuid):
# total coordinates
[51],
]
def test_content_comes_from_attribute_not_template(fake_uuid):
broadcast_message = BroadcastMessage(broadcast_message_json(
id_=fake_uuid,
service_id=fake_uuid,
template_id=fake_uuid,
status='draft',
created_by_id=fake_uuid,
))
assert broadcast_message.content == 'This is a test'