Check for services sending sms messages to tv numbers

tests/app/celery/test_scheduled_tasks.py

@@ -2,6 +2,7 @@ from datetime import datetime, timedelta
 from unittest.mock import call
 
 import pytest
+from collections import namedtuple
 from freezegun import freeze_time
 from mock import mock
 
@@ -497,25 +498,46 @@ def test_check_for_missing_rows_in_completed_jobs_uses_sender_id(mocker, sample_
     )
 
 
-@pytest.mark.parametrize("failure_rates, expected_message", [
+MockServicesSendingToTVNumbers = namedtuple(
+    'ServicesSendingToTVNumbers',
+    [
+        'service_id',
+        'service_name',
+        'notification_count',
+    ]
+)
+
+
+@pytest.mark.parametrize("failure_rates, sms_to_tv_numbers, expected_message", [
     [
         [{"id": "123", "name": "Service 1", "permanent_failure_rate": 0.3}],
+        [],
         "1 service(s) have had high permanent-failure rates for sms messages in last "
-        "24 hours: service id: 123 failure rate: 0.3, "
+        "24 hours:\nservice id: 123 failure rate: 0.3,\n"
+    ],
+    [
+        [],
+        [MockServicesSendingToTVNumbers("123", "Service 1", 300)],
+        "1 service(s) have sent over 100 sms messages to tv numbers in last 24 hours:\n"
+        "service id: 123, count of sms to tv numbers: 300,\n"
     ]
 ])
 def test_check_for_services_with_high_failure_rates_or_sending_to_tv_numbers(
-    mocker, notify_db_session, failure_rates, expected_message
+    mocker, notify_db_session, failure_rates, sms_to_tv_numbers, expected_message
 ):
     mock_logger = mocker.patch('app.celery.tasks.current_app.logger.exception')
     mock_create_ticket = mocker.patch('app.celery.scheduled_tasks.zendesk_client.create_ticket')
     mock_failure_rates = mocker.patch(
         'app.celery.scheduled_tasks.get_services_with_high_failure_rates', return_value=failure_rates
     )
+    mock_sms_to_tv_numbers = mocker.patch(
+        'app.celery.scheduled_tasks.dao_find_services_sending_to_tv_numbers', return_value=sms_to_tv_numbers
+    )
 
     check_for_services_with_high_failure_rates_or_sending_to_tv_numbers()
 
     assert mock_failure_rates.called
+    assert mock_sms_to_tv_numbers.called
     mock_logger.assert_called_once_with(expected_message)
     mock_create_ticket.assert_called_with(
         message=expected_message,