chore: add ENABLE_SQS_MESSAGE_GROUP_IDS and test helper for message group id

Author: joybytes

app/config.py

@@ -139,6 +139,9 @@ class Config:
     # URL of redis instance
     REDIS_URL = os.getenv("REDIS_URL")
     REDIS_ENABLED = False if os.environ.get("REDIS_ENABLED") == "0" else True
+
+    ENABLE_SQS_MESSAGE_GROUP_IDS = os.environ.get("ENABLE_SQS_MESSAGE_GROUP_IDS", "1") == "1"
+
     EXPIRE_CACHE_TEN_MINUTES = 600
     EXPIRE_CACHE_EIGHT_DAYS = 8 * 24 * 60 * 60
 
@@ -587,6 +590,7 @@ class Development(Config):
     SERVER_NAME = os.getenv("SERVER_NAME")
 
     REDIS_ENABLED = os.getenv("REDIS_ENABLED") == "1"
+    ENABLE_SQS_MESSAGE_GROUP_IDS = os.environ.get("ENABLE_SQS_MESSAGE_GROUP_IDS", "1") == "1"
 
     S3_BUCKET_CSV_UPLOAD = "development-notifications-csv-upload"
     S3_BUCKET_CONTACT_LIST = "development-contact-list"

tests/conftest.py

@@ -2,6 +2,7 @@
 import subprocess
 from collections import namedtuple
 from contextlib import contextmanager
+from unittest.mock import patch
 from urllib.parse import urlparse
 
 import freezegun
@@ -265,6 +266,12 @@ def set_config_values(app, dict):
             app.config[key] = old_values[key]
 
 
+@contextmanager
+def _with_message_group_id(task, value):
+    with patch("notifications_utils.celery.NotifyTask.message_group_id", new=value, create=True):
+        yield
+
+
 class Matcher:
     def __init__(self, description, key):
         self.description = description