How to Mock AWS SQS Send Receive Delete in Python

import json
from collections import deque
from uuid import uuid4


class MockSQSQueue:
    def __init__(self, name):
        self.name = name
        self._messages = deque()
        self._in_flight = {}

    def send_message(self, body, attributes=None):
        message_id = str(uuid4())
        message = {
            "MessageId": message_id,
            "Body": body,
            "Attributes": attributes or {},
        }
        self._messages.append(message)
        return {"MessageId": message_id}

    def receive_message(self, max_number=1, visibility_timeout=30):
        received = []
        for _ in range(min(max_number, len(self._messages))):
            message = self._messages.popleft()
            self._in_flight[message["MessageId"]] = message
            received.append(message)
        return received

    def delete_message(self, receipt_handle):
        # In this mock, receipt_handle is the MessageId
        if receipt_handle in self._in_flight:
            del self._in_flight[receipt_handle]
            return True
        return False


if __name__ == "__main__":
    queue = MockSQSQueue("orders")
    queue.send_message(json.dumps({"order_id": 1, "item": "laptop"}))
    queue.send_message(json.dumps({"order_id": 2, "item": "mouse"}))

    msgs = queue.receive_message(max_number=2)
    for msg in msgs:
        print(f"Received: {msg}")

    deleted = [queue.delete_message(m["MessageId"]) for m in msgs]
    print(f"Deleted: {deleted}")
    print(f"Remaining in queue: {len(queue._messages)}")