notifications-admin/tests/app/notify_client/test_notification_client.py

import pytest

from app.notify_client.notification_api_client import NotificationApiClient


@pytest.mark.parametrize("arguments,expected_call", [
    (
        {},
        {'url': '/service/abcd1234/notifications', 'params': {}}
    ),
    (
        {'page': 99},
        {'url': '/service/abcd1234/notifications', 'params': {'page': 99}}
    ),
    (
        {'include_jobs': False},
        {'url': '/service/abcd1234/notifications', 'params': {'include_jobs': False}}
    ),
    (
        {'include_from_test_key': True},
        {'url': '/service/abcd1234/notifications', 'params': {'include_from_test_key': True}}
    ),
    (
        {'page': 48, 'limit_days': 3},
        {'url': '/service/abcd1234/notifications', 'params': {'page': 48, 'limit_days': 3}}
    ),
    (
        {'job_id': 'efgh5678'},
        {'url': '/service/abcd1234/job/efgh5678/notifications', 'params': {}}
    ),
    (
        {'job_id': 'efgh5678', 'page': 48},
        {'url': '/service/abcd1234/job/efgh5678/notifications', 'params': {'page': 48}}
    ),
    (
        {'job_id': 'efgh5678', 'page': 48, 'limit_days': 3},
        {'url': '/service/abcd1234/job/efgh5678/notifications', 'params': {'page': 48}}
    ),
])
def test_client_gets_notifications_for_service_and_job_by_page(mocker, arguments, expected_call):

    mock_get = mocker.patch('app.notify_client.notification_api_client.NotificationApiClient.get')
    NotificationApiClient().get_notifications_for_service('abcd1234', **arguments)
    mock_get.assert_called_once_with(**expected_call)


@pytest.mark.parametrize("arguments,expected_call", [
    (
        {'to': "2028675309"},
        {'url': '/service/abcd1234/notifications', 'data': {'to': "2028675309"}}
    ),
    (
        {'to': "2028675309", 'job_id': 'efgh5678'},
        {'url': '/service/abcd1234/job/efgh5678/notifications', 'data': {'to': "2028675309"}}
    ),
    (
        {'to': "2028675309", 'page': 99},
        {'url': '/service/abcd1234/notifications', 'data': {'to': "2028675309", 'page': 99}}
    ),
    (
        {'to': "2028675309", 'limit_days': 3},
        {'url': '/service/abcd1234/notifications', 'data': {'to': "2028675309", 'limit_days': 3}}
    ),
    (
        {'to': "2028675309", 'job_id': 'efgh5678', 'limit_days': 3},
        {'url': '/service/abcd1234/job/efgh5678/notifications', 'data': {'to': "2028675309"}}
    ),
])
def test_client_gets_notifications_for_service_and_job_by_page_posts_for_to(mocker, arguments, expected_call):

    mock_post = mocker.patch('app.notify_client.notification_api_client.NotificationApiClient.post')
    NotificationApiClient().get_notifications_for_service('abcd1234', **arguments)
    mock_post.assert_called_once_with(**expected_call)


def test_send_notification(mocker, client_request, active_user_with_permissions):
    mock_post = mocker.patch('app.notify_client.notification_api_client.NotificationApiClient.post')
    NotificationApiClient().send_notification(
        'foo',
        template_id='bar',
        recipient='2028675301',
        personalisation=None,
        sender_id=None
    )
    mock_post.assert_called_once_with(
        url='/service/foo/send-notification',
        data={
            'template_id': 'bar',
            'to': '2028675301',
            'personalisation': None,
            'created_by': active_user_with_permissions['id']
        }
    )


def test_get_notification(mocker):
    mock_get = mocker.patch('app.notify_client.notification_api_client.NotificationApiClient.get')
    NotificationApiClient().get_notification('foo', 'bar')
    mock_get.assert_called_once_with(
        url='/service/foo/notifications/bar'
    )


def test_get_notification_count_for_job_id(mocker):
    mock_get = mocker.patch('app.notify_client.notification_api_client.NotificationApiClient.get')
    NotificationApiClient().get_notification_count_for_job_id(service_id='foo', job_id='bar')
    mock_get.assert_called_once_with(
        url='/service/foo/job/bar/notification_count',
    )