darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 15:46:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Split generating authorization headers by type

Browse Source 
In response to [1].

[1]: https://github.com/alphagov/notifications-api/pull/3300#discussion_r681653248
This commit is contained in:
Ben Thorner
2021-08-04 15:12:09 +01:00
parent 5a1636e41f
commit 0312e2a528
33 changed files with 526 additions and 512 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/app/billing/test_billing.py
View File

@@ -12,7 +12,7 @@ from app.dao.date_util import (
get_month_start_and_end_date_in_utc,
)
from app.models import FactBilling
from tests import create_authorization_header
from tests import create_admin_authorization_header
from tests.app.db import (
create_annual_billing,
create_ft_billing,
@@ -37,7 +37,7 @@ def test_create_update_free_sms_fragment_limit_invalid_schema(client, sample_ser
response = client.post('service/{}/billing/free-sms-fragment-limit'.format(sample_service.id),
data={},
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
json_resp = json.loads(response.get_data(as_text=True))
assert response.status_code == 400
@@ -103,7 +103,7 @@ def test_get_free_sms_fragment_limit(
create_annual_billing(service_id=sample_service.id, free_sms_fragment_limit=11000, financial_year_start=2021)
response_get = client.get(
'service/{}/billing/free-sms-fragment-limit'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
json_resp = json.loads(response_get.get_data(as_text=True))
assert response_get.status_code == 200
@@ -117,7 +117,7 @@ def test_get_free_sms_fragment_limit_current_year_creates_new_row_if_annual_bill
):
response_get = client.get(
'service/{}/billing/free-sms-fragment-limit'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
json_resp = json.loads(response_get.get_data(as_text=True))
assert response_get.status_code == 200
@@ -146,7 +146,7 @@ def test_get_yearly_usage_by_monthly_from_ft_billing_populates_deltas(client, no
assert FactBilling.query.count() == 0
response = client.get('service/{}/billing/ft-monthly-usage?year=2018'.format(service.id),
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
assert response.status_code == 200
assert len(json.loads(response.get_data(as_text=True))) == 1
@@ -178,7 +178,7 @@ def test_get_yearly_usage_by_monthly_from_ft_billing(client, notify_db_session):
postage='second')
response = client.get('service/{}/billing/ft-monthly-usage?year=2016'.format(service.id),
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
json_resp = json.loads(response.get_data(as_text=True))
ft_letters = [x for x in json_resp if x['notification_type'] == 'letter']
@@ -235,7 +235,7 @@ def set_up_yearly_data():
def test_get_yearly_billing_usage_summary_from_ft_billing_returns_400_if_missing_year(client, sample_service):
response = client.get(
'/service/{}/billing/ft-yearly-usage-summary'.format(sample_service.id),
headers=[create_authorization_header()]
headers=[create_admin_authorization_header()]
)
assert response.status_code == 400
assert json.loads(response.get_data(as_text=True)) == {
@@ -248,7 +248,7 @@ def test_get_yearly_billing_usage_summary_from_ft_billing_returns_empty_list_if_
):
response = client.get(
'/service/{}/billing/ft-yearly-usage-summary?year=2016'.format(sample_service.id),
headers=[create_authorization_header()]
headers=[create_admin_authorization_header()]
)
assert response.status_code == 200
assert json.loads(response.get_data(as_text=True)) == []
@@ -258,7 +258,7 @@ def test_get_yearly_billing_usage_summary_from_ft_billing(client, notify_db_sess
service = set_up_yearly_data()
response = client.get('/service/{}/billing/ft-yearly-usage-summary?year=2016'.format(service.id),
headers=[create_authorization_header()]
headers=[create_admin_authorization_header()]
)
assert response.status_code == 200
json_response = json.loads(response.get_data(as_text=True))
@@ -280,7 +280,7 @@ def test_get_yearly_billing_usage_summary_from_ft_billing(client, notify_db_sess
def test_get_yearly_usage_by_monthly_from_ft_billing_all_cases(client, notify_db_session):
service = set_up_data_for_all_cases()
response = client.get('service/{}/billing/ft-monthly-usage?year=2018'.format(service.id),
headers=[('Content-Type', 'application/json'), create_authorization_header()])
headers=[('Content-Type', 'application/json'), create_admin_authorization_header()])
assert response.status_code == 200
json_response = json.loads(response.get_data(as_text=True))
@@ -319,7 +319,7 @@ def test_get_yearly_usage_by_monthly_from_ft_billing_all_cases(client, notify_db
def test_get_yearly_billing_usage_summary_from_ft_billing_all_cases(client, notify_db_session):
service = set_up_data_for_all_cases()
response = client.get('/service/{}/billing/ft-yearly-usage-summary?year=2018'.format(service.id),
headers=[create_authorization_header()])
headers=[create_admin_authorization_header()])
assert response.status_code == 200
json_response = json.loads(response.get_data(as_text=True))