import base64
import json
from uuid import uuid4


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

    Returns:
        JSONMessageExt: new SQS message instance
    """

    payload = {
        'content_owner_id': 'theorchardmusic',
        'isrc': str(uuid4()),
        'territories': ['US', 'CA', 'RU', 'UA']
    }

    message_attributes = {
        'Correlation-Id': {
            'DataType': 'String',
            'StringValue': str(uuid4())
        }
    }

    message_body = json.dumps(payload)
    message_body = base64.b64encode(
        message_body.encode('utf-8')).decode('utf-8')

    return {'message_body': message_body,
            'message_attributes': message_attributes}


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 = create_sqs_message()
        sqs_queue.send_message(
            MessageBody=message.get('message_body'),
            MessageAttributes=message.get('message_attributes'))