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)
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)
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")
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)
def setUp(self, redis_class): self.mock_redis = redis_class.from_url.return_value self.client = Client("redis://asdf:1234")
# 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)
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)