fix perf tests

Author: maelorn

mrq/job.py

@@ -655,8 +655,9 @@ def queue_job(main_task_path, params, **kwargs):
 def set_queues_size(size_by_queues, action="incr"):
     if len(size_by_queues) > 0:
         with context.connections.redis.pipeline(transaction=False) as pipe:
+            action_func = getattr(pipe, action)
             for queue in size_by_queues:
-                getattr(pipe, action)("queuesize:%s" % queue, amount=size_by_queues[queue])
+                action_func("queuesize:%s" % queue, amount=size_by_queues[queue])
                 pipe.expire("queuesize:%s" % queue, context.get_current_config().get("queue_ttl"))
             pipe.execute()
 

tests/test_queuesize.py

@@ -3,16 +3,16 @@ def test_job_queue(worker):
   from mrq.context import connections
   worker.start()
   worker.send_task("tests.tasks.general.Wait", {}, block=False)
-  assert connections.redis.get("queuesize:%s" % "default") == "1"
+  assert int(connections.redis.get("queuesize:%s" % "default")) == 1
   worker.wait_for_idle()
-  assert connections.redis.get("queuesize:%s" % "default") == "0"
+  assert int(connections.redis.get("queuesize:%s" % "default")) == 0
 
 def test_job_failed(worker):
   from mrq.context import connections
   worker.start()
   worker.send_task("tests.tasks.general.RaiseException", {}, block=False)
   worker.wait_for_idle()
-  assert connections.redis.get("queuesize:%s" % "default") == "0"
+  assert int(connections.redis.get("queuesize:%s" % "default")) == 0
 
 def test_job_requeue(worker):
   from mrq.context import connections
@@ -21,7 +21,7 @@ def test_job_requeue(worker):
   worker.start()
   job_id = worker.send_task("tests.tasks.general.RaiseException", {}, block=False)
   worker.wait_for_idle()
-  assert connections.redis.get("queuesize:%s" % "default") == "0"
+  assert int(connections.redis.get("queuesize:%s" % "default")) == 0
 
   Job(job_id).requeue()
-  assert connections.redis.get("queuesize:%s" % "default") == "1"
+  assert int(connections.redis.get("queuesize:%s" % "default")) == 1