darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 17:31:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

Mark broadcast message as stubbed when sent from training account

Browse Source
This commit is contained in:
Pea Tyczynska
2021-01-26 17:39:38 +00:00
parent a93a35de8d
commit 26d6b4a958
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/broadcast_message/rest.py
View File

@@ -121,6 +121,7 @@ def create_broadcast_message(service_id):
created_by_id=user.id, created_by_id=user.id,
content=content, content=content,
reference=reference, reference=reference,
stubbed=service.restricted
) )
dao_save_object(broadcast_message) dao_save_object(broadcast_message)

7
tests/app/broadcast_message/test_rest.py
View File

@@ -4,6 +4,7 @@ import uuid
from freezegun import freeze_time from freezegun import freeze_time
import pytest import pytest
from app.dao.broadcast_message_dao import dao_get_broadcast_message_by_id_and_service_id
from app.models import BROADCAST_TYPE, BroadcastStatusType, BroadcastEventMessageType from app.models import BROADCAST_TYPE, BroadcastStatusType, BroadcastEventMessageType
from tests.app.db import create_broadcast_message, create_template, create_service, create_user from tests.app.db import create_broadcast_message, create_template, create_service, create_user
@@ -117,7 +118,9 @@ def test_get_broadcast_messages_for_service(admin_request, sample_broadcast_serv
@freeze_time('2020-01-01') @freeze_time('2020-01-01')
def test_create_broadcast_message(admin_request, sample_broadcast_service): @pytest.mark.parametrize('training_mode_service', [True, False])
def test_create_broadcast_message(admin_request, sample_broadcast_service, training_mode_service):
sample_broadcast_service.restricted = training_mode_service
t = create_template(sample_broadcast_service, BROADCAST_TYPE) t = create_template(sample_broadcast_service, BROADCAST_TYPE)
response = admin_request.post( response = admin_request.post(
@@ -138,6 +141,8 @@ def test_create_broadcast_message(admin_request, sample_broadcast_service):
assert response['personalisation'] == {} assert response['personalisation'] == {}
assert response['areas'] == [] assert response['areas'] == []
broadcast_message = dao_get_broadcast_message_by_id_and_service_id(response["id"], sample_broadcast_service.id)
assert broadcast_message.stubbed == training_mode_service
@pytest.mark.parametrize('data, expected_errors', [ @pytest.mark.parametrize('data, expected_errors', [
( (