"""Test loading an SQS queue."""

import boto3


class SQSAdapter():
    """SQS Publisher Adapter."""

    def __init__(self, url):
        """Connect to queue."""
        session = boto3.Session(profile_name='default')
        sqs = session.resource('sqs')
        self._queue = sqs.Queue(url)

    def publish(self, limit, data):
        for i in range(limit):
            self._queue.send_message(MessageBody=data)

    def close(self):
        del self._queue
        del self._connection