"""Test loading a RabbitMQ queue."""

import pika


class RabbitMQ():

    def __init__(self, host, queue):
        self._connection = pika.BlockingConnection(
            pika.ConnectionParameters(host))
        self._channel = self._connection.channel()
        self._channel.queue_declare(queue=queue)

    def publish(self, limit, queue, data):
        """Publish."""
        for i in range(limit):
            self._channel.basic_publish(
                exchange='', routing_key=queue, body=data)

    def close(self):
        self._channel.close()
        self._connection.close()