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()
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()
