VED-79: Subscribe to MNS notification

Author: Akol125

azure/azure-pr-pipeline.yml

@@ -40,6 +40,10 @@ extends:
           - template: ./templates/post-deploy.yml
             parameters:
               aws_account_type: 'dev'
+          - script: |
+              echo "Creating MNS subscription..."
+              python3 backend/src/mns/subscribe.py
+            displayName: "Subscribe SQS to MNS signal"
       - environment: internal-dev-sandbox
         proxy_path: sandbox
         jinja_templates:

backend/src/mns_service.py

@@ -0,0 +1,42 @@
+import requests
+import uuid
+import json
+from authentication import AppRestrictedAuth
+from models.errors import UnhandledResponseError
+
+SQS_ARN = "arn:aws:sqs:eu-west-2:345594581768:imms-pr-655-id-sync-queue"
+MNS_URL = "https://int.api.service.nhs.uk/multicast-notification-service/subscriptions"
+
+class MnsService:
+    def __init__(self, authenticator: AppRestrictedAuth):
+        self.authenticator = authenticator
+        
+        
+        
+    def subscribeNotification(self)  -> dict | None:
+        access_token = self.authenticator.get_access_token()
+        request_headers = {
+            'Authorization': f'Bearer {access_token}',
+            'X-Correlation-ID': str(uuid.uuid4())
+        }
+    
+        subscription_payload = {
+        "resourceType": "Subscription",
+        "status": "requested",
+        "reason": "Subscribe SQS to MNS test-signal",
+        "criteria": "eventType=mns-test-signal-1",
+        "channel": {
+            "type": "message",
+            "endpoint": SQS_ARN,
+            "payload": "application/json"
+        }
+    }
+        response = requests.post(MNS_URL, headers=request_headers, data=json.dumps(subscription_payload))
+
+        if response.status_code == 200:
+            return response.json()
+        elif response.status_code == 404:
+            return None
+        else:
+            msg = "Please provide the correct resource type for this endpoint"
+            raise UnhandledResponseError(response=response.json(), message=msg)

backend/src/subscribe_mns.py

@@ -0,0 +1,8 @@
+from authentication import AppRestrictedAuth
+from mns_service import MnsService
+
+if __name__ == "__main__":
+    auth = AppRestrictedAuth()
+    mns = MnsService(authenticator=auth)
+    result = mns.subscribeNotification()
+    print("Subscription Result:", result)

backend/tests/test_mns_service.py

@@ -0,0 +1,61 @@
+import unittest
+from unittest.mock import patch, MagicMock, Mock, create_autospec
+from mns_service import MnsService
+from authentication import AppRestrictedAuth
+from models.errors import UnhandledResponseError
+
+
+class TestMnsService(unittest.TestCase):
+    
+    def setUp(self):
+        # Common mock setup
+        self.authenticator = create_autospec(AppRestrictedAuth)
+        self.authenticator.get_access_token.return_value = "mocked_token"
+        self.mock_secret_manager = Mock()
+        self.mock_cache = Mock()
+    
+    @patch("mns_service.requests.post")
+    def test_successful_subscription(self, mock_post):
+        # Arrange
+        mock_response = MagicMock()
+        mock_response.status_code = 200
+        mock_response.json.return_value = {"subscriptionId": "abc123"}
+        mock_post.return_value = mock_response
+
+        service = MnsService(self.authenticator)
+
+        # Act
+        result = service.subscribeNotification()
+
+        # Assert
+        self.assertEqual(result, {"subscriptionId": "abc123"})
+        mock_post.assert_called_once()
+        self.authenticator.get_access_token.assert_called_once()
+
+    @patch("mns_service.requests.post")
+    def test_not_found_subscription(self, mock_post):
+        mock_response = MagicMock()
+        mock_response.status_code = 404
+        mock_post.return_value = mock_response
+
+        service = MnsService(self.authenticator)
+        result = service.subscribeNotification()
+
+        self.assertIsNone(result)
+
+    @patch("mns_service.requests.post")
+    def test_unhandled_error(self, mock_post):
+        mock_response = MagicMock()
+        mock_response.status_code = 500
+        mock_response.json.return_value = {"error": "Server error"}
+        mock_post.return_value = mock_response
+
+        service = MnsService(self.authenticator)
+
+        with self.assertRaises(UnhandledResponseError) as context:
+            service.subscribeNotification()
+
+        self.assertIn("Please provide the correct resource type", str(context.exception))
+
+if __name__ == "__main__":
+    unittest.main()

backend/tests/test_subscription.py

@@ -0,0 +1,32 @@
+import unittest
+from unittest.mock import patch, MagicMock, create_autospec
+from mns_service import MnsService
+from authentication import AppRestrictedAuth
+
+class TestMainSubscriptionCall(unittest.TestCase):
+    
+    def setUp(self):
+        # Common mock setup
+        self.authenticator = create_autospec(AppRestrictedAuth)
+        self.authenticator.get_access_token.return_value = "mocked_token"
+    
+    @patch("mns_service.requests.post")
+    def test_main_subscription_call(self, mock_post):
+        # Arrange
+        mock_response = MagicMock()
+        mock_response.status_code = 200
+        mock_response.json.return_value = {"subscriptionId": "abc123"}
+        mock_post.return_value = mock_response
+
+        mns = MnsService(self.authenticator)
+
+        # Act
+        result = mns.subscribeNotification()
+
+        # Assert
+        self.assertEqual(result, {"subscriptionId": "abc123"})
+        mock_post.assert_called_once()
+        self.authenticator.get_access_token.assert_called_once()
+
+if __name__ == "__main__":
+    unittest.main()