mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-02 17:31:14 -05:00
Added free_sms_fragment_limit model, schema, dao and Rest
This commit is contained in:
46
tests/app/dao/test_annual_billing_dao.py
Normal file
46
tests/app/dao/test_annual_billing_dao.py
Normal file
@@ -0,0 +1,46 @@
|
||||
# from datetime import datetime, timedelta
|
||||
# import uuid
|
||||
# import functools
|
||||
# import pytest
|
||||
|
||||
from app.models import AnnualBilling
|
||||
from app.dao.annual_billing_dao import (
|
||||
dao_create_new_annual_billing_for_year,
|
||||
dao_get_free_sms_fragment_limit_for_year,
|
||||
dao_update_new_free_sms_fragment_limit_for_year
|
||||
)
|
||||
|
||||
|
||||
def test_dao_create_get_free_sms_fragment_limit(notify_db_session, sample_service):
|
||||
year = 2016
|
||||
data = AnnualBilling(
|
||||
free_sms_fragment_limit=250000,
|
||||
financial_year_start=year,
|
||||
service_id=sample_service.id,
|
||||
)
|
||||
dao_create_new_annual_billing_for_year(data)
|
||||
|
||||
free_limit = dao_get_free_sms_fragment_limit_for_year(sample_service.id, year)
|
||||
|
||||
assert free_limit.free_sms_fragment_limit == 250000
|
||||
assert free_limit.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_new_annual_billing_for_year(data)
|
||||
data.free_sms_fragment_limit = new_limit
|
||||
dao_update_new_free_sms_fragment_limit_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
|
||||
Reference in New Issue
Block a user