From 26d6b4a958a6ec4fcd9078f93fafb6842ac617c7 Mon Sep 17 00:00:00 2001 From: Pea Tyczynska Date: Tue, 26 Jan 2021 17:39:38 +0000 Subject: [PATCH] Mark broadcast message as stubbed when sent from training account --- app/broadcast_message/rest.py | 1 + tests/app/broadcast_message/test_rest.py | 7 ++++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/app/broadcast_message/rest.py b/app/broadcast_message/rest.py index b88b48d7b..e7c05d931 100644 --- a/app/broadcast_message/rest.py +++ b/app/broadcast_message/rest.py @@ -121,6 +121,7 @@ def create_broadcast_message(service_id): created_by_id=user.id, content=content, reference=reference, + stubbed=service.restricted ) dao_save_object(broadcast_message) diff --git a/tests/app/broadcast_message/test_rest.py b/tests/app/broadcast_message/test_rest.py index 2ec082267..47802cee5 100644 --- a/tests/app/broadcast_message/test_rest.py +++ b/tests/app/broadcast_message/test_rest.py @@ -4,6 +4,7 @@ import uuid from freezegun import freeze_time 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 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') -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) response = admin_request.post( @@ -138,6 +141,8 @@ def test_create_broadcast_message(admin_request, sample_broadcast_service): assert response['personalisation'] == {} 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', [ (