"""Utilities to create mock SQS messages."""

import json
from uuid import uuid4


def generate_fake_message_body(message_id):
    """Create a fake SQS message.

    message_id (int): Fake message id.

    Returns:
        str: Fake message body encoding to json-string.
    """
    message_body = {
        'message_id': message_id,
        'message_uuid': str(uuid4())
    }
    return json.dumps(message_body)


def put_sqs_messages(sqs_queue, number_of_messages):
    """Put a number of fake SQS messages into the queue.

    Args:
        sqs_queue: SQS queue (or moto SQS queue mock) to write the messages to
        number_of_messages (int): number of messages to be written
    """
    for i in range(0, number_of_messages):
        message_body = generate_fake_message_body(i)
        sqs_queue.send_message(MessageBody=message_body)