From 97832d6a0d1c574666a4f110f54e0782220a0d8f Mon Sep 17 00:00:00 2001 From: Brandon Joyce Date: Wed, 22 Nov 2023 23:47:43 -0500 Subject: [PATCH] Add message_id parameter --- AUTHORS.md | 9 +++++---- pyas2lib/as2.py | 16 ++++++++++++---- pyas2lib/tests/test_basic.py | 8 ++++++++ 3 files changed, 25 insertions(+), 8 deletions(-) diff --git a/AUTHORS.md b/AUTHORS.md index 3c75246..f8b47d9 100644 --- a/AUTHORS.md +++ b/AUTHORS.md @@ -1,4 +1,5 @@ -* Abhishek Ram @abhishek-ram -* Chad Gates @chadgates -* Bruno Ribeiro da Silva @loop0 -* Robin C Samuel @robincsamuel \ No newline at end of file +- Abhishek Ram @abhishek-ram +- Chad Gates @chadgates +- Bruno Ribeiro da Silva @loop0 +- Robin C Samuel @robincsamuel +- Brandon Joyce @brandonjoyce diff --git a/pyas2lib/as2.py b/pyas2lib/as2.py index cb058ad..dec3144 100644 --- a/pyas2lib/as2.py +++ b/pyas2lib/as2.py @@ -330,6 +330,7 @@ def build( content_type="application/edi-consent", additional_headers=None, disposition_notification_to="no-reply@pyas2.com", + 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 "no-reply@pyas2.com") + + :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 = { diff --git a/pyas2lib/tests/test_basic.py b/pyas2lib/tests/test_basic.py index 3eadbb8..3a90e98 100644 --- a/pyas2lib/tests/test_basic.py +++ b/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