Sends sequential number to Vodafone as link test

app/celery/broadcast_message_tasks.py

@@ -2,10 +2,11 @@ import uuid
 
 from flask import current_app
 from notifications_utils.statsd_decorators import statsd
+from sqlalchemy.schema import Sequence
 
-from app import cbc_proxy_client, notify_celery
+from app import cbc_proxy_client, db, notify_celery
 from app.config import QueueNames
-from app.models import BroadcastEventMessageType
+from app.models import BroadcastEventMessageType, BroadcastProvider
 from app.dao.broadcast_message_dao import dao_get_broadcast_event_by_id, create_broadcast_provider_message
 
 
@@ -78,6 +79,10 @@ def send_broadcast_provider_message(broadcast_event_id, provider):
 @notify_celery.task(name='trigger-link-test')
 def trigger_link_test(provider):
     identifier = str(uuid.uuid4())
+    sequential_number = None
+    if provider == BroadcastProvider.VODAFONE:
+        sequence = Sequence('broadcast_provider_message_number_seq')
+        sequential_number = db.session.connection().execute(sequence)
     message = f"Sending a link test to CBC proxy for provider {provider} with ID {identifier}"
     current_app.logger.info(message)
-    cbc_proxy_client.get_proxy(provider).send_link_test(identifier)
+    cbc_proxy_client.get_proxy(provider).send_link_test(identifier, sequential_number)

app/clients/cbc_proxy.py

@@ -42,6 +42,7 @@ class CBCProxyClient:
         proxy_classes = {
             'canary': CBCProxyCanary,
             BroadcastProvider.EE: CBCProxyEE,
+            BroadcastProvider.VODAFONE: CBCProxyVodafone,
         }
         return proxy_classes[provider](self._lambda_client)
 
@@ -61,6 +62,7 @@ class CBCProxyClientBase:
     def send_link_test(
         self,
         identifier,
+        sequential_number=None
     ):
         pass
 
@@ -131,6 +133,7 @@ class CBCProxyEE(CBCProxyClientBase):
     def send_link_test(
         self,
         identifier,
+        sequential_number=None
     ):
         """
         link test - open up a connection to a specific provider, and send them an xml payload with a <msgType> of
@@ -155,3 +158,20 @@ class CBCProxyEE(CBCProxyClientBase):
             'expires': expires,
         }
         self._invoke_lambda(payload=payload)
+
+
+class CBCProxyVodafone(CBCProxyClientBase):
+    lambda_name = 'bt-ee-1-proxy'
+
+    def send_link_test(
+        self,
+        identifier,
+        sequential_number
+    ):
+        """
+        link test - open up a connection to a specific provider, and send them an xml payload with a <msgType> of
+        test.
+        """
+        payload = {'message_type': 'test', 'identifier': identifier, 'message_number': sequential_number}
+
+        self._invoke_lambda(payload=payload)