darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-23 17:01:35 -05:00
Code Issues Packages Projects Releases Wiki Activity

Provider Statistics added.

Browse Source 
Rates command added with a test.

Updated to include added migration.
This commit is contained in:
Nicholas Staples
2016-04-21 11:37:38 +01:00
parent a38e4f4965
commit 3b1423a2ea
15 changed files with 393 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

210
tests/app/dao/test_notification_dao.py
View File

@@ -46,7 +46,7 @@ def test_should_by_able_to_update_reference_by_id(sample_notification):
assert Notification.query.get(sample_notification.id).reference == 'reference'
def test_should_by_able_to_update_status_by_reference(sample_email_template):
def test_should_by_able_to_update_status_by_reference(sample_email_template, ses_provider_name):
data = {
'to': '+44709123456',
'service': sample_email_template.service,
@@ -57,7 +57,10 @@ def test_should_by_able_to_update_status_by_reference(sample_email_template):
}
notification = Notification(**data)
dao_create_notification(notification, sample_email_template.template_type)
dao_create_notification(
notification,
sample_email_template.template_type,
ses_provider_name)
assert Notification.query.get(notification.id).status == "sending"
update_notification_reference_by_id(notification.id, 'reference')
@@ -106,7 +109,7 @@ def test_should_be_able_to_record_statistics_failure_for_sms(sample_notification
).one().sms_failed == 1
def test_should_be_able_to_record_statistics_failure_for_email(sample_email_template):
def test_should_be_able_to_record_statistics_failure_for_email(sample_email_template, ses_provider_name):
data = {
'to': '+44709123456',
'service': sample_email_template.service,
@@ -117,7 +120,7 @@ def test_should_be_able_to_record_statistics_failure_for_email(sample_email_temp
}
notification = Notification(**data)
dao_create_notification(notification, sample_email_template.template_type)
dao_create_notification(notification, sample_email_template.template_type, ses_provider_name)
update_notification_reference_by_id(notification.id, 'reference')
count = update_notification_status_by_reference('reference', 'failed', 'failure')
@@ -142,18 +145,19 @@ def test_should_return_zero_count_if_no_notification_with_reference():
assert update_notification_status_by_reference('something', 'delivered', 'delivered') == 0
def test_should_be_able_to_get_statistics_for_a_service(sample_template):
def test_should_be_able_to_get_statistics_for_a_service(sample_template, mmg_provider_name):
data = {
'to': '+44709123456',
'service': sample_template.service,
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
stats = dao_get_notification_statistics_for_service(sample_template.service.id)
assert len(stats) == 1
@@ -168,7 +172,7 @@ def test_should_be_able_to_get_statistics_for_a_service(sample_template):
assert stats[0].emails_failed == 0
def test_should_be_able_to_get_statistics_for_a_service_for_a_day(sample_template):
def test_should_be_able_to_get_statistics_for_a_service_for_a_day(sample_template, mmg_provider_name):
now = datetime.utcnow()
data = {
'to': '+44709123456',
@@ -176,11 +180,12 @@ def test_should_be_able_to_get_statistics_for_a_service_for_a_day(sample_templat
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': now
'created_at': now,
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
stat = dao_get_notification_statistics_for_service_and_day(
sample_template.service.id, now.date()
)
@@ -194,7 +199,7 @@ def test_should_be_able_to_get_statistics_for_a_service_for_a_day(sample_templat
assert stat.service_id == notification.service_id
def test_should_return_none_if_no_statistics_for_a_service_for_a_day(sample_template):
def test_should_return_none_if_no_statistics_for_a_service_for_a_day(sample_template, mmg_provider_name):
now = datetime.utcnow()
data = {
'to': '+44709123456',
@@ -202,32 +207,34 @@ def test_should_return_none_if_no_statistics_for_a_service_for_a_day(sample_temp
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': now
'created_at': now,
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert not dao_get_notification_statistics_for_service_and_day(
sample_template.service.id, (datetime.utcnow() - timedelta(days=1)).date()
)
def test_should_be_able_to_get_all_statistics_for_a_service(sample_template):
def test_should_be_able_to_get_all_statistics_for_a_service(sample_template, mmg_provider_name):
data = {
'to': '+44709123456',
'service': sample_template.service,
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
notification_2 = Notification(**data)
notification_3 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_3, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_3, sample_template.template_type, mmg_provider_name)
stats = dao_get_notification_statistics_for_service(sample_template.service.id)
assert len(stats) == 1
@@ -235,13 +242,14 @@ def test_should_be_able_to_get_all_statistics_for_a_service(sample_template):
assert stats[0].sms_requested == 3
def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days(sample_template):
def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days(sample_template, mmg_provider_name):
data = {
'to': '+44709123456',
'service': sample_template.service,
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id
'template_id': sample_template.id,
'content_char_count': 160
}
today = datetime.utcnow()
@@ -259,9 +267,9 @@ def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days(sam
'created_at': two_days_ago
})
notification_3 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_3, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_3, sample_template.template_type, mmg_provider_name)
stats = dao_get_notification_statistics_for_service(sample_template.service.id)
assert len(stats) == 3
@@ -280,13 +288,15 @@ def test_should_be_empty_list_if_no_statistics_for_a_service(sample_service):
assert len(dao_get_notification_statistics_for_service(sample_service.id)) == 0
def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days_previous(sample_template):
def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days_previous(sample_template,
mmg_provider_name):
data = {
'to': '+44709123456',
'service': sample_template.service,
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id
'template_id': sample_template.id,
'content_char_count': 160
}
today = datetime.utcnow()
@@ -304,9 +314,9 @@ def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days_pre
'created_at': eight_days_ago
})
notification_3 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_3, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_3, sample_template.template_type, mmg_provider_name)
stats = dao_get_notification_statistics_for_service_and_previous_days(
sample_template.service.id, 7
@@ -320,7 +330,7 @@ def test_should_be_able_to_get_all_statistics_for_a_service_for_several_days_pre
assert stats[1].day == seven_days_ago.date()
def test_save_notification_creates_sms_and_template_stats(sample_template, sample_job):
def test_save_notification_creates_sms_and_template_stats(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
assert NotificationStatistics.query.count() == 0
assert TemplateStatistics.query.count() == 0
@@ -332,11 +342,12 @@ def test_save_notification_creates_sms_and_template_stats(sample_template, sampl
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -364,7 +375,7 @@ def test_save_notification_creates_sms_and_template_stats(sample_template, sampl
assert template_stats.usage_count == 1
def test_save_notification_and_create_email_and_template_stats(sample_email_template, sample_job):
def test_save_notification_and_create_email_and_template_stats(sample_email_template, sample_job, ses_provider_name):
assert Notification.query.count() == 0
assert NotificationStatistics.query.count() == 0
@@ -377,11 +388,12 @@ def test_save_notification_and_create_email_and_template_stats(sample_email_temp
'service_id': sample_email_template.service.id,
'template': sample_email_template,
'template_id': sample_email_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_email_template.template_type)
dao_create_notification(notification, sample_email_template.template_type, ses_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -410,7 +422,7 @@ def test_save_notification_and_create_email_and_template_stats(sample_email_temp
@freeze_time("2016-01-01 00:00:00.000000")
def test_save_notification_handles_midnight_properly(sample_template, sample_job):
def test_save_notification_handles_midnight_properly(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -419,11 +431,12 @@ def test_save_notification_handles_midnight_properly(sample_template, sample_job
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
@@ -435,7 +448,7 @@ def test_save_notification_handles_midnight_properly(sample_template, sample_job
@freeze_time("2016-01-01 23:59:59.999999")
def test_save_notification_handles_just_before_midnight_properly(sample_template, sample_job):
def test_save_notification_handles_just_before_midnight_properly(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -444,11 +457,12 @@ def test_save_notification_handles_just_before_midnight_properly(sample_template
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
@@ -459,7 +473,7 @@ def test_save_notification_handles_just_before_midnight_properly(sample_template
assert stats.day == date(2016, 1, 1)
def test_save_notification_and_increment_email_stats(sample_email_template, sample_job):
def test_save_notification_and_increment_email_stats(sample_email_template, sample_job, ses_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -468,12 +482,13 @@ def test_save_notification_and_increment_email_stats(sample_email_template, samp
'service_id': sample_email_template.service.id,
'template': sample_email_template,
'template_id': sample_email_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
notification_2 = Notification(**data)
dao_create_notification(notification_1, sample_email_template.template_type)
dao_create_notification(notification_1, sample_email_template.template_type, ses_provider_name)
assert Notification.query.count() == 1
@@ -484,7 +499,7 @@ def test_save_notification_and_increment_email_stats(sample_email_template, samp
assert stats1.emails_requested == 1
assert stats1.sms_requested == 0
dao_create_notification(notification_2, sample_email_template.template_type)
dao_create_notification(notification_2, sample_email_template.template_type, ses_provider_name)
assert Notification.query.count() == 2
@@ -496,7 +511,7 @@ def test_save_notification_and_increment_email_stats(sample_email_template, samp
assert stats2.sms_requested == 0
def test_save_notification_and_increment_sms_stats(sample_template, sample_job):
def test_save_notification_and_increment_sms_stats(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -505,12 +520,13 @@ def test_save_notification_and_increment_sms_stats(sample_template, sample_job):
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
notification_2 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
@@ -521,7 +537,7 @@ def test_save_notification_and_increment_sms_stats(sample_template, sample_job):
assert stats1.emails_requested == 0
assert stats1.sms_requested == 1
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 2
@@ -533,7 +549,7 @@ def test_save_notification_and_increment_sms_stats(sample_template, sample_job):
assert stats2.sms_requested == 2
def test_not_save_notification_and_not_create_stats_on_commit_error(sample_template, sample_job):
def test_not_save_notification_and_not_create_stats_on_commit_error(sample_template, sample_job, mmg_provider_name):
random_id = str(uuid.uuid4())
assert Notification.query.count() == 0
@@ -544,12 +560,13 @@ def test_not_save_notification_and_not_create_stats_on_commit_error(sample_templ
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
with pytest.raises(SQLAlchemyError):
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 0
assert Job.query.get(sample_job.id).notifications_sent == 0
@@ -557,7 +574,7 @@ def test_not_save_notification_and_not_create_stats_on_commit_error(sample_templ
assert TemplateStatistics.query.count() == 0
def test_save_notification_and_increment_job(sample_template, sample_job):
def test_save_notification_and_increment_job(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -566,11 +583,12 @@ def test_save_notification_and_increment_job(sample_template, sample_job):
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -584,12 +602,12 @@ def test_save_notification_and_increment_job(sample_template, sample_job):
assert Job.query.get(sample_job.id).notifications_sent == 1
notification_2 = Notification(**data)
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 2
assert Job.query.get(sample_job.id).notifications_sent == 2
def test_should_not_increment_job_if_notification_fails_to_persist(sample_template, sample_job):
def test_should_not_increment_job_if_notification_fails_to_persist(sample_template, sample_job, mmg_provider_name):
random_id = str(uuid.uuid4())
assert Notification.query.count() == 0
@@ -601,11 +619,12 @@ def test_should_not_increment_job_if_notification_fails_to_persist(sample_templa
'service': sample_template.service,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
assert Job.query.get(sample_job.id).notifications_sent == 1
@@ -613,14 +632,14 @@ def test_should_not_increment_job_if_notification_fails_to_persist(sample_templa
notification_2 = Notification(**data)
with pytest.raises(SQLAlchemyError):
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
assert Job.query.get(sample_job.id).notifications_sent == 1
assert Job.query.get(sample_job.id).updated_at == job_last_updated_at
def test_save_notification_and_increment_correct_job(notify_db, notify_db_session, sample_template):
def test_save_notification_and_increment_correct_job(notify_db, notify_db_session, sample_template, mmg_provider_name):
job_1 = sample_job(notify_db, notify_db_session, sample_template.service)
job_2 = sample_job(notify_db, notify_db_session, sample_template.service)
@@ -632,11 +651,12 @@ def test_save_notification_and_increment_correct_job(notify_db, notify_db_sessio
'service': sample_template.service,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -651,7 +671,7 @@ def test_save_notification_and_increment_correct_job(notify_db, notify_db_sessio
assert Job.query.get(job_2.id).notifications_sent == 0
def test_save_notification_with_no_job(sample_template):
def test_save_notification_with_no_job(sample_template, mmg_provider_name):
assert Notification.query.count() == 0
data = {
'to': '+44709123456',
@@ -659,11 +679,12 @@ def test_save_notification_with_no_job(sample_template):
'service': sample_template.service,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -682,7 +703,7 @@ def test_get_notification(sample_notification):
assert sample_notification == notifcation_from_db
def test_save_notification_no_job_id(sample_template):
def test_save_notification_no_job_id(sample_template, mmg_provider_name):
assert Notification.query.count() == 0
to = '+44709123456'
data = {
@@ -691,11 +712,12 @@ def test_save_notification_no_job_id(sample_template):
'service': sample_template.service,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert Notification.query.count() == 1
notification_from_db = Notification.query.all()[0]
@@ -786,7 +808,7 @@ def test_should_not_delete_failed_notifications_before_seven_days(notify_db, not
@freeze_time("2016-03-30")
def test_save_new_notification_creates_template_stats(sample_template, sample_job):
def test_save_new_notification_creates_template_stats(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
assert TemplateStatistics.query.count() == 0
data = {
@@ -796,11 +818,12 @@ def test_save_new_notification_creates_template_stats(sample_template, sample_jo
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert TemplateStatistics.query.count() == 1
template_stats = TemplateStatistics.query.filter(TemplateStatistics.service_id == sample_template.service.id,
@@ -812,7 +835,7 @@ def test_save_new_notification_creates_template_stats(sample_template, sample_jo
@freeze_time("2016-03-30")
def test_save_new_notification_creates_template_stats_per_day(sample_template, sample_job):
def test_save_new_notification_creates_template_stats_per_day(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
assert TemplateStatistics.query.count() == 0
data = {
@@ -822,11 +845,12 @@ def test_save_new_notification_creates_template_stats_per_day(sample_template, s
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
assert TemplateStatistics.query.count() == 1
template_stats = TemplateStatistics.query.filter(TemplateStatistics.service_id == sample_template.service.id,
@@ -840,7 +864,7 @@ def test_save_new_notification_creates_template_stats_per_day(sample_template, s
with freeze_time('2016-03-31'):
assert TemplateStatistics.query.count() == 1
new_notification = Notification(**data)
dao_create_notification(new_notification, sample_template.template_type)
dao_create_notification(new_notification, sample_template.template_type, mmg_provider_name)
assert TemplateStatistics.query.count() == 2
first_stats = TemplateStatistics.query.filter(TemplateStatistics.day == datetime(2016, 3, 30)).first()
@@ -856,7 +880,7 @@ def test_save_new_notification_creates_template_stats_per_day(sample_template, s
assert second_stats.usage_count == 1
def test_save_another_notification_increments_template_stats(sample_template, sample_job):
def test_save_another_notification_increments_template_stats(sample_template, sample_job, mmg_provider_name):
assert Notification.query.count() == 0
assert TemplateStatistics.query.count() == 0
data = {
@@ -866,12 +890,13 @@ def test_save_another_notification_increments_template_stats(sample_template, sa
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
notification_2 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
assert TemplateStatistics.query.count() == 1
template_stats = TemplateStatistics.query.filter(TemplateStatistics.service_id == sample_template.service.id,
@@ -880,7 +905,7 @@ def test_save_another_notification_increments_template_stats(sample_template, sa
assert template_stats.template_id == sample_template.id
assert template_stats.usage_count == 1
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
assert TemplateStatistics.query.count() == 1
template_stats = TemplateStatistics.query.filter(TemplateStatistics.service_id == sample_template.service.id,
@@ -889,7 +914,8 @@ def test_save_another_notification_increments_template_stats(sample_template, sa
def test_successful_notification_inserts_followed_by_failure_does_not_increment_template_stats(sample_template,
sample_job):
sample_job,
mmg_provider_name):
assert Notification.query.count() == 0
assert NotificationStatistics.query.count() == 0
assert TemplateStatistics.query.count() == 0
@@ -901,15 +927,16 @@ def test_successful_notification_inserts_followed_by_failure_does_not_increment_
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification_1 = Notification(**data)
notification_2 = Notification(**data)
notification_3 = Notification(**data)
dao_create_notification(notification_1, sample_template.template_type)
dao_create_notification(notification_2, sample_template.template_type)
dao_create_notification(notification_3, sample_template.template_type)
dao_create_notification(notification_1, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_2, sample_template.template_type, mmg_provider_name)
dao_create_notification(notification_3, sample_template.template_type, mmg_provider_name)
assert NotificationStatistics.query.count() == 1
notication_stats = NotificationStatistics.query.filter(
@@ -928,7 +955,7 @@ def test_successful_notification_inserts_followed_by_failure_does_not_increment_
try:
# Mess up db in really bad way
db.session.execute('DROP TABLE TEMPLATE_STATISTICS')
dao_create_notification(failing_notification, sample_template.template_type)
dao_create_notification(failing_notification, sample_template.template_type, mmg_provider_name)
except Exception as e:
# There should be no additional notification stats or counts
assert NotificationStatistics.query.count() == 1
@@ -939,7 +966,9 @@ def test_successful_notification_inserts_followed_by_failure_does_not_increment_
@freeze_time("2016-03-30")
def test_get_template_stats_for_service_returns_stats_in_reverse_date_order(sample_template, sample_job):
def test_get_template_stats_for_service_returns_stats_in_reverse_date_order(sample_template,
sample_job,
mmg_provider_name):
template_stats = dao_get_template_statistics_for_service(sample_template.service.id)
assert len(template_stats) == 0
@@ -950,21 +979,22 @@ def test_get_template_stats_for_service_returns_stats_in_reverse_date_order(samp
'service_id': sample_template.service.id,
'template': sample_template,
'template_id': sample_template.id,
'created_at': datetime.utcnow()
'created_at': datetime.utcnow(),
'content_char_count': 160
}
notification = Notification(**data)
dao_create_notification(notification, sample_template.template_type)
dao_create_notification(notification, sample_template.template_type, mmg_provider_name)
# move on one day
with freeze_time('2016-03-31'):
new_notification = Notification(**data)
dao_create_notification(new_notification, sample_template.template_type)
dao_create_notification(new_notification, sample_template.template_type, mmg_provider_name)
# move on one more day
with freeze_time('2016-04-01'):
new_notification = Notification(**data)
dao_create_notification(new_notification, sample_template.template_type)
dao_create_notification(new_notification, sample_template.template_type, mmg_provider_name)
template_stats = dao_get_template_statistics_for_service(sample_template.service_id)
assert len(template_stats) == 3