notifications-api/tests/app/notifications/test_notification_statistics_rest.py

import json
from datetime import (
    date,
    timedelta
)

from flask import url_for
from tests import create_authorization_header
from tests.app.conftest import sample_notification_statistics as create_sample_notification_statistics

from freezegun import freeze_time


def test_get_notification_statistics_returns_empty_list_if_no_stats(notify_api,
                                                                    notify_db,
                                                                    notify_db_session,
                                                                    sample_template,
                                                                    sample_email_template):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:

            path = '/service/{}/notifications-statistics'.format(sample_email_template.service.id)

            auth_header = create_authorization_header(
                service_id=sample_email_template.service_id)

            response = client.get(path, headers=[auth_header])
            assert response.status_code == 200

            stats = json.loads(response.get_data(as_text=True))
            assert stats['data'] == []


def test_get_week_aggregate_statistics(notify_api,
                                       notify_db,
                                       notify_db_session,
                                       sample_service):
    with notify_api.test_request_context():
        sample_notification_statistics = create_sample_notification_statistics(
            notify_db,
            notify_db_session,
            day=date(date.today().year, 4, 1))
        with notify_api.test_client() as client:
            endpoint = url_for(
                'notifications-statistics.get_notification_statistics_for_service_seven_day_aggregate',
                service_id=sample_service.id)
            auth_header = create_authorization_header(
                service_id=sample_service.id)

            resp = client.get(endpoint, headers=[auth_header])
            assert resp.status_code == 200
            json_resp = json.loads(resp.get_data(as_text=True))
            week_len_index = len(json_resp['data']) - 1
            assert json_resp['data'][week_len_index]['emails_requested'] == 2
            assert json_resp['data'][week_len_index]['sms_requested'] == 2
            assert json_resp['data'][week_len_index]['week_start'] == date(date.today().year, 4, 1).strftime('%Y-%m-%d')
            assert json_resp['data'][week_len_index]['week_end'] == date(date.today().year, 4, 7).strftime('%Y-%m-%d')


def test_get_week_aggregate_statistics_date_from(notify_api,
                                                 notify_db,
                                                 notify_db_session,
                                                 sample_service):
    with notify_api.test_request_context():
        sample_notification_statistics = create_sample_notification_statistics(
            notify_db,
            notify_db_session,
            day=date(date.today().year, 4, 1))
        date_from_str = date(date.today().year, 4, 1).strftime('%Y-%m-%d')
        with notify_api.test_client() as client:
            endpoint = url_for(
                'notifications-statistics.get_notification_statistics_for_service_seven_day_aggregate',
                service_id=sample_service.id,
                date_from=date_from_str)
            auth_header = create_authorization_header(
                service_id=sample_service.id)

            resp = client.get(endpoint, headers=[auth_header])
            assert resp.status_code == 200
            json_resp = json.loads(resp.get_data(as_text=True))
            week_len_index = len(json_resp['data']) - 1
            assert json_resp['data'][week_len_index]['emails_requested'] == 2
            assert json_resp['data'][week_len_index]['sms_requested'] == 2
            assert json_resp['data'][week_len_index]['week_start'] == date_from_str
            assert json_resp['data'][week_len_index]['week_end'] == date(date.today().year, 4, 7).strftime('%Y-%m-%d')


def test_get_week_aggregate_statistics_date_in_future(notify_api,
                                                      notify_db,
                                                      notify_db_session,
                                                      sample_service):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            endpoint = url_for(
                'notifications-statistics.get_notification_statistics_for_service_seven_day_aggregate',
                service_id=sample_service.id,
                date_from=(date.today() + timedelta(days=1)).strftime('%Y-%m-%d'))
            auth_header = create_authorization_header(
                service_id=sample_service.id)

            resp = client.get(endpoint, headers=[auth_header])
            assert resp.status_code == 400
            json_resp = json.loads(resp.get_data(as_text=True))
            assert json_resp['result'] == 'error'
            assert json_resp['message']['date_from'][0] == 'Date cannot be in the future'


def test_get_week_aggregate_statistics_invalid_week_count(notify_api,
                                                          notify_db,
                                                          notify_db_session,
                                                          sample_service):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            endpoint = url_for(
                'notifications-statistics.get_notification_statistics_for_service_seven_day_aggregate',
                service_id=sample_service.id,
                week_count=-1)
            auth_header = create_authorization_header(
                service_id=sample_service.id)

            resp = client.get(endpoint, headers=[auth_header])
            assert resp.status_code == 400
            json_resp = json.loads(resp.get_data(as_text=True))
            assert json_resp['result'] == 'error'
            assert json_resp['message']['week_count'][0] == 'Not a positive integer'


@freeze_time('2016-01-01')
def test_get_notification_statistics_for_specific_day(notify_api,
                                                      notify_db,
                                                      notify_db_session,
                                                      sample_template):
    the_day = date.today()

    sample_notification_statistics = create_sample_notification_statistics(
        notify_db,
        notify_db_session,
        day=the_day)

    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            path = '/service/{}/notifications-statistics/day/{}'.format(sample_template.service_id, the_day)
            auth_header = create_authorization_header(service_id=sample_template.service_id)
            response = client.get(path, headers=[auth_header])
            assert response.status_code == 200
            stats = json.loads(response.get_data(as_text=True))

            assert stats['data']['id'] == str(sample_notification_statistics.id)
            assert stats['data']['day'] == the_day.strftime('%Y-%m-%d')

            another_day = the_day - timedelta(days=1)
            path = '/service/{}/notifications-statistics/day/{}'.format(sample_template.service_id, another_day)

            response = client.get(path, headers=[auth_header])
            assert response.status_code == 404


@freeze_time('2016-01-01')
def test_get_notification_statistics_for_specific_day_returns_404_if_no_stats(notify_api,
                                                                              notify_db,
                                                                              notify_db_session,
                                                                              sample_template):
    the_day = date.today()

    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            path = '/service/{}/notifications-statistics/day/{}'.format(sample_template.service_id, the_day)
            auth_header = create_authorization_header(service_id=sample_template.service_id)
            response = client.get(path, headers=[auth_header])
            assert response.status_code == 404


@freeze_time('2016-01-01')
def test_get_notification_statistics_for_specific_day_returns_400_for_incorrect_date(notify_api,
                                                                                     notify_db,
                                                                                     notify_db_session,
                                                                                     sample_template):
    the_day = date.today()
    incorrect_date_format = the_day.strftime('%d-%m-%Y')

    create_sample_notification_statistics(
        notify_db,
        notify_db_session,
        day=the_day)

    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            path = '/service/{}/notifications-statistics/day/{}'.format(
                sample_template.service_id,
                incorrect_date_format)
            auth_header = create_authorization_header(service_id=sample_template.service_id)
            response = client.get(path, headers=[auth_header])
            assert response.status_code == 400

            another_dodgy_date = 'fish'
            path = '/service/{}/notifications-statistics/day/{}'.format(
                sample_template.service_id,
                another_dodgy_date)

            response = client.get(path, headers=[auth_header])
            assert response.status_code == 400