Messages larger than 1mb caught.

TODO: find a way so that the message is not removed if its larger than 1mb
Signed-off-by: jounaidr <[email protected]>

ssm/ssm2.py

@@ -16,6 +16,7 @@
 from __future__ import print_function
 
 import pkg_resources
+from pympler import asizeof
 
 from ssm import crypto
 from ssm.message_directory import MessageDirectory
@@ -390,8 +391,13 @@ def _send_msg_ams(self, text, msgid):
             message = AmsMessage(data=to_send,
                                  attributes={'empaid': msgid}).dict()
 
-            argo_response = self._ams.publish(self._dest, message, retry=3)
-            return argo_response['messageIds'][0]
+            # Check the encrypted and wrapped message is less than 1mb
+            if asizeof.asizeof(message) < 1048576:
+                argo_response = self._ams.publish(self._dest, message, retry=3)
+                return argo_response['messageIds'][0]
+
+            else:
+                log.error('Message %s could not be sent as its larger than 1MB', msgid)
 
     def pull_msg_ams(self):
         """Pull 1 message from the AMS and acknowledge it."""