Explorar el Código

bench_worker: Use amqplib if running under PyPy

Ask Solem hace 13 años
padre
commit
7938a95135
Se han modificado 1 ficheros con 5 adiciones y 1 borrados
  1. 5 1
      funtests/benchmarks/bench_worker.py

+ 5 - 1
funtests/benchmarks/bench_worker.py

@@ -15,8 +15,12 @@ from celery import Celery
 
 DEFAULT_ITS = 20000
 
+BROKER_TRANSPORT = "librabbitmq"
+if hasattr(sys, "pypy_version_info"):
+    BROKER_TRANSPORT = "amqplib"
+
 celery = Celery(__name__)
-celery.conf.update(BROKER_TRANSPORT="librabbitmq",
+celery.conf.update(BROKER_TRANSPORT=BROKER_TRANSPORT,
                    BROKER_POOL_LIMIT=10,
                    CELERYD_POOL="solo",
                    CELERY_PREFETCH_MULTIPLIER=0,