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

from app.notify_client.complaint_api_client import ComplaintApiClient


def test_get_all_complaints(mocker):
    client = ComplaintApiClient()

    mock = mocker.patch('app.notify_client.complaint_api_client.ComplaintApiClient.get')

    client.get_all_complaints()
    mock.assert_called_once_with('/complaint', params={'page': 1})


def test_get_all_complaints_with_a_page_number_specified(mocker):
    client = ComplaintApiClient()

    mock = mocker.patch('app.notify_client.complaint_api_client.ComplaintApiClient.get')

    client.get_all_complaints(page=3)
    mock.assert_called_once_with('/complaint', params={'page': 3})


def test_get_complaint_count(mocker):
    client = ComplaintApiClient()
    mock = mocker.patch.object(client, 'get')
    params_dict = {'start_date': '2018-06-01', 'end_date': '2018-06-15'}

    client.get_complaint_count(params_dict=params_dict)
    mock.assert_called_once_with('/complaint/count-by-date-range', params=params_dict)