darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 18:01:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix tests to create notifications properly

Browse Source 
migrated away from calling fixtures as functions
This commit is contained in:
Leo Hemsted
2018-09-25 15:59:08 +01:00
parent 3739c573ea
commit 69e08fa23b
3 changed files with 46 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
tests/app/celery/test_reporting_tasks.py
View File

@@ -1,5 +1,10 @@
from datetime import datetime, timedelta, date from datetime import datetime, timedelta, date
from tests.app.conftest import sample_notification from decimal import Decimal
import pytest
from freezegun import freeze_time
from sqlalchemy import desc
from app.celery.reporting_tasks import create_nightly_billing, create_nightly_notification_status from app.celery.reporting_tasks import create_nightly_billing, create_nightly_notification_status
from app.dao.fact_billing_dao import get_rate from app.dao.fact_billing_dao import get_rate
from app.models import ( from app.models import (
@@ -9,12 +14,8 @@ from app.models import (
EMAIL_TYPE, EMAIL_TYPE,
SMS_TYPE, FactNotificationStatus SMS_TYPE, FactNotificationStatus
) )
from decimal import Decimal
import pytest
from app.models import LetterRate, Rate from app.models import LetterRate, Rate
from app import db from app import db
from freezegun import freeze_time
from sqlalchemy import desc
from tests.app.db import create_service, create_template, create_notification from tests.app.db import create_service, create_template, create_notification
@@ -38,8 +39,6 @@ def mocker_get_rate(
[(1.0, 1, 2, [1]), [(1.0, 1, 2, [1]),
(2.0, 2, 1, [1, 2])]) (2.0, 2, 1, [1, 2])])
def test_create_nightly_billing_sms_rate_multiplier( def test_create_nightly_billing_sms_rate_multiplier(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
mocker, mocker,
@@ -53,11 +52,8 @@ def test_create_nightly_billing_sms_rate_multiplier(
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
# These are sms notifications # These are sms notifications
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by='mmg', sent_by='mmg',
@@ -65,11 +61,8 @@ def test_create_nightly_billing_sms_rate_multiplier(
rate_multiplier=1.0, rate_multiplier=1.0,
billable_units=1, billable_units=1,
) )
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by='mmg', sent_by='mmg',
@@ -94,8 +87,6 @@ def test_create_nightly_billing_sms_rate_multiplier(
def test_create_nightly_billing_different_templates( def test_create_nightly_billing_different_templates(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
sample_email_template, sample_email_template,
@@ -104,11 +95,8 @@ def test_create_nightly_billing_different_templates(
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by='mmg', sent_by='mmg',
@@ -116,11 +104,8 @@ def test_create_nightly_billing_different_templates(
rate_multiplier=1.0, rate_multiplier=1.0,
billable_units=1, billable_units=1,
) )
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_email_template, template=sample_email_template,
status='delivered', status='delivered',
sent_by='ses', sent_by='ses',
@@ -148,8 +133,6 @@ def test_create_nightly_billing_different_templates(
def test_create_nightly_billing_different_sent_by( def test_create_nightly_billing_different_sent_by(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
sample_email_template, sample_email_template,
@@ -159,11 +142,8 @@ def test_create_nightly_billing_different_sent_by(
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
# These are sms notifications # These are sms notifications
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by='mmg', sent_by='mmg',
@@ -171,11 +151,8 @@ def test_create_nightly_billing_different_sent_by(
rate_multiplier=1.0, rate_multiplier=1.0,
billable_units=1, billable_units=1,
) )
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by='firetext', sent_by='firetext',
@@ -201,8 +178,6 @@ def test_create_nightly_billing_different_sent_by(
def test_create_nightly_billing_letter( def test_create_nightly_billing_letter(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_letter_template, sample_letter_template,
mocker): mocker):
@@ -210,11 +185,8 @@ def test_create_nightly_billing_letter(
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_letter_template, template=sample_letter_template,
status='delivered', status='delivered',
sent_by='dvla', sent_by='dvla',
@@ -239,8 +211,6 @@ def test_create_nightly_billing_letter(
def test_create_nightly_billing_null_sent_by_sms( def test_create_nightly_billing_null_sent_by_sms(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
mocker): mocker):
@@ -248,11 +218,8 @@ def test_create_nightly_billing_null_sent_by_sms(
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=yesterday, created_at=yesterday,
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,
@@ -280,8 +247,6 @@ def test_create_nightly_billing_null_sent_by_sms(
@freeze_time('2018-01-15T03:30:00') @freeze_time('2018-01-15T03:30:00')
def test_create_nightly_billing_consolidate_from_3_days_delta( def test_create_nightly_billing_consolidate_from_3_days_delta(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
mocker): mocker):
@@ -290,11 +255,8 @@ def test_create_nightly_billing_consolidate_from_3_days_delta(
# create records from 11th to 15th # create records from 11th to 15th
for i in range(0, 11): for i in range(0, 11):
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=datetime.now() - timedelta(days=i), created_at=datetime.now() - timedelta(days=i),
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,
@@ -331,7 +293,7 @@ def test_get_rate_for_letter_latest(notify_db_session):
assert rate == Decimal(0.33) assert rate == Decimal(0.33)
def test_get_rate_for_sms_and_email(notify_db, notify_db_session): def test_get_rate_for_sms_and_email(notify_db_session):
sms_rate = Rate(valid_from=datetime(2017, 12, 1), sms_rate = Rate(valid_from=datetime(2017, 12, 1),
rate=Decimal(0.15), rate=Decimal(0.15),
notification_type=SMS_TYPE) notification_type=SMS_TYPE)
@@ -356,19 +318,14 @@ def test_get_rate_for_sms_and_email(notify_db, notify_db_session):
@freeze_time('2018-03-27T03:30:00') @freeze_time('2018-03-27T03:30:00')
# summer time starts on 2018-03-25 # summer time starts on 2018-03-25
def test_create_nightly_billing_use_BST( def test_create_nightly_billing_use_BST(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
mocker): mocker):
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=datetime(2018, 3, 25, 12, 0), created_at=datetime(2018, 3, 25, 12, 0),
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,
@@ -377,11 +334,8 @@ def test_create_nightly_billing_use_BST(
billable_units=1, billable_units=1,
) )
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=datetime(2018, 3, 25, 23, 5), created_at=datetime(2018, 3, 25, 23, 5),
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,
@@ -405,19 +359,14 @@ def test_create_nightly_billing_use_BST(
@freeze_time('2018-01-15T03:30:00') @freeze_time('2018-01-15T03:30:00')
def test_create_nightly_billing_update_when_record_exists( def test_create_nightly_billing_update_when_record_exists(
notify_db,
notify_db_session,
sample_service, sample_service,
sample_template, sample_template,
mocker): mocker):
mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate) mocker.patch('app.dao.fact_billing_dao.get_rate', side_effect=mocker_get_rate)
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=datetime.now() - timedelta(days=1), created_at=datetime.now() - timedelta(days=1),
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,
@@ -437,11 +386,8 @@ def test_create_nightly_billing_update_when_record_exists(
assert records[0].billable_units == 1 assert records[0].billable_units == 1
assert not records[0].updated_at assert not records[0].updated_at
sample_notification( create_notification(
notify_db,
notify_db_session,
created_at=datetime.now() - timedelta(days=1), created_at=datetime.now() - timedelta(days=1),
service=sample_service,
template=sample_template, template=sample_template,
status='delivered', status='delivered',
sent_by=None, sent_by=None,

37
tests/app/performance_platform/test_total_sent_notifications.py
View File

@@ -1,5 +1,4 @@
from datetime import datetime from datetime import datetime
from functools import partial
from freezegun import freeze_time from freezegun import freeze_time
@@ -9,9 +8,7 @@ from app.performance_platform.total_sent_notifications import (
get_total_sent_notifications_for_day get_total_sent_notifications_for_day
) )
from tests.app.conftest import ( from tests.app.db import create_template, create_notification
sample_notification_history as create_notification_history
)
def test_send_total_notifications_sent_for_day_stats_stats_creates_correct_call(mocker, client): def test_send_total_notifications_sent_for_day_stats_stats_creates_correct_call(mocker, client):
@@ -37,32 +34,24 @@ def test_send_total_notifications_sent_for_day_stats_stats_creates_correct_call(
@freeze_time("2016-01-11 12:30:00") @freeze_time("2016-01-11 12:30:00")
def test_get_total_sent_notifications_yesterday_returns_expected_totals_dict( def test_get_total_sent_notifications_yesterday_returns_expected_totals_dict(sample_service):
notify_db, sms = create_template(sample_service, template_type='sms')
notify_db_session, email = create_template(sample_service, template_type='email')
sample_template letter = create_template(sample_service, template_type='letter')
):
notification_history = partial(
create_notification_history,
notify_db,
notify_db_session,
sample_template,
status='delivered'
)
notification_history(notification_type='email') create_notification(email, status='delivered')
notification_history(notification_type='sms') create_notification(sms, status='delivered')
# Create some notifications for the day before # Create some notifications for the day before
yesterday = datetime(2016, 1, 10, 15, 30, 0, 0) yesterday = datetime(2016, 1, 10, 15, 30, 0, 0)
ereyesterday = datetime(2016, 1, 9, 15, 30, 0, 0) ereyesterday = datetime(2016, 1, 9, 15, 30, 0, 0)
with freeze_time(yesterday): with freeze_time(yesterday):
notification_history(notification_type='letter') create_notification(letter, status='delivered')
notification_history(notification_type='sms') create_notification(sms, status='delivered')
notification_history(notification_type='sms') create_notification(sms, status='delivered')
notification_history(notification_type='email') create_notification(email, status='delivered')
notification_history(notification_type='email') create_notification(email, status='delivered')
notification_history(notification_type='email') create_notification(email, status='delivered')
total_count_dict = get_total_sent_notifications_for_day(yesterday) total_count_dict = get_total_sent_notifications_for_day(yesterday)

48
tests/app/test_model.py
View File

@@ -21,16 +21,14 @@ from app.models import (
NOTIFICATION_TECHNICAL_FAILURE, NOTIFICATION_TECHNICAL_FAILURE,
PRECOMPILED_TEMPLATE_NAME PRECOMPILED_TEMPLATE_NAME
) )
from tests.app.conftest import (
sample_template as create_sample_template,
sample_notification_with_job as create_sample_notification_with_job
)
from tests.app.db import ( from tests.app.db import (
create_notification, create_notification,
create_service, create_service,
create_inbound_number, create_inbound_number,
create_reply_to_email, create_reply_to_email,
create_letter_contact create_letter_contact,
create_template
) )
@@ -99,26 +97,17 @@ def test_status_conversion(initial_statuses, expected_statuses):
('sms', '+447700900855'), ('sms', '+447700900855'),
('email', 'foo@bar.com'), ('email', 'foo@bar.com'),
]) ])
def test_notification_for_csv_returns_correct_type(notify_db, notify_db_session, template_type, recipient): def test_notification_for_csv_returns_correct_type(sample_service, template_type, recipient):
template = create_sample_template(notify_db, notify_db_session, template_type=template_type) template = create_template(sample_service, template_type=template_type)
notification = create_sample_notification_with_job( notification = create_notification(template, to_field=recipient)
notify_db,
notify_db_session,
template=template,
to_field=recipient
)
serialized = notification.serialize_for_csv() serialized = notification.serialize_for_csv()
assert serialized['template_type'] == template_type assert serialized['template_type'] == template_type
@freeze_time("2016-01-01 11:09:00.000000") @freeze_time("2016-01-01 11:09:00.000000")
def test_notification_for_csv_returns_correct_job_row_number(notify_db, notify_db_session): def test_notification_for_csv_returns_correct_job_row_number(sample_job):
notification = create_sample_notification_with_job( notification = create_notification(sample_job.template, sample_job, job_row_number=0)
notify_db,
notify_db_session,
job_row_number=0
)
serialized = notification.serialize_for_csv() serialized = notification.serialize_for_csv()
assert serialized['row_number'] == 1 assert serialized['row_number'] == 1
@@ -139,32 +128,21 @@ def test_notification_for_csv_returns_correct_job_row_number(notify_db, notify_d
('letter', 'delivered', 'Received') ('letter', 'delivered', 'Received')
]) ])
def test_notification_for_csv_returns_formatted_status( def test_notification_for_csv_returns_formatted_status(
notify_db, sample_service,
notify_db_session,
template_type, template_type,
status, status,
expected_status expected_status
): ):
template = create_sample_template(notify_db, notify_db_session, template_type=template_type) template = create_template(sample_service, template_type=template_type)
notification = create_sample_notification_with_job( notification = create_notification(template, status=status)
notify_db,
notify_db_session,
status=status,
template=template
)
serialized = notification.serialize_for_csv() serialized = notification.serialize_for_csv()
assert serialized['status'] == expected_status assert serialized['status'] == expected_status
@freeze_time("2017-03-26 23:01:53.321312") @freeze_time("2017-03-26 23:01:53.321312")
def test_notification_for_csv_returns_bst_correctly(notify_db, notify_db_session): def test_notification_for_csv_returns_bst_correctly(sample_template):
notification = create_sample_notification_with_job( notification = create_notification(sample_template)
notify_db,
notify_db_session,
job_row_number=100,
status='permanent-failure'
)
serialized = notification.serialize_for_csv() serialized = notification.serialize_for_csv()
assert serialized['created_at'] == 'Monday 27 March 2017 at 00:01' assert serialized['created_at'] == 'Monday 27 March 2017 at 00:01'