def test_uninstrument_new_brokers(self): new_broker = LocalBroker() remoulade.set_broker(new_broker) new_broker.declare_actor(actor_div) actor_div.send(1, 1) spans = self.sorted_spans(self.memory_exporter.get_finished_spans()) self.assertEqual(len(spans), 0)
def local_broker(): broker = LocalBroker() broker.emit_after("process_boot") remoulade.set_broker(broker) yield broker broker.flush_all() broker.emit_before("process_stop") broker.close()
def setUp(self): super().setUp() RemouladeInstrumentor().instrument() broker = LocalBroker() remoulade.set_broker(broker) broker.declare_actor(actor_div)
def rabbitmq_broker(): broker = RabbitmqBroker(max_priority=10) check_rabbitmq(broker) broker.emit_after("process_boot") remoulade.set_broker(broker) yield broker broker.flush_all() broker.emit_before("process_stop") broker.close()
def rabbitmq_broker(): rmq_url = os.getenv("REMOULADE_TEST_RABBITMQ_URL") or "amqp://*****:*****@localhost:5784" broker = RabbitmqBroker(max_priority=10, url=rmq_url) check_rabbitmq(broker) broker.emit_after("process_boot") remoulade.set_broker(broker) yield broker broker.flush_all() broker.emit_before("process_stop") broker.close()
import celery import pylibmc import remoulade from remoulade.brokers.rabbitmq import RabbitmqBroker from remoulade.brokers.redis import RedisBroker logger = logging.getLogger("example") counter_key = "latench-bench-counter" memcache_client = pylibmc.Client(["localhost"], binary=True) memcache_pool = pylibmc.ClientPool(memcache_client, 8) random.seed(1337) if os.getenv("REDIS") == "1": broker = RedisBroker() remoulade.set_broker(broker) celery_app = celery.Celery(broker="redis:///") else: broker = RabbitmqBroker(host="127.0.0.1") remoulade.set_broker(broker) celery_app = celery.Celery(broker="amqp:///") def fib_bench(n): p, q = 0, 1 while n > 0: p, q = q, p + q n -= 1 with memcache_pool.reserve() as client: