mns subscription

Author: Akol125

mns_subscription/src/mns_service.py

@@ -47,5 +47,5 @@ def subscribe_notification(self) -> dict | None:
         elif response.status_code == 404:
             return None
         else:
-            msg = "Please provide the correct resource type for this endpoint"
+            msg = f"MNS subscription failed"
             raise UnhandledResponseError(response=response.json(), message=msg)

mns_subscription/tests/test_mns_service.py

@@ -17,7 +17,7 @@ def setUp(self):
     def test_successful_subscription(self, mock_post):
         # Arrange
         mock_response = MagicMock()
-        mock_response.status_code = 200
+        mock_response.status_code = 201
         mock_response.json.return_value = {"subscriptionId": "abc123"}
         mock_post.return_value = mock_response
 
@@ -54,7 +54,7 @@ def test_unhandled_error(self, mock_post):
         with self.assertRaises(UnhandledResponseError) as context:
             service.subscribe_notification()
 
-        self.assertIn("MNS subscription failed with status", str(context.exception))
+        self.assertIn("MNS subscription failed", str(context.exception))
 
 
 if __name__ == "__main__":