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

WIP trying to make end to end test

Browse Source
This commit is contained in:
Pea Tyczynska
2022-04-08 16:34:22 +01:00
parent b8de45128b
commit 253a9c9bc6
1 changed files with 33 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,6 +2,7 @@ import uuid
import pytest import pytest
from freezegun import freeze_time from freezegun import freeze_time
from unittest.mock import ANY, Mock, call
from app.dao.broadcast_message_dao import ( from app.dao.broadcast_message_dao import (
dao_get_broadcast_message_by_id_and_service_id, dao_get_broadcast_message_by_id_and_service_id,
@@ -18,6 +19,8 @@ from tests.app.db import (
create_user, create_user,
) )
from tests.conftest import set_config
def test_get_broadcast_message( def test_get_broadcast_message(
admin_request, admin_request,
@@ -386,39 +389,53 @@ def test_create_broadcast_message_400s_if_no_content_or_template(
# here we test that flow for cancelling broadcast works correctly from hitting the endpoint up util calling # here we test that flow for cancelling broadcast works correctly from hitting the endpoint up util calling
# lambdas with correct payloads # lambdas with correct payloads
def test_integration_for_cancel_broadcast_message( def test_integration_for_cancel_broadcast_message(
admin_request, sample_broadcast_service, mocker admin_request, sample_broadcast_service, mocker, notify_api
): ):
t = create_template(sample_broadcast_service, BROADCAST_TYPE, content='emergency broadcast') template = create_template(sample_broadcast_service, BROADCAST_TYPE, content='emergency broadcast')
bm = create_broadcast_message(t, status=BroadcastStatusType.BROADCASTING) broadcast_message = create_broadcast_message(template, status=BroadcastStatusType.BROADCASTING)
canceller = create_user(email='canceller@gov.uk') canceller = create_user(email='canceller@gov.uk')
sample_broadcast_service.users.append(canceller) sample_broadcast_service.users.append(canceller)
mock_task = mocker.patch('app.celery.broadcast_message_tasks.send_broadcast_event.apply_async') mocker.patch('app.celery.broadcast_message_tasks.notify_celery.send_task')
response = admin_request.post( mock_send_ticket_to_zendesk = mocker.patch(
'broadcast_message.update_broadcast_message_status', 'app.celery.broadcast_message_tasks.zendesk_client.send_ticket_to_zendesk',
_data={'status': BroadcastStatusType.CANCELLED, 'created_by': str(canceller.id)}, autospec=True,
service_id=t.service_id, )
broadcast_message_id=bm.id, mock_send_broadcast_provider_message = mocker.patch(
_expected_status=200 'app.celery.broadcast_message_tasks.send_broadcast_provider_message',
) )
assert len(bm.events) == 1 with set_config(notify_api, 'ENABLED_CBCS', ['ee', 'vodafone']):
cancel_event = bm.events[0] response = admin_request.post(
'broadcast_message.update_broadcast_message_status',
_data={'status': BroadcastStatusType.CANCELLED, 'created_by': str(canceller.id)},
service_id=template.service_id,
broadcast_message_id=broadcast_message.id,
_expected_status=200
)
cancel_id = str(cancel_event.id) assert len(broadcast_message.events) == 1
cancel_event = broadcast_message.events[0]
mock_task.assert_called_once_with(kwargs={'broadcast_event_id': cancel_id}, queue='broadcast-tasks')
assert response['status'] == BroadcastStatusType.CANCELLED assert response['status'] == BroadcastStatusType.CANCELLED
assert response['cancelled_at'] is not None assert response['cancelled_at'] is not None
assert response['cancelled_by_id'] == str(canceller.id) assert response['cancelled_by_id'] == str(canceller.id)
assert cancel_event.service_id == sample_broadcast_service.id assert cancel_event.service_id == sample_broadcast_service.id
assert cancel_event.transmitted_areas == bm.areas assert cancel_event.transmitted_areas == broadcast_message.areas
assert cancel_event.message_type == BroadcastEventMessageType.CANCEL assert cancel_event.message_type == BroadcastEventMessageType.CANCEL
assert cancel_event.transmitted_finishes_at == bm.finishes_at assert cancel_event.transmitted_finishes_at == broadcast_message.finishes_at
assert cancel_event.transmitted_content == {"body": "emergency broadcast"} assert cancel_event.transmitted_content == {"body": "emergency broadcast"}
assert mock_send_broadcast_provider_message.apply_async.call_args_list == [
call(kwargs={'broadcast_event_id': cancel_event.id, 'provider': 'ee'}, queue='broadcast-tasks'),
call(kwargs={'broadcast_event_id': cancel_event.id, 'provider': 'vodafone'}, queue='broadcast-tasks')
]
# we're on test env so this isn't called
assert mock_send_ticket_to_zendesk.called is False
@pytest.mark.parametrize('status', [ @pytest.mark.parametrize('status', [
BroadcastStatusType.DRAFT, BroadcastStatusType.DRAFT,