mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 00:41:35 -05:00
a186fc95be
Since the response has changed I have created new endpoints so that the deployments for Admin are more managable. Removed print statements from some tests.
160 lines
9.9 KiB
Python
160 lines
9.9 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
from app.dao.notification_usage_dao import (get_rates_for_year, get_yearly_billing_data,
|
|
get_monthly_billing_data)
|
|
from app.models import Rate
|
|
from tests.app.db import create_notification
|
|
|
|
|
|
def test_get_rates_for_year(notify_db, notify_db_session):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.50)
|
|
set_up_rate(notify_db, datetime(2017, 6, 1), 1.75)
|
|
rates = get_rates_for_year(datetime(2016, 4, 1), datetime(2017, 3, 31), 'sms')
|
|
assert len(rates) == 1
|
|
assert datetime.strftime(rates[0].valid_from, '%Y-%m-%d %H:%M:%S') == "2016-04-01 00:00:00"
|
|
assert rates[0].rate == 1.50
|
|
rates = get_rates_for_year(datetime(2017, 4, 1), datetime(2018, 3, 31), 'sms')
|
|
assert len(rates) == 1
|
|
assert datetime.strftime(rates[0].valid_from, '%Y-%m-%d %H:%M:%S') == "2017-06-01 00:00:00"
|
|
assert rates[0].rate == 1.75
|
|
|
|
|
|
def test_get_yearly_billing_data(notify_db, notify_db_session, sample_template, sample_email_template):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.40)
|
|
set_up_rate(notify_db, datetime(2016, 6, 1), 1.58)
|
|
# previous year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 3, 31), sent_at=datetime(2016, 3, 31),
|
|
status='sending', billable_units=1)
|
|
# current year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 4, 2), sent_at=datetime(2016, 4, 2),
|
|
status='sending', billable_units=1)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 5, 18), sent_at=datetime(2016, 5, 18),
|
|
status='sending', billable_units=2)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 7, 22), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=3)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 9, 15), sent_at=datetime(2016, 9, 15),
|
|
status='sending', billable_units=4)
|
|
create_notification(template=sample_template, created_at=datetime(2017, 3, 31), sent_at=datetime(2017, 3, 31),
|
|
status='sending', billable_units=5)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 9, 15), sent_at=datetime(2016, 9, 15),
|
|
status='sending', billable_units=0)
|
|
create_notification(template=sample_email_template, created_at=datetime(2017, 3, 31), sent_at=datetime(2017, 3, 31),
|
|
status='sending', billable_units=0)
|
|
# next year
|
|
create_notification(template=sample_template, created_at=datetime(2017, 4, 1), sent_at=datetime(2017, 4, 1),
|
|
status='sending', billable_units=6)
|
|
results = get_yearly_billing_data(sample_template.service_id, 2016)
|
|
assert len(results) == 3
|
|
assert results[0] == (3, 'sms', None, False, None, 1.40)
|
|
assert results[1] == (12, 'sms', None, False, None, 1.58)
|
|
assert results[2] == (2, 'email', None, False, None, 0)
|
|
|
|
|
|
def test_get_yearly_billing_data_with_one_rate(notify_db, notify_db_session, sample_template):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.40)
|
|
# previous year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 3, 31), sent_at=datetime(2016, 3, 31),
|
|
status='sending', billable_units=1)
|
|
# current year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 4, 2), sent_at=datetime(2016, 4, 2),
|
|
status='sending', billable_units=1)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 5, 18), sent_at=datetime(2016, 5, 18),
|
|
status='sending', billable_units=2)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 7, 22), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=3)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 9, 15), sent_at=datetime(2016, 9, 15),
|
|
status='sending', billable_units=4)
|
|
create_notification(template=sample_template, created_at=datetime(2017, 3, 31, 22, 59, 59),
|
|
sent_at=datetime(2017, 3, 31), status='sending', billable_units=5)
|
|
# next year
|
|
create_notification(template=sample_template, created_at=datetime(2017, 3, 31, 23, 00, 00),
|
|
sent_at=datetime(2017, 3, 31), status='sending', billable_units=6)
|
|
create_notification(template=sample_template, created_at=datetime(2017, 4, 1), sent_at=datetime(2017, 4, 1),
|
|
status='sending', billable_units=7)
|
|
results = get_yearly_billing_data(sample_template.service_id, 2016)
|
|
assert len(results) == 2
|
|
assert results[0] == (15, 'sms', None, False, None, 1.4)
|
|
assert results[1] == (0, 'email', None, False, None, 0)
|
|
|
|
|
|
def test_get_yearly_billing_data_with_no_sms_notifications(notify_db, notify_db_session, sample_email_template):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.40)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 7, 31), sent_at=datetime(2016, 3, 31),
|
|
status='sending', billable_units=0)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 10, 2), sent_at=datetime(2016, 4, 2),
|
|
status='sending', billable_units=0)
|
|
|
|
results = get_yearly_billing_data(sample_email_template.service_id, 2016)
|
|
assert len(results) == 2
|
|
assert results[0] == (0, 'sms', None, False, None, 0)
|
|
assert results[1] == (2, 'email', None, False, None, 0)
|
|
|
|
|
|
def test_get_monthly_billing_data(notify_db, notify_db_session, sample_template, sample_email_template):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.40)
|
|
# previous year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 3, 31), sent_at=datetime(2016, 3, 31),
|
|
status='sending', billable_units=1)
|
|
# current year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 4, 2), sent_at=datetime(2016, 4, 2),
|
|
status='sending', billable_units=1)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 5, 18), sent_at=datetime(2016, 5, 18),
|
|
status='sending', billable_units=2)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 7, 22), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=3)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 7, 30), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=4)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 8, 22), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=0)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 8, 30), sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=0)
|
|
# next year
|
|
create_notification(template=sample_template, created_at=datetime(2017, 3, 31, 23, 00, 00),
|
|
sent_at=datetime(2017, 3, 31), status='sending', billable_units=6)
|
|
results = get_monthly_billing_data(sample_template.service_id, 2016)
|
|
assert len(results) == 4
|
|
assert results[0] == ('April', 1, 'sms', None, False, None, 1.4)
|
|
assert results[1] == ('May', 2, 'sms', None, False, None, 1.4)
|
|
assert results[2] == ('July', 7, 'sms', None, False, None, 1.4)
|
|
assert results[3] == ('August', 2, 'email', None, False, None, 0)
|
|
|
|
|
|
def test_get_monthly_billing_data_with_multiple_rates(notify_db, notify_db_session, sample_template,
|
|
sample_email_template):
|
|
set_up_rate(notify_db, datetime(2016, 4, 1), 1.40)
|
|
set_up_rate(notify_db, datetime(2016, 6, 5), 1.75)
|
|
# previous year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 3, 31), sent_at=datetime(2016, 3, 31),
|
|
status='sending', billable_units=1)
|
|
# current year
|
|
create_notification(template=sample_template, created_at=datetime(2016, 4, 2), sent_at=datetime(2016, 4, 2),
|
|
status='sending', billable_units=1)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 5, 18), sent_at=datetime(2016, 5, 18),
|
|
status='sending', billable_units=2)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 6, 1), sent_at=datetime(2016, 6, 1),
|
|
status='sending', billable_units=3)
|
|
create_notification(template=sample_template, created_at=datetime(2016, 6, 15), sent_at=datetime(2016, 6, 15),
|
|
status='sending', billable_units=4)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 8, 22),
|
|
sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=0)
|
|
create_notification(template=sample_email_template, created_at=datetime(2016, 8, 30),
|
|
sent_at=datetime(2016, 7, 22),
|
|
status='sending', billable_units=0)
|
|
# next year
|
|
create_notification(template=sample_template, created_at=datetime(2017, 3, 31, 23, 00, 00),
|
|
sent_at=datetime(2017, 3, 31), status='sending', billable_units=6)
|
|
results = get_monthly_billing_data(sample_template.service_id, 2016)
|
|
assert len(results) == 5
|
|
assert results[0] == ('April', 1, 'sms', None, False, None, 1.4)
|
|
assert results[1] == ('May', 2, 'sms', None, False, None, 1.4)
|
|
assert results[2] == ('June', 3, 'sms', None, False, None, 1.4)
|
|
assert results[3] == ('June', 4, 'sms', None, False, None, 1.75)
|
|
assert results[4] == ('August', 2, 'email', None, False, None, 0)
|
|
|
|
|
|
def set_up_rate(notify_db, start_date, value):
|
|
rate = Rate(id=uuid.uuid4(), valid_from=start_date, rate=value, notification_type='sms')
|
|
notify_db.session.add(rate)
|