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

Update montly billing with letters

Browse Source
This commit is contained in:
Rebecca Law
2017-12-14 17:17:05 +00:00
parent a836ed037e
commit 012d80003a
7 changed files with 51 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/app/dao/test_monthly_billing.py
View File

@@ -164,7 +164,7 @@ def test_add_monthly_billing_for_single_month_populates_correctly(
monthly_billing = MonthlyBilling.query.order_by(MonthlyBilling.notification_type).all()
assert len(monthly_billing) == 2
assert len(monthly_billing) == 3
_assert_monthly_billing(
monthly_billing[0], sample_template.service.id, 'email', JAN_2017_MONTH_START, JAN_2017_MONTH_END
)
@@ -181,6 +181,11 @@ def test_add_monthly_billing_for_single_month_populates_correctly(
"total_cost": 1 * 2 * 0.0158
})
_assert_monthly_billing(
monthly_billing[2], sample_template.service.id, 'letter', JAN_2017_MONTH_START, JAN_2017_MONTH_END
)
assert monthly_billing[2].monthly_totals == []
def test_add_monthly_billing_for_multiple_months_populate_correctly(
sample_template, sample_email_template
@@ -203,7 +208,7 @@ def test_add_monthly_billing_for_multiple_months_populate_correctly(
MonthlyBilling.start_date
).all()
assert len(monthly_billing) == 4
assert len(monthly_billing) == 6
_assert_monthly_billing(
monthly_billing[0], sample_template.service.id, 'email', FEB_2016_MONTH_START, FEB_2016_MONTH_END
)
@@ -236,6 +241,16 @@ def test_add_monthly_billing_for_multiple_months_populate_correctly(
"total_cost": 0.72
})
_assert_monthly_billing(
monthly_billing[4], sample_template.service.id, 'letter', FEB_2016_MONTH_START, FEB_2016_MONTH_END
)
assert monthly_billing[4].monthly_totals == []
_assert_monthly_billing(
monthly_billing[5], sample_template.service.id, 'letter', MAR_2016_MONTH_START, MAR_2016_MONTH_END
)
assert monthly_billing[5].monthly_totals == []
def test_add_monthly_billing_with_multiple_rates_populate_correctly(
sample_template
@@ -252,7 +267,7 @@ def test_add_monthly_billing_with_multiple_rates_populate_correctly(
monthly_billing = MonthlyBilling.query.order_by(MonthlyBilling.notification_type).all()
assert len(monthly_billing) == 2
assert len(monthly_billing) == 3
_assert_monthly_billing(
monthly_billing[0], sample_template.service.id, 'email', JAN_2017_MONTH_START, JAN_2017_MONTH_END
)
@@ -276,6 +291,11 @@ def test_add_monthly_billing_with_multiple_rates_populate_correctly(
"total_cost": 0.246
})
_assert_monthly_billing(
monthly_billing[2], sample_template.service.id, 'letter', JAN_2017_MONTH_START, JAN_2017_MONTH_END
)
assert monthly_billing[0].monthly_totals == []
def test_update_monthly_billing_overwrites_old_totals(sample_template):
create_rate(APR_2016_MONTH_START, 0.123, SMS_TYPE)

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

@@ -1,5 +1,4 @@
import uuid
from _decimal import Decimal
from datetime import datetime, timedelta
from freezegun import freeze_time
@@ -218,7 +217,7 @@ def test_get_monthly_billing_data_where_start_date_before_rate_returns_empty(
def test_billing_letter_data_per_month_query(
notify_db_session
):
rate = create_letter_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),
@@ -233,4 +232,5 @@ def test_billing_letter_data_per_month_query(
end_date=datetime(2017, 2, 28))
assert len(results) == 1
assert results[0].rate == Decimal('0.31')
print(results[0].rate)
assert results[0].rate == 0.31