mirror of
https://github.com/GSA/notifications-api.git
synced 2026-09-10 10:03:38 -04:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from app.service.utils import get_current_financial_year_start_year
|
|
from app.models import AnnualBilling
|
|
from app.dao.annual_billing_dao import (
|
|
dao_create_or_update_annual_billing_for_year,
|
|
dao_get_free_sms_fragment_limit_for_year
|
|
)
|
|
|
|
|
|
def test_sample_service_has_free_sms_fragment_limit(notify_db_session, sample_service):
|
|
|
|
free_limit = dao_get_free_sms_fragment_limit_for_year(sample_service.id, get_current_financial_year_start_year())
|
|
|
|
assert free_limit.free_sms_fragment_limit == 250000
|
|
assert free_limit.financial_year_start == get_current_financial_year_start_year()
|
|
assert free_limit.service_id == sample_service.id
|
|
|
|
|
|
def test_dao_update_free_sms_fragment_limit(notify_db_session, sample_service):
|
|
year = 2016
|
|
old_limit = 1000
|
|
new_limit = 9999
|
|
|
|
data = AnnualBilling(
|
|
free_sms_fragment_limit=old_limit,
|
|
financial_year_start=year,
|
|
service_id=sample_service.id,
|
|
)
|
|
|
|
dao_create_or_update_annual_billing_for_year(data)
|
|
data.free_sms_fragment_limit = new_limit
|
|
dao_create_or_update_annual_billing_for_year(data)
|
|
new_free_limit = dao_get_free_sms_fragment_limit_for_year(sample_service.id, year)
|
|
|
|
assert new_free_limit.free_sms_fragment_limit == new_limit
|