+ {% call banner_wrapper(type='dangerous') %}
+ {% set recipients_remaining_messages = 0 %}
+ {% include "partials/check/too-many-messages-annual.html" %}
+ {% endcall %}
{% elif error == 'message-too-long' %}
{# the only row_errors we can get when sending one off messages is that the message is too long #}
{{ govuk_back_link(back_link) }}
diff --git a/app/templates/views/templates/_template.html b/app/templates/views/templates/_template.html
index 86c19a127e..038db7fb62 100644
--- a/app/templates/views/templates/_template.html
+++ b/app/templates/views/templates/_template.html
@@ -1,4 +1,13 @@
{% from 'components/message-count-label.html' import message_count_label %}
+{% from 'components/remaining-messages-summary.html' import remaining_messages_summary with context %}
+
+
+
{% if template._template.archived %}
@@ -13,20 +22,22 @@
{% else %}
{% if current_user.has_permissions('send_messages', restrict_admin_usage=True) %}
-
{{ _('Ready to send?') }}
+
{{ heading }}
+
+ {{ remaining_messages_summary(dailyLimit, dailyUsed, yearlyLimit, yearlyUsed, notification_type, yearlyRemaining == 0 or dailyRemaining == 0) }}
-
+ {% if yearlyRemaining > 0 and dailyRemaining > 0 %}
+
+ {% endif %}
{% endif %}
{% endif %}
-
+
{{ template|string|translate_preview_template }}
-
-
-
+
\ No newline at end of file
diff --git a/app/translations/csv/fr.csv b/app/translations/csv/fr.csv
index 57aead4f21..6b2b5dc2cb 100644
--- a/app/translations/csv/fr.csv
+++ b/app/translations/csv/fr.csv
@@ -2027,4 +2027,14 @@
"Annual usage","Utilisation annuelle"
"resets at 7pm Eastern Time","Réinitialisation à 19 h, heure de l’Est"
"Visit usage report","Consulter le rapport d’utilisation"
-"Month by month totals","Totaux mensuels"
\ No newline at end of file
+"Month by month totals","Totaux mensuels"
+"
{} can only send
{} more {} until annual limit resets","FR:
{} can only send
{} more {} until annual limit resets"
+"To send some of these messages now, edit the spreadsheet to
{} recipients maximum. ","FR: To send some of these messages now, edit the spreadsheet to
{} recipients maximum. "
+"To send to recipients you removed, wait until
April 1, {} or contact them some other way.","FR: To send to recipients you removed, wait until
April 1, {} or contact them some other way"
+"These messages exceed the annual limit","FR: These messages exceed the annual limit"
+"
{} cannot send any more {} until
April 1, {}","FR:
{} cannot send any more {} until
April 1, {}"
+"For more information, visit the
usage report for {}.","FR: For more information, visit the
usage report for {}."
+"This message exceeds your annual limit","FR: This message exceeds your annual limit"
+"email messages","courriels"
+"Sending paused until annual limit resets","FR: Sending paused until annual limit resets"
+"Sending paused until 7pm ET. You can schedule more messages to send later.","FR: Sending paused until 7pm ET. You can schedule more messages to send later."
diff --git a/tests/app/main/views/test_send.py b/tests/app/main/views/test_send.py
index 78df4b10cc..fdc833ff55 100644
--- a/tests/app/main/views/test_send.py
+++ b/tests/app/main/views/test_send.py
@@ -41,6 +41,7 @@
mock_get_service_letter_template,
mock_get_service_template,
normalize_spaces,
+ set_config,
)
template_types = ["email", "sms"]
@@ -3401,3 +3402,296 @@ class Object(object):
multiple_choise_options = [x.text.strip() for x in options]
assert multiple_choise_options == expected_filenames
+
+
+class TestAnnualLimitsSend:
+ @pytest.mark.parametrize(
+ "num_being_sent, num_sent_today, num_sent_this_year, expect_to_see_annual_limit_msg, expect_to_see_daily_limit_msg",
+ [
+ # annual limit for mock_get_live_service is 10,000email/10,000sms
+ # daily limit for mock_get_live_service is 1,000email/1,000sms
+ # 1000 have already been sent today, trying to send 100 more [over both limits]
+ (100, 1000, 10000, True, False),
+ # No sent yet today or this year, trying to send 1001 [over both limits]
+ (10001, 0, 0, True, False),
+ # 600 have already been sent this year, trying to send 500 more [over annual limit but not daily]
+ (500, 0, 9600, True, False),
+ # No sent yet today or this year, trying to send 1001 [over daily limit but not annual]
+ (1001, 0, 0, False, True),
+ # No sent yet today or this year, trying to send 100 [over neither limit]
+ (100, 0, 0, False, False),
+ ],
+ ids=[
+ "email_over_both_limits",
+ "email_over_both_limits2",
+ "email_over_annual_but_not_daily",
+ "email_over_daily_but_not_annual",
+ "email_over_neither",
+ ],
+ )
+ def test_email_send_fails_approrpiately_when_over_limits(
+ self,
+ mocker,
+ client_request,
+ mock_get_live_service, # set email_annual_limit and sms_annual_limit to 1000
+ mock_get_users_by_service,
+ mock_get_service_email_template_without_placeholders,
+ mock_get_template_statistics,
+ mock_get_job_doesnt_exist,
+ mock_get_jobs,
+ mock_s3_set_metadata,
+ mock_notification_counts_client,
+ mock_daily_sms_fragment_count,
+ mock_daily_email_count,
+ fake_uuid,
+ num_being_sent,
+ num_sent_today,
+ num_sent_this_year,
+ expect_to_see_annual_limit_msg,
+ expect_to_see_daily_limit_msg,
+ app_,
+ ):
+ with set_config(app_, "FF_ANNUAL_LIMIT", True):
+ mocker.patch(
+ "app.main.views.send.s3download",
+ return_value=",\n".join(
+ ["email address"] + ([mock_get_users_by_service(None)[0]["email_address"]] * num_being_sent)
+ ),
+ )
+
+ mock_notification_counts_client.get_limit_stats.return_value = {
+ "email": {
+ "annual": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": 10000
+ - num_sent_this_year
+ - num_sent_today, # The number of email notifications remaining this year
+ },
+ "daily": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": 1000 - num_sent_today, # The number of email notifications remaining today
+ },
+ }
+ }
+
+ # mock that we've already sent `emails_sent_today` emails today
+ mock_daily_email_count.return_value = num_sent_today
+ mock_daily_sms_fragment_count.return_value = 900 # not used in test but needs a value
+
+ with client_request.session_transaction() as session:
+ session["file_uploads"] = {
+ fake_uuid: {
+ "template_id": fake_uuid,
+ "notification_count": 1,
+ "valid": True,
+ }
+ }
+
+ page = client_request.get(
+ "main.check_messages",
+ service_id=SERVICE_ONE_ID,
+ template_id=fake_uuid,
+ upload_id=fake_uuid,
+ original_file_name="valid.csv",
+ _test_page_title=False,
+ )
+
+ if expect_to_see_annual_limit_msg:
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is not None
+ else:
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is None
+
+ if expect_to_see_daily_limit_msg:
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is not None
+ else:
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is None
+
+ @pytest.mark.parametrize(
+ "num_being_sent, num_sent_today, num_sent_this_year, expect_to_see_annual_limit_msg, expect_to_see_daily_limit_msg",
+ [
+ # annual limit for mock_get_live_service is 10,000email/10,000sms
+ # daily limit for mock_get_live_service is 1,000email/1,000sms
+ # 1000 have already been sent today, trying to send 100 more [over both limits]
+ (100, 1000, 10000, True, False),
+ # No sent yet today or this year, trying to send 1001 [over both limits]
+ (10001, 0, 0, True, False),
+ # 600 have already been sent this year, trying to send 500 more [over annual limit but not daily]
+ (500, 0, 9600, True, False),
+ # No sent yet today or this year, trying to send 1001 [over daily limit but not annual]
+ (1001, 0, 0, False, True),
+ # No sent yet today or this year, trying to send 100 [over neither limit]
+ (100, 0, 0, False, False),
+ ],
+ ids=[
+ "sms_over_both_limits",
+ "sms_over_both_limits2",
+ "sms_over_annual_but_not_daily",
+ "sms_over_daily_but_not_annual",
+ "sms_over_neither",
+ ],
+ )
+ def test_sms_send_fails_approrpiately_when_over_limits(
+ self,
+ mocker,
+ client_request,
+ mock_get_live_service, # set email_annual_limit and sms_annual_limit to 1000
+ mock_get_users_by_service,
+ mock_get_service_sms_template_without_placeholders,
+ mock_get_template_statistics,
+ mock_get_job_doesnt_exist,
+ mock_get_jobs,
+ mock_s3_set_metadata,
+ mock_notification_counts_client,
+ mock_daily_sms_fragment_count,
+ mock_daily_email_count,
+ fake_uuid,
+ num_being_sent,
+ num_sent_today,
+ num_sent_this_year,
+ expect_to_see_annual_limit_msg,
+ expect_to_see_daily_limit_msg,
+ app_,
+ ):
+ with set_config(app_, "FF_ANNUAL_LIMIT", True): # REMOVE LINE WHEN FF REMOVED
+ mocker.patch(
+ "app.main.views.send.s3download",
+ return_value=",\n".join(
+ ["phone number"] + ([mock_get_users_by_service(None)[0]["mobile_number"]] * num_being_sent)
+ ),
+ )
+ mock_notification_counts_client.get_limit_stats.return_value = {
+ "sms": {
+ "annual": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": 10000
+ - num_sent_this_year
+ - num_sent_today, # The number of email notifications remaining this year
+ },
+ "daily": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": 1000 - num_sent_today, # The number of email notifications remaining today
+ },
+ }
+ }
+ # mock that we've already sent `num_sent_today` emails today
+ mock_daily_email_count.return_value = 900 # not used in test but needs a value
+ mock_daily_sms_fragment_count.return_value = num_sent_today
+
+ with client_request.session_transaction() as session:
+ session["file_uploads"] = {
+ fake_uuid: {
+ "template_id": fake_uuid,
+ "notification_count": 1,
+ "valid": True,
+ }
+ }
+
+ page = client_request.get(
+ "main.check_messages",
+ service_id=SERVICE_ONE_ID,
+ template_id=fake_uuid,
+ upload_id=fake_uuid,
+ original_file_name="valid.csv",
+ _test_page_title=False,
+ )
+
+ if expect_to_see_annual_limit_msg:
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is not None
+ else:
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is None
+
+ if expect_to_see_daily_limit_msg:
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is not None
+ else:
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is None
+
+ @pytest.mark.parametrize(
+ "num_to_send, remaining_daily, remaining_annual, error_shown",
+ [
+ (2, 2, 2, "none"),
+ (5, 5, 4, "annual"),
+ (5, 4, 5, "daily"),
+ (5, 4, 4, "annual"),
+ ],
+ )
+ def test_correct_error_displayed(
+ self,
+ mocker,
+ client_request,
+ mock_get_live_service, # set email_annual_limit and sms_annual_limit to 1000
+ mock_get_users_by_service,
+ mock_get_service_email_template_without_placeholders,
+ mock_get_template_statistics,
+ mock_get_job_doesnt_exist,
+ mock_get_jobs,
+ mock_s3_set_metadata,
+ mock_daily_email_count,
+ mock_notification_counts_client,
+ fake_uuid,
+ num_to_send,
+ remaining_daily,
+ remaining_annual,
+ error_shown,
+ app_,
+ ):
+ with set_config(app_, "FF_ANNUAL_LIMIT", True): # REMOVE LINE WHEN FF REMOVED
+ # mock that `num_sent_this_year` have already been sent this year
+ mock_notification_counts_client.get_limit_stats.return_value = {
+ "email": {
+ "annual": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_annual, # The number of email notifications remaining this year
+ },
+ "daily": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_daily, # The number of email notifications remaining today
+ },
+ }
+ }
+
+ # only change this value when we're expecting an error
+ if error_shown != "none":
+ mock_daily_email_count.return_value = 1000 - (
+ num_to_send - 1
+ ) # svc limit is 1000 - exceeding the daily limit is calculated based off of this
+ else:
+ mock_daily_email_count.return_value = 0 # none sent
+
+ mocker.patch(
+ "app.main.views.send.s3download",
+ return_value=",\n".join(
+ ["email address"] + ([mock_get_users_by_service(None)[0]["email_address"]] * num_to_send)
+ ),
+ )
+ with client_request.session_transaction() as session:
+ session["file_uploads"] = {
+ fake_uuid: {
+ "template_id": fake_uuid,
+ "notification_count": 1,
+ "valid": True,
+ }
+ }
+ page = client_request.get(
+ "main.check_messages",
+ service_id=SERVICE_ONE_ID,
+ template_id=fake_uuid,
+ upload_id=fake_uuid,
+ original_file_name="valid.csv",
+ _test_page_title=False,
+ )
+
+ if error_shown == "annual":
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is not None
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is None
+ elif error_shown == "daily":
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is None
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is not None
+ elif error_shown == "none":
+ assert page.find(attrs={"data-testid": "exceeds-annual"}) is None
+ assert page.find(attrs={"data-testid": "exceeds-daily"}) is None
diff --git a/tests/app/main/views/test_templates.py b/tests/app/main/views/test_templates.py
index b48234af1f..54df21e58f 100644
--- a/tests/app/main/views/test_templates.py
+++ b/tests/app/main/views/test_templates.py
@@ -1,6 +1,6 @@
from datetime import datetime
from functools import partial
-from unittest.mock import ANY, MagicMock, Mock
+from unittest.mock import ANY, MagicMock, Mock, patch
import pytest
from flask import url_for
@@ -54,6 +54,12 @@
DEFAULT_PROCESS_TYPE = TemplateProcessTypes.BULK.value
+@pytest.fixture
+def mock_notification_counts_client():
+ with patch("app.main.views.templates.notification_counts_client") as mock:
+ yield mock
+
+
class TestRedisPreviewUtilities:
def test_set_get(self, fake_uuid, mocker):
mock_redis_obj = MockRedis()
@@ -2742,3 +2748,64 @@ def test_should_hide_category_name_from_template_list_if_marked_hidden(
# assert that "HIDDEN_CATEGORY" is not found anywhere in the page using beautifulsoup
assert "HIDDEN_CATEGORY" not in page.text
assert not page.find(text="HIDDEN_CATEGORY")
+
+
+class TestAnnualLimits:
+ @pytest.mark.parametrize(
+ "remaining_daily, remaining_annual, buttons_shown",
+ [
+ (10, 100, True), # Within both limits
+ (0, 100, False), # Exceeds daily limit
+ (10, 0, False), # Exceeds annual limit
+ (0, 0, False), # Exceeds both limits
+ (1, 1, True), # Exactly at both limits
+ ],
+ )
+ def test_should_hide_send_buttons_when_appropriate(
+ self,
+ client_request,
+ mock_get_service_template,
+ mock_get_template_folders,
+ mock_notification_counts_client,
+ fake_uuid,
+ remaining_daily,
+ remaining_annual,
+ buttons_shown,
+ ):
+ mock_notification_counts_client.get_limit_stats.return_value = {
+ "email": {
+ "annual": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_annual, # The number of email notifications remaining this year
+ },
+ "daily": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_daily, # The number of email notifications remaining today
+ },
+ },
+ "sms": {
+ "annual": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_annual, # The number of email notifications remaining this year
+ },
+ "daily": {
+ "limit": 1, # doesn't matter for our test
+ "sent": 1, # doesn't matter for our test
+ "remaining": remaining_daily, # The number of email notifications remaining today
+ },
+ },
+ }
+
+ page = client_request.get(
+ ".view_template",
+ service_id=SERVICE_ONE_ID,
+ template_id=fake_uuid,
+ _test_page_title=False,
+ )
+ if buttons_shown:
+ assert page.find(attrs={"data-testid": "send-buttons"}) is not None
+ else:
+ assert page.find(attrs={"data-testid": "send-buttons"}) is None
diff --git a/tests/app/notify_client/test_notification_counts_client.py b/tests/app/notify_client/test_notification_counts_client.py
new file mode 100644
index 0000000000..342aaa2917
--- /dev/null
+++ b/tests/app/notify_client/test_notification_counts_client.py
@@ -0,0 +1,199 @@
+from datetime import datetime
+from unittest.mock import Mock, patch
+
+import pytest
+
+from app.notify_client.notification_counts_client import NotificationCounts
+
+
+@pytest.fixture
+def mock_redis():
+ with patch("app.notify_client.notification_counts_client.redis_client") as mock:
+ yield mock
+
+
+@pytest.fixture
+def mock_template_stats():
+ with patch("app.notify_client.notification_counts_client.template_statistics_client") as mock:
+ yield mock
+
+
+@pytest.fixture
+def mock_service_api():
+ with patch("app.notify_client.notification_counts_client.service_api_client") as mock:
+ yield mock
+
+
+@pytest.fixture
+def mock_get_all_notification_counts_for_today():
+ with patch("app.notify_client.notification_counts_client.get_all_notification_counts_for_today") as mock:
+ yield mock
+
+
+class TestNotificationCounts:
+ def test_get_all_notification_counts_for_today_redis_has_data(self, mock_redis):
+ # Setup
+ mock_redis.get.side_effect = [5, 10] # sms, email
+ wrapper = NotificationCounts()
+
+ # Execute
+ result = wrapper.get_all_notification_counts_for_today("service-123")
+
+ # Assert
+ assert result == {"sms": 5, "email": 10}
+ assert mock_redis.get.call_count == 2
+
+ @pytest.mark.parametrize(
+ "redis_side_effect, expected_result",
+ [
+ ([None, None], {"sms": 10, "email": 10}),
+ ([None, 10], {"sms": 10, "email": 10}), # Falls back to API if either is None
+ ([10, None], {"sms": 10, "email": 10}), # Falls back to API if either is None
+ ],
+ )
+ def test_get_all_notification_counts_for_today_redis_missing_data(
+ self, mock_redis, mock_template_stats, redis_side_effect, expected_result
+ ):
+ # Setup
+ mock_redis.get.side_effect = redis_side_effect
+ mock_template_stats.get_template_statistics_for_service.return_value = [
+ {"template_id": "a1", "template_type": "sms", "count": 3, "status": "delivered"},
+ {"template_id": "a2", "template_type": "email", "count": 7, "status": "temporary-failure"},
+ {"template_id": "a3", "template_type": "email", "count": 3, "status": "delivered"},
+ {"template_id": "a4", "template_type": "sms", "count": 7, "status": "delivered"},
+ ]
+
+ wrapper = NotificationCounts()
+
+ # Execute
+ result = wrapper.get_all_notification_counts_for_today("service-123")
+
+ # Assert
+ assert result == {"sms": 10, "email": 10}
+ mock_template_stats.get_template_statistics_for_service.assert_called_once()
+
+ def test_get_all_notification_counts_for_year(self, mock_service_api):
+ # Setup
+ mock_service_api.get_monthly_notification_stats.return_value = {
+ "data": {
+ "2024-01": {
+ "sms": {"sent": 1, "temporary-failure:": 22},
+ "email": {"delivered": 1, "permanent-failure": 1, "sending": 12, "technical-failure": 1},
+ },
+ "2024-02": {"sms": {"sent": 1}, "email": {"delivered": 1}},
+ }
+ }
+ wrapper = NotificationCounts()
+
+ with patch.object(wrapper, "get_all_notification_counts_for_today") as mock_today:
+ mock_today.return_value = {"sms": 5, "email": 5}
+
+ # Execute
+ result = wrapper.get_all_notification_counts_for_year("service-123", 2024)
+
+ # Assert
+ assert result["sms"] == 29 # 1 + 22 + 1 + 5
+ assert result["email"] == 21 # 1 + 1 + 12 + 1 + 1 + 5
+
+ def test_get_limit_stats(self, mocker):
+ # Setup
+ mock_service = Mock(id="service-1", email_annual_limit=1000, sms_annual_limit=500, message_limit=100, sms_daily_limit=50)
+
+ mock_notification_client = NotificationCounts()
+
+ # Mock the dependency methods
+
+ mocker.patch.object(
+ mock_notification_client, "get_all_notification_counts_for_today", return_value={"email": 20, "sms": 10}
+ )
+ mocker.patch.object(
+ mock_notification_client, "get_all_notification_counts_for_year", return_value={"email": 200, "sms": 100}
+ )
+
+ # Execute
+ result = mock_notification_client.get_limit_stats(mock_service)
+
+ # Assert
+ assert result == {
+ "email": {
+ "annual": {
+ "limit": 1000,
+ "sent": 200,
+ "remaining": 800,
+ },
+ "daily": {
+ "limit": 100,
+ "sent": 20,
+ "remaining": 80,
+ },
+ },
+ "sms": {
+ "annual": {
+ "limit": 500,
+ "sent": 100,
+ "remaining": 400,
+ },
+ "daily": {
+ "limit": 50,
+ "sent": 10,
+ "remaining": 40,
+ },
+ },
+ }
+
+ @pytest.mark.parametrize(
+ "today_counts,year_counts,expected_remaining",
+ [
+ (
+ {"email": 0, "sms": 0},
+ {"email": 0, "sms": 0},
+ {"email": {"annual": 1000, "daily": 100}, "sms": {"annual": 500, "daily": 50}},
+ ),
+ (
+ {"email": 100, "sms": 50},
+ {"email": 1000, "sms": 500},
+ {"email": {"annual": 0, "daily": 0}, "sms": {"annual": 0, "daily": 0}},
+ ),
+ (
+ {"email": 50, "sms": 25},
+ {"email": 500, "sms": 250},
+ {"email": {"annual": 500, "daily": 50}, "sms": {"annual": 250, "daily": 25}},
+ ),
+ ],
+ )
+ def test_get_limit_stats_remaining_calculations(self, mocker, today_counts, year_counts, expected_remaining):
+ # Setup
+ mock_service = Mock(id="service-1", email_annual_limit=1000, sms_annual_limit=500, message_limit=100, sms_daily_limit=50)
+
+ mock_notification_client = NotificationCounts()
+
+ mocker.patch.object(mock_notification_client, "get_all_notification_counts_for_today", return_value=today_counts)
+ mocker.patch.object(mock_notification_client, "get_all_notification_counts_for_year", return_value=year_counts)
+
+ # Execute
+ result = mock_notification_client.get_limit_stats(mock_service)
+
+ # Assert remaining counts
+ assert result["email"]["annual"]["remaining"] == expected_remaining["email"]["annual"]
+ assert result["email"]["daily"]["remaining"] == expected_remaining["email"]["daily"]
+ assert result["sms"]["annual"]["remaining"] == expected_remaining["sms"]["annual"]
+ assert result["sms"]["daily"]["remaining"] == expected_remaining["sms"]["daily"]
+
+ def test_get_limit_stats_dependencies_called(self, mocker):
+ # Setup
+ mock_service = Mock(id="service-1", email_annual_limit=1000, sms_annual_limit=500, message_limit=100, sms_daily_limit=50)
+ mock_notification_client = NotificationCounts()
+
+ mock_today = mocker.patch.object(
+ mock_notification_client, "get_all_notification_counts_for_today", return_value={"email": 0, "sms": 0}
+ )
+ mock_year = mocker.patch.object(
+ mock_notification_client, "get_all_notification_counts_for_year", return_value={"email": 0, "sms": 0}
+ )
+
+ # Execute
+ mock_notification_client.get_limit_stats(mock_service)
+
+ # Assert dependencies called
+ mock_today.assert_called_once_with(mock_service.id)
+ mock_year.assert_called_once_with(mock_service.id, datetime.now().year)