darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-24 16:23:44 -04:00
Code Issues Packages Projects Releases Wiki Activity

Test filtering letter_rates by post_class

Browse Source
This commit is contained in:
Pea Tyczynska
2018-09-17 14:12:06 +01:00
parent b3191dbc2a
commit 39ebfb6826
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,7 +23,9 @@ def test_reporting_should_have_decorated_tasks_functions():
assert create_nightly_billing.__wrapped__.__name__ == 'create_nightly_billing' assert create_nightly_billing.__wrapped__.__name__ == 'create_nightly_billing'
def mocker_get_rate(non_letter_rates, letter_rates, notification_type, date, crown=None, rate_multiplier=None, post_class="second"): def mocker_get_rate(
non_letter_rates, letter_rates, notification_type, date, crown=None, rate_multiplier=None, post_class="second"
):
if notification_type == LETTER_TYPE: if notification_type == LETTER_TYPE:
return Decimal(2.1) return Decimal(2.1)
elif notification_type == SMS_TYPE: elif notification_type == SMS_TYPE:
@@ -329,6 +331,21 @@ def test_get_rate_for_letter_latest(notify_db_session):
assert rate == Decimal(0.33) assert rate == Decimal(0.33)
@pytest.mark.parametrize(
"letter_post_class,expected_rate", [("first", 0.42), ("second", 0.33)]
)
def test_get_rate_filters_letters_byl_post_class(notify_db_session, letter_post_class, expected_rate):
non_letter_rates = [(r.notification_type, r.valid_from, r.rate) for r in
Rate.query.order_by(desc(Rate.valid_from)).all()]
letter_rates = [
(datetime(2017, 12, 1), True, 1, Decimal(0.42), 'first'),
(datetime(2017, 12, 1), True, 1, Decimal(0.33), 'second')
]
rate = get_rate(non_letter_rates, letter_rates, LETTER_TYPE, datetime(2018, 1, 1), True, 1, letter_post_class)
assert rate == Decimal(expected_rate)
def test_get_rate_for_sms_and_email(notify_db, notify_db_session): def test_get_rate_for_sms_and_email(notify_db, 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),