mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-30 06:21:50 -05:00
incorporate reviewers comments
This commit is contained in:
@@ -17,6 +17,7 @@ from tests import create_authorization_header
|
||||
from app.dao.annual_billing_dao import (dao_get_free_sms_fragment_limit_for_year,
|
||||
dao_create_or_update_annual_billing_for_year)
|
||||
from app.models import AnnualBilling
|
||||
import uuid
|
||||
|
||||
APR_2016_MONTH_START = datetime(2016, 3, 31, 23, 00, 00)
|
||||
APR_2016_MONTH_END = datetime(2016, 4, 30, 22, 59, 59, 99999)
|
||||
@@ -359,3 +360,12 @@ def test_get_free_sms_fragment_limit_no_year_data_return_404(client, sample_serv
|
||||
json_resp = json.loads(response_get.get_data(as_text=True))
|
||||
|
||||
assert response_get.status_code == 404
|
||||
|
||||
|
||||
def test_get_free_sms_fragment_limit_unknown_service_id_return_404(client):
|
||||
|
||||
response_get = client.get(
|
||||
'service/{}/billing/free-sms-fragment-limit'.format(uuid.uuid4()),
|
||||
headers=[('Content-Type', 'application/json'), create_authorization_header()])
|
||||
json_resp = json.loads(response_get.get_data(as_text=True))
|
||||
assert response_get.status_code == 404
|
||||
|
||||
Reference in New Issue
Block a user