Esempio n. 1
0
    def test_client_passes_additional_args_to_client(self, mock_redis_class):
        self.client = Client("redis://url", 4, socket_timeout=5)

        mock_redis_class.from_url.assert_called_with("redis://url",
                                                     4,
                                                     decode_responses=True,
                                                     socket_timeout=5)
Esempio n. 2
0
    def setUp(self, mock_redis_class):
        self.mock_strict_redis = mock_redis_class.from_url.return_value

        self.task_callback = mock.Mock()

        self.client = Client("redis://asdf:1234")
        self.runner = forking_runner.ForkingRunner(self.client, "some.queue",
                                                   self.task_callback)
Esempio n. 3
0
    def setUp(self, mock_redis_queue_class, mock_strict_redis):
        self.mock_strict_redis = mock_strict_redis.from_url.return_value

        self.mock_redis_queue_class = mock_redis_queue_class
        self.mock_redis_queue = mock_redis_queue_class.return_value

        self.client = Client("redis://asdf:1234")
        self.queue = self.client.get_queue("some.queue")
Esempio n. 4
0
    def setUp(self, mock_redis_queue_class, mock_strict_redis):
        self.mock_redis_queue = mock_redis_queue_class.return_value
        self.mock_strict_redis = mock_strict_redis.from_url.return_value

        self.mock_strict_redis.hgetall.return_value = {
            "queue": "some.queue",
            "status": "reserved",
            "node": "host_1234",
            "parameters": "some parameters"
        }

        self.client = Client("redis://asdf:1234")
        self.task = self.client.get_task("some_task")
        self.processor = self.client.get_processor(self.task)
Esempio n. 5
0
    def setUp(self, redis_class):
        self.mock_redis = redis_class.from_url.return_value

        self.client = Client("redis://asdf:1234")
Esempio n. 6
0
# This hasn't actually been run; it's just a scratch-pad, for now, to
# figure out what API we want

from __future__ import print_function
from blueque import Client
from blueque import forking_runner

import time


def do_work(task):
    print(task.id, task.parameters)

    time.sleep(1000)

    return "result"


if __name__ == "__main__":
    client = Client("redis://localhost")

    forking_runner.run(client, "some.queue", do_work, 4)
Esempio n. 7
0
    def test_client_connects_with_requested_information(
            self, mock_redis_class):
        self.client = Client("redis://url")

        mock_redis_class.from_url.assert_called_with("redis://url",
                                                     decode_responses=True)