"""Utilities to create mock SQS messages."""
from uuid import uuid4

from label_copy_export.connectors import sqs


def create_sqs_message(payload):
    """Create a fake SQS message.

    Returns:
        JSONMessageExt: new SQS message instance
    """
    message_attributes = {
        'Correlation-Id': {
            'data_type': 'String',
            'string_value': str(uuid4())
        }
    }

    message = sqs.JSONMessageExt()
    message.set_body(payload)
    message.message_attributes = message_attributes
    return message


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
    """
    payload = {
        'release_id': '123456789',
    }
    for i in range(0, number_of_messages):
        message = create_sqs_message(payload)
        sqs_queue.write(message)