Adding MessageGroupId and MessageDeduplicationId (#3163)

Adding MessageGroupId and MessageDeduplicationId when sent from batch (#3101)
2020-07-22 08:09:12 -04:00 · 2020-07-22 08:09:12 -04:00 · bf8eb11dc3
commit bf8eb11dc3
parent b09c8034e6
3 changed files with 28 additions and 0 deletions
--- a/moto/sqs/models.py
+++ b/moto/sqs/models.py
@ -685,6 +685,8 @@ class SQSBackend(BaseBackend):
                entry["MessageBody"],
                message_attributes=entry["MessageAttributes"],
                delay_seconds=entry["DelaySeconds"],
+                group_id=entry.get("MessageGroupId"),
+                deduplication_id=entry.get("MessageDeduplicationId"),
            )
            message.user_id = entry["Id"]

--- a/moto/sqs/responses.py
+++ b/moto/sqs/responses.py
@ -291,6 +291,16 @@ class SQSResponse(BaseResponse):
                        [None],
                    )[0],
                    "MessageAttributes": message_attributes,
+                    "MessageGroupId": self.querystring.get(
+                        "SendMessageBatchRequestEntry.{}.MessageGroupId".format(index),
+                        [None],
+                    )[0],
+                    "MessageDeduplicationId": self.querystring.get(
+                        "SendMessageBatchRequestEntry.{}.MessageDeduplicationId".format(
+                            index
+                        ),
+                        [None],
+                    )[0],
                }

        if entries == {}: