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

Added query for billing usage of letter notifications.

Browse Source
This commit is contained in:
Rebecca Law
2017-12-11 16:55:23 +00:00
parent 70890e9dfc
commit a836ed037e
3 changed files with 85 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/app/dao/test_notification_usage_dao.py
View File

@@ -1,4 +1,5 @@
import uuid
from _decimal import Decimal
from datetime import datetime, timedelta
from freezegun import freeze_time
@@ -7,13 +8,14 @@ from app.dao.date_util import get_financial_year
from app.dao.notification_usage_dao import (
get_rates_for_daterange,
get_billing_data_for_month,
get_monthly_billing_data
get_monthly_billing_data,
billing_letter_data_per_month_query
)
from app.models import (
Rate,
SMS_TYPE,
)
from tests.app.db import create_notification, create_rate
from tests.app.db import create_notification, create_rate, create_letter_rate, create_template, create_service
def test_get_rates_for_daterange(notify_db, notify_db_session):
@@ -211,3 +213,24 @@ def test_get_monthly_billing_data_where_start_date_before_rate_returns_empty(
)
assert not results
def test_billing_letter_data_per_month_query(
notify_db_session
):
rate = create_letter_rate()
service = create_service()
template = create_template(service=service, template_type='letter')
create_notification(template=template, billable_units=1, created_at=datetime(2017, 2, 1, 13, 21),
status='delivered')
create_notification(template=template, billable_units=1, created_at=datetime(2017, 2, 1, 13, 21),
status='delivered')
create_notification(template=template, billable_units=1, created_at=datetime(2017, 2, 1, 13, 21),
status='delivered')
results = billing_letter_data_per_month_query(service_id=service.id,
start_date=datetime(2017, 2, 1),
end_date=datetime(2017, 2, 28))
assert len(results) == 1
assert results[0].rate == Decimal('0.31')