mirror of
https://github.com/GSA/notifications-api.git
synced 2026-03-02 06:50:08 -05:00
Merge branch 'master' into async-workers
This commit is contained in:
@@ -542,7 +542,9 @@ def dao_fetch_monthly_historical_usage_by_template_for_service(service_id, year)
|
||||
Template, Notification.template_id == Template.id,
|
||||
).filter(
|
||||
Notification.created_at >= start_date,
|
||||
Notification.service_id == service_id
|
||||
Notification.service_id == service_id,
|
||||
# we don't want to include test keys
|
||||
Notification.key_type != KEY_TYPE_TEST
|
||||
).group_by(
|
||||
Notification.template_id,
|
||||
Template.name,
|
||||
@@ -562,7 +564,7 @@ def dao_fetch_monthly_historical_usage_by_template_for_service(service_id, year)
|
||||
add_to_stats = False
|
||||
|
||||
if add_to_stats:
|
||||
new_stat = type("", (), {})()
|
||||
new_stat = type("StatsTemplateUsageByMonth", (), {})()
|
||||
new_stat.template_id = today_result.template_id
|
||||
new_stat.template_type = today_result.template_type
|
||||
new_stat.name = today_result.name
|
||||
|
||||
@@ -64,8 +64,13 @@ from app.models import (
|
||||
)
|
||||
|
||||
from tests.app.db import (
|
||||
create_inbound_number, create_user, create_service, create_service_with_inbound_number,
|
||||
create_service_with_defined_sms_sender
|
||||
create_inbound_number,
|
||||
create_user,
|
||||
create_service,
|
||||
create_service_with_inbound_number,
|
||||
create_service_with_defined_sms_sender,
|
||||
create_template,
|
||||
create_notification as create_notification_db
|
||||
)
|
||||
from tests.app.conftest import (
|
||||
sample_notification as create_notification,
|
||||
@@ -1477,3 +1482,27 @@ def test_dao_fetch_monthly_historical_usage_by_template_for_service_only_returns
|
||||
)
|
||||
|
||||
assert len(result) == 1
|
||||
|
||||
|
||||
@freeze_time("2018-01-01 11:09:00.000000")
|
||||
def test_dao_fetch_monthly_historical_usage_by_template_for_service_ignores_test_api_keys(sample_service):
|
||||
template_1 = create_template(sample_service, template_name='1')
|
||||
template_2 = create_template(sample_service, template_name='2')
|
||||
template_3 = create_template(sample_service, template_name='3')
|
||||
|
||||
create_notification_db(template_1, key_type=KEY_TYPE_TEST)
|
||||
create_notification_db(template_2, key_type=KEY_TYPE_TEAM)
|
||||
create_notification_db(template_3, key_type=KEY_TYPE_NORMAL)
|
||||
|
||||
results = sorted(
|
||||
dao_fetch_monthly_historical_usage_by_template_for_service(sample_service.id, 2017),
|
||||
key=lambda x: x.name
|
||||
)
|
||||
|
||||
assert len(results) == 2
|
||||
# template_1 only used with test keys
|
||||
assert results[0].template_id == template_2.id
|
||||
assert results[0].count == 1
|
||||
|
||||
assert results[1].template_id == template_3.id
|
||||
assert results[1].count == 1
|
||||
|
||||
Reference in New Issue
Block a user