diff --git a/app/service/service_broadcast_settings_schema.py b/app/service/service_broadcast_settings_schema.py index 494b08a59..cc98a718f 100644 --- a/app/service/service_broadcast_settings_schema.py +++ b/app/service/service_broadcast_settings_schema.py @@ -4,7 +4,7 @@ service_broadcast_settings_schema = { "type": "object", "title": "Set a services broadcast settings", "properties": { - "broadcast_channel": {"enum": ["test", "severe"]}, + "broadcast_channel": {"enum": ["test", "severe", "government"]}, "service_mode": {"enum": ["training", "live"]}, "provider_restriction": {"enum": ["three", "o2", "vodafone", "ee", "all"]} }, diff --git a/tests/app/celery/test_broadcast_message_tasks.py b/tests/app/celery/test_broadcast_message_tasks.py index 8d65108b6..eca8b2ce2 100644 --- a/tests/app/celery/test_broadcast_message_tasks.py +++ b/tests/app/celery/test_broadcast_message_tasks.py @@ -221,7 +221,7 @@ def test_send_broadcast_provider_message_sends_data_correctly( ['o2', 'O2'], ['vodafone', 'Vodafone'], ]) -@pytest.mark.parametrize('channel', ['test', 'severe']) +@pytest.mark.parametrize('channel', ['test', 'severe', 'government']) def test_send_broadcast_provider_message_uses_channel_set_on_broadcast_service( notify_db, mocker, sample_broadcast_service, provider, provider_capitalised, channel ): diff --git a/tests/app/service/test_rest.py b/tests/app/service/test_rest.py index b8ed01941..c6d666a2d 100644 --- a/tests/app/service/test_rest.py +++ b/tests/app/service/test_rest.py @@ -292,8 +292,10 @@ def test_get_service_by_id(admin_request, sample_service): @pytest.mark.parametrize('broadcast_channel,allowed_broadcast_provider', ( ('test', 'all'), ('severe', 'all'), + ('government', 'all'), ('test', 'ee'), ('severe', 'three'), + ('government', 'vodafone'), )) def test_get_service_by_id_for_broadcast_service_returns_broadcast_keys( notify_db, admin_request, sample_broadcast_service, broadcast_channel, allowed_broadcast_provider @@ -3709,7 +3711,7 @@ def test_get_returned_letter(admin_request, sample_letter_template): assert response[4]['uploaded_letter_file_name'] == 'filename.pdf' -@pytest.mark.parametrize('channel', ["test", "severe"]) +@pytest.mark.parametrize('channel', ["test", "severe", "government"]) def test_set_as_broadcast_service_sets_broadcast_channel( admin_request, sample_service, broadcast_organisation, channel ): @@ -3759,7 +3761,7 @@ def test_set_as_broadcast_service_updates_channel_for_broadcast_service( assert records[0].channel == "test" -@pytest.mark.parametrize('channel', ["government", "extreme", "exercise", "random", ""]) +@pytest.mark.parametrize('channel', ["extreme", "exercise", "random", ""]) def test_set_as_broadcast_service_rejects_unknown_channels( admin_request, sample_service, broadcast_organisation, channel ):