"""Test loading a redis queue."""

import redis


class RedisAdapter():

    def __init__(self, host):
        self._connection = redis.Redis(host=host)

    def publish(self, limit, queue, data):
        """Publish."""
        for i in range(limit):
            self._connection.lpush(queue, data)

    def close(self):
        del self._connection