def test_setup_custom_values(self, getLogger, WatchedFileHandler): logger = logging.getLogger.return_value = Mock() logger.handlers = [] log.setup_logging(loglevel=logging.DEBUG, logfile='/var/logfile') logger.setLevel.assert_called_with(logging.DEBUG) self.assertTrue(logger.addHandler.called) self.assertTrue(WatchedFileHandler.called)
def test_setup_custom_values(self, getLogger, WatchedFileHandler): logger = logging.getLogger.return_value = Mock() logger.handlers = [] setup_logging(loglevel=logging.DEBUG, logfile='/var/logfile') logger.setLevel.assert_called_with(logging.DEBUG) self.assertTrue(logger.addHandler.called) self.assertTrue(WatchedFileHandler.called)
def test_set_up_default_values(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [] log.setup_logging() logger.setLevel.assert_called_with(logging.ERROR) self.assertTrue(logger.addHandler.called) ah_args, _ = logger.addHandler.call_args handler = ah_args[0] self.assertIsInstance(handler, logging.StreamHandler) self.assertIs(handler.stream, sys.__stderr__)
def test_set_up_default_values(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [] setup_logging() logger.setLevel.assert_called_with(logging.ERROR) logger.addHandler.assert_called() ah_args, _ = logger.addHandler.call_args handler = ah_args[0] assert isinstance(handler, logging.StreamHandler) assert handler.stream is sys.__stderr__
def test_set_up_default_values(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [] setup_logging() logger.setLevel.assert_called_with(logging.ERROR) self.assertTrue(logger.addHandler.called) ah_args, _ = logger.addHandler.call_args handler = ah_args[0] self.assertIsInstance(handler, logging.StreamHandler) self.assertIs(handler.stream, sys.__stderr__)
def start(self): from kombu.log import setup_logging setup_logging() with Connection(self.broker_url) as conn: self.log.info("Connected to %s" % self.broker_url) self.conn = conn self.rkey = ".".join(["components", self.name, "inbox"]) self.ex = Exchange("components", 'topic', durable=True) self.q = Queue('inbox', exchange=self.ex, routing_key=self.rkey) maybe_declare( self.ex, conn ) self.log.info("Declared Exchange: '%s' (%s)" % (self.ex.name, self.ex.type) ) maybe_declare( self.q, conn ) self.log.info("Declared Queue: %s with routing key: '%s' " % (self.q, self.rkey) ) with Consumer(conn.channel(), queues=self.q, callbacks=[self.on_receive]) as c: while self.continue_to_drain: try: conn.drain_events() c.consume() except socket.timeout as e: pass
# This Source Code Form is subject to the terms of the Mozilla Public # License, v. 2.0. If a copy of the MPL was not distributed with this # file, You can obtain one at http://mozilla.org/MPL/2.0/. from __future__ import with_statement import time from kombu import Connection from kombu.mixins import ConsumerMixin from kombu.log import setup_logging from .queues import hg_queues from pushes.utils import handlePushes, PushJS logger = setup_logging('INFO') class Worker(ConsumerMixin): def __init__(self, connection, settings): self.connection = connection self.retries = 0 self.max_retries = settings.MAX_HG_RETRIES self.sentry = None if hasattr(settings, 'RAVEN_CONFIG'): from raven import Client self.sentry = Client(**settings.RAVEN_CONFIG) def get_consumers(self, Consumer, channel): return [Consumer(queues=hg_queues,
def test_logger_already_setup(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [Mock()] log.setup_logging() self.assertFalse(logger.setLevel.called)
def run_from_commandline(self, loglevel='INFO', logfile=None): setup_logging(loglevel, logfile) try: self.run() except KeyboardInterrupt: self.info('[Quit requested by user]')
def test_logger_already_setup(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [Mock()] setup_logging() self.assertFalse(logger.setLevel.called)
def run_from_commandline(self, loglevel=None, logfile=None): setup_logging(loglevel, logfile) try: self.run() except KeyboardInterrupt: self.info("[Quit requested by user]")
def test_logger_already_setup(self, getLogger): logger = logging.getLogger.return_value = Mock() logger.handlers = [Mock()] setup_logging() logger.setLevel.assert_not_called()
from __future__ import with_statement from kombu.mixins import ConsumerMixin from kombu.utils import kwdict from queues import task_queues class Worker(ConsumerMixin): def get_consumers(self, Consumer, channel): return Consumer(queues=task_queues, callbacks=[self.process_task]) def process_task(body, message): fun = body["fun"] args = body["args"] kwargs = body["kwargs"] fun(*args, **kwdict(kwargs)) message.ack() if __name__ == "__main__": from kombu import BrokerConnection from kombu.log import setup_logging setup_logging(loglevel="INFO") with BrokerConnection("amqp://*****:*****@localhost:5672//") as conn: Worker(conn).run()