Add message_id parameter

AUTHORS.md

@@ -1,4 +1,5 @@
-* Abhishek Ram <[email protected]> @abhishek-ram
-* Chad Gates @chadgates
-* Bruno Ribeiro da Silva <[email protected]> @loop0 
-* Robin C Samuel @robincsamuel
+- Abhishek Ram <[email protected]> @abhishek-ram
+- Chad Gates @chadgates
+- Bruno Ribeiro da Silva <[email protected]> @loop0
+- Robin C Samuel @robincsamuel
+- Brandon Joyce @brandonjoyce

pyas2lib/as2.py

@@ -330,6 +330,7 @@ def build(
         content_type="application/edi-consent",
         additional_headers=None,
         disposition_notification_to="[email protected]",
+        message_id=None,
     ):
 
         """Function builds the AS2 message. Compresses, signs and encrypts
@@ -354,6 +355,10 @@ def build(
         :param disposition_notification_to:
             Email address for disposition-notification-to header entry.
             (default "[email protected]")
+
+        :param message_id:
+            The message id to be used for the message. If not provided a
+            unique message id is generated. (default None)
         """
 
         # Validations
@@ -372,10 +377,13 @@ def build(
                 "Encryption of messages is enabled but encrypt key is not set for the receiver."
             )
 
-        # Generate message id using UUID 1 as it uses both hostname and time
-        self.message_id = (
-            email_utils.make_msgid(domain=self.sender.domain).lstrip("<").rstrip(">")
-        )
+        if message_id:
+            self.message_id = message_id  
+        else:
+            # Generate message id using UUID 1 as it uses both hostname and time
+            self.message_id = (
+                email_utils.make_msgid(domain=self.sender.domain).lstrip("<").rstrip(">")
+            )
 
         # Set up the message headers
         as2_headers = {

pyas2lib/tests/test_basic.py

@@ -200,6 +200,14 @@ def test_plain_message_without_domain(self):
         out_message.build(self.test_data)
         self.assertEqual(out_message.message_id.split("@")[1], socket.getfqdn())
 
+    def test_plain_message_with_custom_message_id(self):
+        """Test Message building with a custom message id"""
+
+        # Build an As2 message to be transmitted to partner
+        out_message = as2.Message(self.org, self.partner)
+        out_message.build(self.test_data, message_id="some_custom_id")
+        self.assertEqual(out_message.message_id, "some_custom_id")
+
     def find_org(self, as2_id):
         return self.org