SQS: add SequenceNumber to FIFO queue SendMessage responses (getmoto#9789)

Author: bpandola

moto/sqs/responses.py

@@ -152,6 +152,7 @@ def send_message(self) -> ActionResult:
         system_message_attributes = self._get_param("MessageSystemAttributes")
         validate_message_attributes(system_message_attributes)
         queue_name = self._get_queue_name()
+        queue = self.sqs_backend.get_queue(queue_name)
         message = self.sqs_backend.send_message(
             queue_name,
             message,
@@ -167,11 +168,13 @@ def send_message(self) -> ActionResult:
         }
         if len(message.message_attributes) > 0:
             resp["MD5OfMessageAttributes"] = message.attribute_md5
+        if queue.fifo_queue and message.sequence_number:
+            resp["SequenceNumber"] = message.sequence_number
         return ActionResult(resp)
 
     def send_message_batch(self) -> ActionResult:
         queue_name = self._get_queue_name()
-        self.sqs_backend.get_queue(queue_name)
+        queue = self.sqs_backend.get_queue(queue_name)
         entries = self._get_param("Entries", [])
         entries = {str(idx): entry for idx, entry in enumerate(entries)}
         for entry in entries.values():
@@ -209,6 +212,8 @@ def send_message_batch(self) -> ActionResult:
             }
             if len(msg.message_attributes) > 0:
                 msg_dict["MD5OfMessageAttributes"] = msg.attribute_md5
+            if queue.fifo_queue:
+                msg_dict["SequenceNumber"] = msg.sequence_number
             resp["Successful"].append(msg_dict)
         return ActionResult(resp)
 

tests/test_sqs/test_sqs.py

@@ -462,6 +462,7 @@ def test_message_send_without_attributes():
     assert msg.get("MD5OfMessageBody") == "58fd9edd83341c29f1aebba81c31e257"
     assert "MD5OfMessageAttributes" not in msg
     assert " \n" not in msg.get("MessageId")
+    assert "SequenceNumber" not in msg
 
     messages = queue.receive_messages()
     assert len(messages) == 1
@@ -1357,12 +1358,13 @@ def test_fifo_send_receive_message_with_attribute_name(attribute_name, expected)
 
     body = "this is a test message"
 
-    client.send_message(
+    resp = client.send_message(
         QueueUrl=queue_url,
         MessageBody=body,
         MessageDeduplicationId="123",
         MessageGroupId="456",
     )
+    assert "SequenceNumber" in resp
 
     message = client.receive_message(
         QueueUrl=queue_url, AttributeNames=[attribute_name], MaxNumberOfMessages=2
@@ -1850,7 +1852,7 @@ def test_delete_message_using_old_receipt_handle():
 
 
 @mock_aws
-def test_send_message_batch():
+def test_fifo_send_message_batch():
     client = boto3.client("sqs", region_name=REGION)
     response = client.create_queue(
         QueueName=f"{str(uuid4())[0:6]}.fifo", Attributes={"FifoQueue": "true"}
@@ -1887,6 +1889,8 @@ def test_send_message_batch():
     )
 
     assert sorted([entry["Id"] for entry in response["Successful"]]) == ["id_1", "id_2"]
+    for entry in response["Successful"]:
+        assert "SequenceNumber" in entry
 
     response = client.receive_message(
         QueueUrl=queue_url,
@@ -3346,6 +3350,8 @@ def test_message_delay_is_more_than_15_minutes():
 
     assert sorted([entry["Id"] for entry in response["Successful"]]) == ["id_1"]
     assert sorted([entry["Id"] for entry in response["Failed"]]) == ["id_2"]
+    for entry in response["Successful"]:
+        assert "SequenceNumber" not in entry
 
     time.sleep(1.1)